eSlider
fec5325c7a
- bin/chats/: env-based paths, no /mnt/ /home/ hardcodes - bin/edelweiss-pilot: remove curasoft, use DOCS_BASE env var - bin/facts/crm: use KNOWLEDGE_MESH_SEED env var - compose.edelweiss.yml: remove curasoft volumes, use DOCS_BASE - docs/chat-import-plan.md: link to Gitea issue, no secrets - bin/seed-edelweiss-facts.py: removed (curasoft-only)
88 lines
2.4 KiB
Go
88 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func runSyncTelegram(args []string) int {
|
|
fs := flag.NewFlagSet("chats sync telegram", flag.ContinueOnError)
|
|
limit := fs.Int("limit", 0, "max messages per chat (0 = all)")
|
|
phone := fs.String("phone", "", "phone number (default env TELEGRAM_PHONE)")
|
|
help := fs.Bool("help", false, "")
|
|
fs.SetOutput(os.Stderr)
|
|
if err := fs.Parse(args); err != nil {
|
|
return 2
|
|
}
|
|
if *help {
|
|
fmt.Fprintln(os.Stderr, "usage: chats sync telegram [--limit N] [--phone PHONE]")
|
|
return 0
|
|
}
|
|
|
|
apiIDStr := envVar("TELEGRAM_API_ID", "")
|
|
apiHash := envVar("TELEGRAM_API_HASH", "")
|
|
sessionStr := envVar("TELEGRAM_SESSION_STRING", "")
|
|
phoneNum := *phone
|
|
if phoneNum == "" {
|
|
phoneNum = envVar("TELEGRAM_PHONE", "")
|
|
}
|
|
if apiIDStr == "" || apiHash == "" || phoneNum == "" {
|
|
fmt.Fprintln(os.Stderr, "chats: need TELEGRAM_API_ID, TELEGRAM_API_HASH, TELEGRAM_PHONE in env")
|
|
return 2
|
|
}
|
|
apiID, err := strconv.Atoi(apiIDStr)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "chats: invalid TELEGRAM_API_ID %q\n", apiIDStr)
|
|
return 2
|
|
}
|
|
|
|
mcpDir := envVar("TELEGRAM_MCP_DIR", "")
|
|
if mcpDir == "" {
|
|
fmt.Fprintln(os.Stderr, "chats: set TELEGRAM_MCP_DIR to telegram-mcp directory")
|
|
return 1
|
|
}
|
|
if _, err := os.Stat(filepath.Join(mcpDir, "main.py")); err != nil {
|
|
fmt.Fprintf(os.Stderr, "chats: TELEGRAM_MCP_DIR=%s: main.py not found\n", mcpDir)
|
|
return 1
|
|
}
|
|
|
|
if sessionStr == "" {
|
|
envPath := filepath.Join(mcpDir, ".env")
|
|
if data, err := os.ReadFile(envPath); err == nil {
|
|
for _, line := range strings.Split(string(data), "\n") {
|
|
if strings.HasPrefix(line, "TELEGRAM_SESSION_STRING=") {
|
|
sessionStr = strings.TrimPrefix(line, "TELEGRAM_SESSION_STRING=")
|
|
sessionStr = strings.Trim(sessionStr, "\"'")
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if sessionStr == "" {
|
|
sessionStr = envVar("TELEGRAM_SESSION_STRING", "")
|
|
}
|
|
if sessionStr == "" {
|
|
fmt.Fprintln(os.Stderr, "chats: TELEGRAM_SESSION_STRING not found; set env or in TELEGRAM_MCP_DIR/.env")
|
|
return 1
|
|
}
|
|
|
|
src := NewTelegramMCPSource(apiID, apiHash, phoneNum, sessionStr, mcpDir)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Minute)
|
|
defer cancel()
|
|
|
|
start := time.Now()
|
|
if err := src.Sync(ctx, chatsDir(), *limit); err != nil {
|
|
fmt.Fprintf(os.Stderr, "chats sync telegram: %v\n", err)
|
|
return 1
|
|
}
|
|
fmt.Printf("chats sync telegram: completed in %s\n", time.Since(start).Round(time.Millisecond))
|
|
return 0
|
|
}
|