Add support for custom Telegram tdata directory path (#31)

Co-authored-by: User <user@example.com>

Author: zanzhz1101

docs/telegram.md

@@ -75,6 +75,7 @@ vget telegram status                 # Show login state
   - macOS: `~/Library/Application Support/Telegram Desktop/tdata/`
   - Linux: `~/.local/share/TelegramDesktop/tdata/`
   - Windows: `%APPDATA%/Telegram Desktop/tdata/`
+  - **Custom path**: Set via `vget config set telegram.tdata_path /path/to/tdata`
 - Imports session using Desktop's API credentials (2040)
 - Session stored in `~/.config/vget/telegram/desktop-session.json`
 

internal/cli/telegram.go

@@ -13,6 +13,7 @@ import (
 	"github.com/gotd/td/telegram"
 	"github.com/gotd/td/tg"
 	tgpkg "github.com/guiyumin/vget/internal/core/extractor/telegram"
+	"github.com/guiyumin/vget/internal/core/config"
 	"github.com/spf13/cobra"
 )
 
@@ -63,11 +64,17 @@ func runTelegramLogin(cmd *cobra.Command, args []string) error {
 		return nil
 	}
 
-	// Find Telegram Desktop tdata directory
-	tdataPath := getTelegramDesktopPath()
+	// Check config for custom Telegram directory
+	cfg := config.LoadOrDefault()
+	tdataPath := cfg.Telegram.TDataPath
+	
+	// If no custom path, use default locations
 	if tdataPath == "" {
-		return fmt.Errorf("could not find Telegram Desktop data directory.\n" +
-			"Make sure Telegram Desktop is installed and you're logged in")
+		tdataPath = getTelegramDesktopPath()
+		if tdataPath == "" {
+			return fmt.Errorf("could not find Telegram Desktop data directory.\n"+
+				"Make sure Telegram Desktop is installed and you're logged in")
+		}
 	}
 
 	fmt.Printf("Found Telegram Desktop at: %s\n", tdataPath)

internal/core/config/config.go

@@ -81,6 +81,9 @@ type Config struct {
 
 	// Bilibili configuration
 	Bilibili BilibiliConfig `yaml:"bilibili,omitempty"`
+
+	// Telegram configuration
+	Telegram TelegramConfig `yaml:"telegram,omitempty"`
 }
 
 // BilibiliConfig holds Bilibili authentication settings
@@ -89,6 +92,12 @@ type BilibiliConfig struct {
 	Cookie string `yaml:"cookie,omitempty"`
 }
 
+// TelegramConfig holds Telegram authentication settings
+type TelegramConfig struct {
+	// TDataPath is the custom path to Telegram Desktop tdata directory
+	TDataPath string `yaml:"tdata_path,omitempty"`
+}
+
 // TorrentConfig holds configuration for remote torrent client integration
 type TorrentConfig struct {
 	// Enabled determines if torrent dispatch feature is active

internal/server/server.go

@@ -542,7 +542,8 @@ func (s *Server) handleGetConfig(c *gin.Context) {
 			"express":               cfg.Express,
 			"torrent_enabled":       cfg.Torrent.Enabled,
 			"bilibili_cookie":       cfg.Bilibili.Cookie,
-		},
+			"telegram_tdata_path":   cfg.Telegram.TDataPath,
+			},
 		Message: "config retrieved",
 	})
 }
@@ -1170,6 +1171,8 @@ func (s *Server) setConfigValue(cfg *config.Config, key, value string) error {
 		cfg.Server.APIKey = value
 	case "bilibili.cookie", "bilibili_cookie":
 		cfg.Bilibili.Cookie = value
+	case "telegram.tdata_path", "telegram_tdata_path":
+		cfg.Telegram.TDataPath = value
 	default:
 		return fmt.Errorf("unknown config key: %s", key)
 	}

ui/src/components/ConfigEditor.tsx

@@ -39,6 +39,7 @@ interface ConfigEditorProps {
   initialApiKey: string;
   initialKuaidi100Key: string;
   initialKuaidi100Customer: string;
+  initialTelegramTdataPath: string;
   serverPort: number;
   webdavServers: Record<string, WebDAVServer>;
   // Callbacks
@@ -62,6 +63,7 @@ export interface ConfigValues {
   apiKey: string;
   kuaidi100Key: string;
   kuaidi100Customer: string;
+  telegramTdataPath: string;
 }
 
 export function ConfigEditor({
@@ -74,6 +76,7 @@ export function ConfigEditor({
   initialApiKey,
   initialKuaidi100Key,
   initialKuaidi100Customer,
+  initialTelegramTdataPath,
   serverPort,
   webdavServers,
   onSave,
@@ -100,6 +103,9 @@ export function ConfigEditor({
   const [pendingKuaidi100Customer, setPendingKuaidi100Customer] = useState(
     initialKuaidi100Customer || ""
   );
+  const [pendingTelegramTdataPath, setPendingTelegramTdataPath] = useState(
+    initialTelegramTdataPath || ""
+  );
 
   // WebDAV add form
   const [newWebDAVName, setNewWebDAVName] = useState("");
@@ -120,6 +126,7 @@ export function ConfigEditor({
         apiKey: pendingApiKey,
         kuaidi100Key: pendingKuaidi100Key,
         kuaidi100Customer: pendingKuaidi100Customer,
+        telegramTdataPath: pendingTelegramTdataPath,
       });
     } finally {
       setSavingConfig(false);
@@ -136,6 +143,7 @@ export function ConfigEditor({
     setPendingApiKey(initialApiKey || "");
     setPendingKuaidi100Key(initialKuaidi100Key || "");
     setPendingKuaidi100Customer(initialKuaidi100Customer || "");
+    setPendingTelegramTdataPath(initialTelegramTdataPath || "");
     // Reset WebDAV form
     setNewWebDAVName("");
     setNewWebDAVUrl("");
@@ -296,6 +304,24 @@ export function ConfigEditor({
         </div>
       </div>
 
+      {/* Telegram Section */}
+      <div className="text-sm font-semibold text-zinc-900 dark:text-white mt-4 mb-2 pt-3 border-t border-zinc-300 dark:border-zinc-700">
+        Telegram
+      </div>
+      <div className="flex flex-col sm:flex-row sm:items-center gap-1 sm:gap-3">
+        <span className="sm:min-w-25 text-sm text-zinc-700 dark:text-zinc-200">
+          TData Path
+        </span>
+        <input
+          type="text"
+          className={inputBaseClass}
+          placeholder="Custom Telegram Desktop tdata directory path"
+          value={pendingTelegramTdataPath}
+          onChange={(e) => setPendingTelegramTdataPath(e.target.value)}
+          disabled={!isConnected || savingConfig}
+        />
+      </div>
+
       {/* WebDAV Servers Section */}
       <div className="mt-4 pt-4 border-t border-zinc-300 dark:border-zinc-700">
         <div className="text-sm font-semibold text-zinc-900 dark:text-white mb-3">

ui/src/context/AppContext.tsx

@@ -57,6 +57,7 @@ interface AppContextType {
   kuaidi100Customer: string;
   configExists: boolean;
   torrentEnabled: boolean;
+  telegramTdataPath: string;
 
   // Translations
   t: UITranslations;
@@ -112,6 +113,7 @@ export function AppProvider({ children }: { children: ReactNode }) {
   const [kuaidi100Key, setKuaidi100Key] = useState("");
   const [kuaidi100Customer, setKuaidi100Customer] = useState("");
   const [torrentEnabled, setTorrentEnabled] = useState(false);
+  const [telegramTdataPath, setTelegramTdataPath] = useState("");
   const [toasts, setToasts] = useState<ToastData[]>([]);
 
   const showToast = useCallback((type: ToastType, message: string) => {
@@ -159,6 +161,7 @@ export function AppProvider({ children }: { children: ReactNode }) {
         setKuaidi100Key(kuaidi100Cfg?.key || "");
         setKuaidi100Customer(kuaidi100Cfg?.customer || "");
         setTorrentEnabled(configRes.data.torrent_enabled || false);
+        setTelegramTdataPath(configRes.data.telegram_tdata_path || "");
       }
       if (i18nRes.code === 200) {
         // Merge with defaults to ensure new keys are available
@@ -246,6 +249,9 @@ export function AppProvider({ children }: { children: ReactNode }) {
           values.kuaidi100Customer
         );
       }
+      if (values.telegramTdataPath) {
+        await setConfigValue("telegram.tdata_path", values.telegramTdataPath);
+      }
       refresh();
     },
     [refresh]
@@ -292,6 +298,7 @@ export function AppProvider({ children }: { children: ReactNode }) {
         kuaidi100Customer,
         configExists,
         torrentEnabled,
+        telegramTdataPath,
         t,
         serverT,
         darkMode,

ui/src/pages/ConfigPage.tsx

@@ -16,6 +16,7 @@ export function ConfigPage() {
     apiKey,
     kuaidi100Key,
     kuaidi100Customer,
+    telegramTdataPath,
     webdavServers,
     saveConfig,
     addWebDAV,
@@ -52,6 +53,7 @@ export function ConfigPage() {
         initialApiKey={apiKey}
         initialKuaidi100Key={kuaidi100Key}
         initialKuaidi100Customer={kuaidi100Customer}
+        initialTelegramTdataPath={telegramTdataPath}
         serverPort={serverPort}
         webdavServers={webdavServers}
         onSave={handleSaveConfig}