main.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772
  1. // Package main is the entry point for the 3x-ui web panel application.
  2. // It initializes the database, web server, and handles command-line operations for managing the panel.
  3. package main
  4. import (
  5. "context"
  6. "flag"
  7. "fmt"
  8. "log"
  9. "net/http"
  10. _ "net/http/pprof"
  11. "os"
  12. "os/signal"
  13. "strings"
  14. "syscall"
  15. _ "unsafe"
  16. "github.com/mhsanaei/3x-ui/v3/internal/config"
  17. "github.com/mhsanaei/3x-ui/v3/internal/crypto/nodetoken"
  18. "github.com/mhsanaei/3x-ui/v3/internal/database"
  19. "github.com/mhsanaei/3x-ui/v3/internal/logger"
  20. "github.com/mhsanaei/3x-ui/v3/internal/sub"
  21. "github.com/mhsanaei/3x-ui/v3/internal/tunnelmonitor"
  22. "github.com/mhsanaei/3x-ui/v3/internal/util/crypto"
  23. "github.com/mhsanaei/3x-ui/v3/internal/util/sys"
  24. "github.com/mhsanaei/3x-ui/v3/internal/web"
  25. "github.com/mhsanaei/3x-ui/v3/internal/web/global"
  26. "github.com/mhsanaei/3x-ui/v3/internal/web/service"
  27. "github.com/mhsanaei/3x-ui/v3/internal/web/service/panel"
  28. "github.com/mhsanaei/3x-ui/v3/internal/web/service/tgbot"
  29. "github.com/joho/godotenv"
  30. "github.com/op/go-logging"
  31. )
  32. // cliFallbackTokenName is the single token the CLI regenerates, so `-getApiToken`
  33. // cannot accumulate admin-equivalent credentials that are never revoked.
  34. const cliFallbackTokenName = "cli-fallback"
  35. // installTokenName is minted once on a panel with no tokens and is deliberately
  36. // not the rotated slot, so the credential the installer recorded keeps working.
  37. const installTokenName = "install"
  38. // initNodeTokenCrypto loads the process codec, preferring the key file over
  39. // the environment and failing closed when an enabled policy lacks a key.
  40. func initNodeTokenCrypto() error {
  41. mode, err := nodetoken.ParseMode(config.GetNodeTokenEncryptionMode())
  42. if err != nil {
  43. return err
  44. }
  45. if mode == nodetoken.ModeOff {
  46. c, _ := nodetoken.NewCodec(nodetoken.ModeOff, nil)
  47. nodetoken.Init(c)
  48. return nil
  49. }
  50. ring, ferr := (nodetoken.FileKeySource{Path: config.GetNodeTokenKeyFile()}).Load()
  51. if ferr != nil {
  52. var eerr error
  53. if ring, eerr = (nodetoken.EnvKeySource{Var: config.GetNodeTokenKeyEnv()}).Load(); eerr != nil {
  54. return fmt.Errorf("load node-token key: file: %w; env: %w", ferr, eerr)
  55. }
  56. }
  57. c, err := nodetoken.NewCodec(mode, ring)
  58. if err != nil {
  59. return err
  60. }
  61. nodetoken.Init(c)
  62. // The CLI runs before package logger initialization, so use log.Printf.
  63. log.Printf("node-token encryption enabled (mode=%s, active-key=%s)", config.GetNodeTokenEncryptionMode(), c.ActiveKeyID())
  64. return nil
  65. }
  66. // runWebServer initializes and starts the web server for the 3x-ui panel.
  67. func runWebServer() {
  68. log.Printf("Starting %v %v", config.GetName(), config.GetPanelVersion())
  69. switch config.GetLogLevel() {
  70. case config.Debug:
  71. logger.InitLogger(logging.DEBUG)
  72. case config.Info:
  73. logger.InitLogger(logging.INFO)
  74. case config.Notice:
  75. logger.InitLogger(logging.NOTICE)
  76. case config.Warning:
  77. logger.InitLogger(logging.WARNING)
  78. case config.Error:
  79. logger.InitLogger(logging.ERROR)
  80. default:
  81. log.Fatalf("Unknown log level: %v", config.GetLogLevel())
  82. }
  83. _ = godotenv.Load()
  84. for _, line := range sys.ApplyMemoryTuning() {
  85. logger.Info(line)
  86. }
  87. if os.Getenv("XUI_PPROF") == "true" {
  88. go func() {
  89. logger.Info("pprof profiling server listening on 127.0.0.1:6060")
  90. if err := http.ListenAndServe("127.0.0.1:6060", nil); err != nil {
  91. logger.Warning("pprof server stopped: ", err)
  92. }
  93. }()
  94. }
  95. if err := initNodeTokenCrypto(); err != nil {
  96. log.Fatalf("Error initializing node-token encryption: %v", err)
  97. }
  98. err := database.InitDB(config.GetDBPath())
  99. if err != nil {
  100. log.Fatalf("Error initializing database: %v", err)
  101. }
  102. server := web.NewServer()
  103. global.SetWebServer(server)
  104. err = server.Start()
  105. if err != nil {
  106. log.Fatalf("Error starting web server: %v", err)
  107. }
  108. sub.SetDistFS(web.EmbeddedDist())
  109. service.RegisterSubLinkProvider(sub.NewLinkProvider())
  110. subServer := sub.NewServer()
  111. global.SetSubServer(subServer)
  112. err = subServer.Start()
  113. if err != nil {
  114. log.Fatalf("Error starting sub server: %v", err)
  115. }
  116. sigCh := make(chan os.Signal, 8)
  117. // Trap shutdown signals
  118. signal.Notify(sigCh, syscall.SIGHUP, syscall.SIGTERM, sys.SIGUSR1, os.Interrupt)
  119. global.SetRestartHook(func() {
  120. select {
  121. case sigCh <- syscall.SIGHUP:
  122. default:
  123. }
  124. })
  125. var stopTunnelHealthMonitor context.CancelFunc
  126. monitorCfg := tunnelmonitor.ConfigFromEnv()
  127. if monitorCfg.Enabled {
  128. if monitorCfg.ProxyURL == "" {
  129. logger.Warning("Tunnel health monitor enabled without XUI_TUNNEL_HEALTH_PROXY: the probe measures host connectivity, not the xray tunnel, so failures will restart xray without fixing host network issues")
  130. }
  131. monitorCtx, cancel := context.WithCancel(context.Background())
  132. stopTunnelHealthMonitor = cancel
  133. monitor, err := tunnelmonitor.New(monitorCfg, func(_ context.Context) error {
  134. logger.Warning("Tunnel health monitor threshold reached, restarting xray-core")
  135. return server.RestartXray()
  136. })
  137. if err != nil {
  138. logger.Warning("Tunnel health monitor disabled: ", err)
  139. } else {
  140. go monitor.Run(monitorCtx)
  141. }
  142. }
  143. for {
  144. sig := <-sigCh
  145. switch sig {
  146. case syscall.SIGHUP:
  147. logger.Info("Received SIGHUP signal. Restarting servers...")
  148. err := server.StopPanelOnly()
  149. if err != nil {
  150. logger.Debug("Error stopping web server:", err)
  151. }
  152. err = subServer.Stop()
  153. if err != nil {
  154. logger.Debug("Error stopping sub server:", err)
  155. }
  156. server = web.NewServer()
  157. global.SetWebServer(server)
  158. err = server.StartPanelOnly()
  159. if err != nil {
  160. log.Fatalf("Error restarting web server: %v", err)
  161. }
  162. log.Println("Web server restarted successfully.")
  163. sub.SetDistFS(web.EmbeddedDist())
  164. subServer = sub.NewServer()
  165. global.SetSubServer(subServer)
  166. err = subServer.Start()
  167. if err != nil {
  168. log.Fatalf("Error restarting sub server: %v", err)
  169. }
  170. log.Println("Sub server restarted successfully.")
  171. case sys.SIGUSR1:
  172. logger.Info("Received USR1 signal, restarting xray-core...")
  173. err := server.RestartXray()
  174. if err != nil {
  175. logger.Error("Failed to restart xray-core:", err)
  176. }
  177. default:
  178. if stopTunnelHealthMonitor != nil {
  179. stopTunnelHealthMonitor()
  180. }
  181. // --- FIX FOR TELEGRAM BOT CONFLICT (409) on full shutdown ---
  182. tgbot.StopBot()
  183. // ------------------------------------------------------------
  184. _ = server.Stop()
  185. _ = subServer.Stop()
  186. log.Println("Shutting down servers.")
  187. return
  188. }
  189. }
  190. }
  191. // resetSetting resets all panel settings to their default values.
  192. func resetSetting() error {
  193. err := database.InitDB(config.GetDBPath())
  194. if err != nil {
  195. fmt.Println("Failed to initialize database:", err)
  196. return err
  197. }
  198. settingService := service.SettingService{}
  199. err = settingService.ResetSettings()
  200. if err != nil {
  201. fmt.Println("Failed to reset settings:", err)
  202. return err
  203. } else {
  204. fmt.Println("Settings successfully reset.")
  205. }
  206. return nil
  207. }
  208. // showSetting displays the current panel settings if show is true.
  209. func showSetting(show bool) {
  210. if show {
  211. settingService := service.SettingService{}
  212. port, err := settingService.GetPort()
  213. if err != nil {
  214. fmt.Println("get current port failed, error info:", err)
  215. }
  216. webBasePath, err := settingService.GetBasePath()
  217. if err != nil {
  218. fmt.Println("get webBasePath failed, error info:", err)
  219. }
  220. certFile, err := settingService.GetCertFile()
  221. if err != nil {
  222. fmt.Println("get cert file failed, error info:", err)
  223. }
  224. keyFile, err := settingService.GetKeyFile()
  225. if err != nil {
  226. fmt.Println("get key file failed, error info:", err)
  227. }
  228. userService := panel.UserService{}
  229. userModel, err := userService.GetFirstUser()
  230. if err != nil {
  231. fmt.Println("get current user info failed, error info:", err)
  232. }
  233. if userModel.Username == "" || userModel.Password == "" {
  234. fmt.Println("current username or password is empty")
  235. }
  236. fmt.Println("current panel settings as follows:")
  237. if certFile == "" || keyFile == "" {
  238. fmt.Println("Warning: Panel is not secure with SSL")
  239. } else {
  240. fmt.Println("Panel is secure with SSL")
  241. }
  242. hasDefaultCredential := func() bool {
  243. return userModel.Username == "admin" && crypto.CheckPasswordHash(userModel.Password, "admin")
  244. }()
  245. fmt.Println("hasDefaultCredential:", hasDefaultCredential)
  246. fmt.Println("port:", port)
  247. fmt.Println("webBasePath:", webBasePath)
  248. }
  249. }
  250. // updateTgbotEnableSts enables or disables the Telegram bot notifications based on the status parameter.
  251. func updateTgbotEnableSts(status bool) {
  252. settingService := service.SettingService{}
  253. currentTgSts, err := settingService.GetTgbotEnabled()
  254. if err != nil {
  255. fmt.Println(err)
  256. return
  257. }
  258. logger.Infof("current enabletgbot status[%v],need update to status[%v]", currentTgSts, status)
  259. if currentTgSts != status {
  260. err := settingService.SetTgbotEnabled(status)
  261. if err != nil {
  262. fmt.Println(err)
  263. return
  264. } else {
  265. logger.Infof("SetTgbotEnabled[%v] success", status)
  266. }
  267. }
  268. }
  269. // updateTgbotSetting updates Telegram bot settings including token, chat ID, and runtime schedule.
  270. func updateTgbotSetting(tgBotToken string, tgBotChatid string, tgBotRuntime string) {
  271. err := database.InitDB(config.GetDBPath())
  272. if err != nil {
  273. fmt.Println("Error initializing database:", err)
  274. return
  275. }
  276. settingService := service.SettingService{}
  277. if tgBotToken != "" {
  278. err := settingService.SetTgBotToken(tgBotToken)
  279. if err != nil {
  280. fmt.Printf("Error setting Telegram bot token: %v\n", err)
  281. return
  282. }
  283. logger.Info("Successfully updated Telegram bot token.")
  284. }
  285. if tgBotRuntime != "" {
  286. err := settingService.SetTgbotRuntime(tgBotRuntime)
  287. if err != nil {
  288. fmt.Printf("Error setting Telegram bot runtime: %v\n", err)
  289. return
  290. }
  291. logger.Infof("Successfully updated Telegram bot runtime to [%s].", tgBotRuntime)
  292. }
  293. if tgBotChatid != "" {
  294. err := settingService.SetTgBotChatId(tgBotChatid)
  295. if err != nil {
  296. fmt.Printf("Error setting Telegram bot chat ID: %v\n", err)
  297. return
  298. }
  299. logger.Info("Successfully updated Telegram bot chat ID.")
  300. }
  301. }
  302. // encryptNodeTokens re-encrypts stored tokens after enablement or rotation.
  303. // It requires migration|required mode and a configured key.
  304. func encryptNodeTokens() {
  305. _ = godotenv.Load()
  306. if err := initNodeTokenCrypto(); err != nil {
  307. fmt.Println("node-token encryption init failed:", err)
  308. os.Exit(1)
  309. }
  310. if err := database.InitDB(config.GetDBPath()); err != nil {
  311. fmt.Println("database initialization failed:", err)
  312. os.Exit(1)
  313. }
  314. changed, skipped, err := (&service.NodeService{}).MigrateNodeTokensToActiveKey()
  315. if err != nil {
  316. fmt.Println("token migration failed:", err)
  317. os.Exit(1)
  318. }
  319. fmt.Printf("node-token migration complete: %d re-encrypted, %d already current/skipped\n", changed, skipped)
  320. }
  321. // updateSetting updates various panel settings including port, credentials, base path, listen IP, and two-factor authentication.
  322. func updateSetting(port int, username string, password string, webBasePath string, listenIP string, resetTwoFactor bool) error {
  323. err := database.InitDB(config.GetDBPath())
  324. if err != nil {
  325. fmt.Println("Database initialization failed:", err)
  326. return err
  327. }
  328. settingService := service.SettingService{}
  329. userService := panel.UserService{}
  330. if port > 0 {
  331. err := settingService.SetPort(port)
  332. if err != nil {
  333. fmt.Println("Failed to set port:", err)
  334. } else {
  335. fmt.Printf("Port set successfully: %v\n", port)
  336. }
  337. }
  338. if username != "" || password != "" {
  339. err := userService.UpdateFirstUser(username, password)
  340. if err != nil {
  341. fmt.Println("Failed to update username and password:", err)
  342. } else {
  343. fmt.Println("Username and password updated successfully")
  344. }
  345. }
  346. if webBasePath != "" {
  347. err := settingService.SetBasePath(webBasePath)
  348. if err != nil {
  349. fmt.Println("Failed to set base URI path:", err)
  350. } else {
  351. fmt.Println("Base URI path set successfully")
  352. }
  353. }
  354. if resetTwoFactor {
  355. err := settingService.SetTwoFactorEnable(false)
  356. if err != nil {
  357. fmt.Println("Failed to reset two-factor authentication:", err)
  358. } else {
  359. _ = settingService.SetTwoFactorToken("")
  360. fmt.Println("Two-factor authentication reset successfully")
  361. }
  362. }
  363. if listenIP != "" {
  364. err := settingService.SetListen(listenIP)
  365. if err != nil {
  366. fmt.Println("Failed to set listen IP:", err)
  367. } else {
  368. fmt.Printf("listen %v set successfully\n", listenIP)
  369. }
  370. }
  371. return nil
  372. }
  373. // updateCert updates the SSL certificate files for the panel.
  374. func updateCert(publicKey string, privateKey string) {
  375. err := database.InitDB(config.GetDBPath())
  376. if err != nil {
  377. fmt.Println(err)
  378. return
  379. }
  380. if (privateKey != "" && publicKey != "") || (privateKey == "" && publicKey == "") {
  381. settingService := service.SettingService{}
  382. err = settingService.SetCertFile(publicKey)
  383. if err != nil {
  384. fmt.Println("set certificate public key failed:", err)
  385. } else {
  386. fmt.Println("set certificate public key success")
  387. }
  388. err = settingService.SetKeyFile(privateKey)
  389. if err != nil {
  390. fmt.Println("set certificate private key failed:", err)
  391. } else {
  392. fmt.Println("set certificate private key success")
  393. }
  394. err = settingService.SetSubCertFile(publicKey)
  395. if err != nil {
  396. fmt.Println("set certificate for subscription public key failed:", err)
  397. } else {
  398. fmt.Println("set certificate for subscription public key success")
  399. }
  400. err = settingService.SetSubKeyFile(privateKey)
  401. if err != nil {
  402. fmt.Println("set certificate for subscription private key failed:", err)
  403. } else {
  404. fmt.Println("set certificate for subscription private key success")
  405. }
  406. } else {
  407. fmt.Println("both public and private key should be entered.")
  408. }
  409. }
  410. // GetCertificate displays the current SSL certificate settings if getCert is true.
  411. func GetCertificate(getCert bool) {
  412. if getCert {
  413. settingService := service.SettingService{}
  414. certFile, err := settingService.GetCertFile()
  415. if err != nil {
  416. fmt.Println("get cert file failed, error info:", err)
  417. }
  418. keyFile, err := settingService.GetKeyFile()
  419. if err != nil {
  420. fmt.Println("get key file failed, error info:", err)
  421. }
  422. fmt.Println("cert:", certFile)
  423. fmt.Println("key:", keyFile)
  424. }
  425. }
  426. // GetListenIP displays the current panel listen IP address if getListen is true.
  427. func GetListenIP(getListen bool) {
  428. if getListen {
  429. settingService := service.SettingService{}
  430. ListenIP, err := settingService.GetListen()
  431. if err != nil {
  432. log.Printf("Failed to retrieve listen IP: %v", err)
  433. return
  434. }
  435. fmt.Println("listenIP:", ListenIP)
  436. }
  437. }
  438. func GetApiToken(getApiToken bool, tokenName string) {
  439. if !getApiToken {
  440. return
  441. }
  442. // An explicit name applies to both branches below; without one each keeps
  443. // the name it already used, so every existing invocation is unaffected.
  444. name := strings.TrimSpace(tokenName)
  445. err := database.InitDB(config.GetDBPath())
  446. if err != nil {
  447. fmt.Println("open database failed, error info:", err)
  448. return
  449. }
  450. apiTokenService := panel.ApiTokenService{}
  451. tokens, err := apiTokenService.List()
  452. if err != nil {
  453. fmt.Println("get apiToken failed, error info:", err)
  454. return
  455. }
  456. if len(tokens) > 0 {
  457. fmt.Printf("There are %d API token(s) configured. Existing tokens cannot be retrieved in plaintext because only hashes are stored.\n", len(tokens))
  458. fmt.Println("If you have lost your token, you can manage and generate new tokens through the Panel UI (Settings -> API Tokens).")
  459. // Rotate one token per name so repeated calls cannot pile up
  460. // indefinitely many admin-equivalent tokens that never expire.
  461. rotated := name
  462. if rotated == "" {
  463. rotated = cliFallbackTokenName
  464. }
  465. created, err := apiTokenService.RecreateByName(rotated)
  466. if err != nil {
  467. fmt.Println("Failed to create a fallback API token:", err)
  468. return
  469. }
  470. fmt.Printf("\nThe API token %q has been regenerated (any previous one is now invalid):\n", rotated)
  471. fmt.Println("apiToken:", created.Token)
  472. return
  473. }
  474. if name == "" {
  475. name = installTokenName
  476. }
  477. created, err := apiTokenService.Create(name, "", 0)
  478. if err != nil {
  479. fmt.Println("create apiToken failed, error info:", err)
  480. return
  481. }
  482. fmt.Println("apiToken:", created.Token)
  483. }
  484. // migrateDb performs database migration operations for the 3x-ui panel.
  485. func migrateDb() {
  486. inboundService := service.InboundService{}
  487. logger.InitLogger(logging.INFO)
  488. err := database.InitDB(config.GetDBPath())
  489. if err != nil {
  490. log.Fatal(err)
  491. }
  492. fmt.Println("Start migrating database...")
  493. inboundService.MigrateDB()
  494. fmt.Println("Migration done!")
  495. }
  496. // loadServiceEnvFile loads the systemd EnvironmentFile so CLI subcommands like
  497. // "x-ui setting" hit the same database backend as the panel. godotenv.Load does
  498. // not override variables already in the environment, so it is a no-op for the
  499. // systemd-managed service.
  500. func loadServiceEnvFile() {
  501. for _, path := range config.GetEnvFilePaths() {
  502. if _, err := os.Stat(path); err != nil {
  503. continue
  504. }
  505. if err := godotenv.Load(path); err != nil {
  506. log.Printf("warning: failed to load env file %s: %v", path, err)
  507. }
  508. return
  509. }
  510. }
  511. // main is the entry point of the 3x-ui application.
  512. // It parses command-line arguments to run the web server, migrate database, or update settings.
  513. func main() {
  514. loadServiceEnvFile()
  515. if len(os.Args) < 2 {
  516. runWebServer()
  517. return
  518. }
  519. var showVersion bool
  520. flag.BoolVar(&showVersion, "v", false, "show version")
  521. runCmd := flag.NewFlagSet("run", flag.ExitOnError)
  522. migrateDbCmd := flag.NewFlagSet("migrate-db", flag.ExitOnError)
  523. var migrateDsn string
  524. var migrateSrc string
  525. var migrateDump string
  526. var migrateRestore string
  527. var migrateOut string
  528. migrateDbCmd.StringVar(&migrateDsn, "dsn", "", "Destination PostgreSQL DSN (postgres://user:pass@host:port/db?sslmode=disable)")
  529. migrateDbCmd.StringVar(&migrateSrc, "src", "", "Source SQLite file (defaults to the configured x-ui.db)")
  530. migrateDbCmd.StringVar(&migrateDump, "dump", "", "Write a portable SQL text dump of --src to this file (.db -> .dump)")
  531. migrateDbCmd.StringVar(&migrateRestore, "restore", "", "Rebuild a SQLite database from this SQL text dump (.dump -> .db); requires --out")
  532. migrateDbCmd.StringVar(&migrateOut, "out", "", "Destination SQLite file for --restore (must not already exist)")
  533. settingCmd := flag.NewFlagSet("setting", flag.ExitOnError)
  534. var port int
  535. var username string
  536. var password string
  537. var webBasePath string
  538. var listenIP string
  539. var getListen bool
  540. var webCertFile string
  541. var webKeyFile string
  542. var tgbottoken string
  543. var tgbotchatid string
  544. var enabletgbot bool
  545. var tgbotRuntime string
  546. var reset bool
  547. var show bool
  548. var getCert bool
  549. var getApiToken bool
  550. var tokenName string
  551. var resetTwoFactor bool
  552. settingCmd.BoolVar(&reset, "reset", false, "Reset all settings")
  553. settingCmd.BoolVar(&show, "show", false, "Display current settings")
  554. settingCmd.IntVar(&port, "port", 0, "Set panel port number")
  555. settingCmd.StringVar(&username, "username", "", "Set login username")
  556. settingCmd.StringVar(&password, "password", "", "Set login password")
  557. settingCmd.StringVar(&webBasePath, "webBasePath", "", "Set base path for Panel")
  558. settingCmd.StringVar(&listenIP, "listenIP", "", "set panel listenIP IP")
  559. settingCmd.BoolVar(&resetTwoFactor, "resetTwoFactor", false, "Reset two-factor authentication settings")
  560. settingCmd.BoolVar(&getListen, "getListen", false, "Display current panel listenIP IP")
  561. settingCmd.BoolVar(&getCert, "getCert", false, "Display current certificate settings")
  562. settingCmd.BoolVar(&getApiToken, "getApiToken", false, "Print an API token for CLI use, regenerating it and invalidating the previous one; on a panel with no tokens yet it mints one instead")
  563. settingCmd.StringVar(&tokenName, "tokenName", "", "Name of the token -getApiToken acts on (default: "+cliFallbackTokenName+", or "+installTokenName+" on a panel with no tokens)")
  564. settingCmd.StringVar(&webCertFile, "webCert", "", "Set path to public key file for panel")
  565. settingCmd.StringVar(&webKeyFile, "webCertKey", "", "Set path to private key file for panel")
  566. settingCmd.StringVar(&tgbottoken, "tgbottoken", "", "Set token for Telegram bot")
  567. settingCmd.StringVar(&tgbotRuntime, "tgbotRuntime", "", "Set cron time for Telegram bot notifications")
  568. settingCmd.StringVar(&tgbotchatid, "tgbotchatid", "", "Set chat ID for Telegram bot notifications")
  569. settingCmd.BoolVar(&enabletgbot, "enabletgbot", false, "Enable notifications via Telegram bot")
  570. oldUsage := flag.Usage
  571. flag.Usage = func() {
  572. oldUsage()
  573. fmt.Print(commandHelp())
  574. }
  575. flag.Parse()
  576. if showVersion {
  577. fmt.Println(config.GetPanelVersion())
  578. return
  579. }
  580. switch os.Args[1] {
  581. case "run":
  582. err := runCmd.Parse(os.Args[2:])
  583. if err != nil {
  584. fmt.Println(err)
  585. return
  586. }
  587. runWebServer()
  588. case "migrate":
  589. migrateDb()
  590. case "encrypt-tokens":
  591. encryptNodeTokens()
  592. case "migrate-db":
  593. if err := migrateDbCmd.Parse(os.Args[2:]); err != nil {
  594. fmt.Println(err)
  595. return
  596. }
  597. src := migrateSrc
  598. if src == "" {
  599. src = config.GetDBPath()
  600. }
  601. switch {
  602. case migrateDump != "":
  603. if err := database.DumpSQLite(src, migrateDump); err != nil {
  604. fmt.Println("dump failed:", err)
  605. os.Exit(1)
  606. }
  607. fmt.Printf("Dumped %s -> %s\n", src, migrateDump)
  608. case migrateRestore != "":
  609. if migrateOut == "" {
  610. fmt.Println("--out is required when using --restore: the destination .db path (must not exist)")
  611. return
  612. }
  613. if err := database.RestoreSQLite(migrateRestore, migrateOut); err != nil {
  614. fmt.Println("restore failed:", err)
  615. os.Exit(1)
  616. }
  617. fmt.Printf("Restored %s -> %s\n", migrateRestore, migrateOut)
  618. case migrateDsn != "":
  619. if err := database.MigrateData(src, migrateDsn); err != nil {
  620. fmt.Println("migration failed:", err)
  621. os.Exit(1)
  622. }
  623. default:
  624. fmt.Println("nothing to do: pass --dump <file>, --restore <file> --out <db>, or --dsn <postgres-dsn>")
  625. }
  626. case "setting":
  627. err := settingCmd.Parse(os.Args[2:])
  628. if err != nil {
  629. fmt.Println(err)
  630. return
  631. }
  632. // flag stops parsing at the first non-flag argument, so the `-getApiToken true`
  633. // form drops every flag written after it. Say so instead of acting on a default.
  634. if rest := settingCmd.Args(); len(rest) > 0 {
  635. fmt.Printf("warning: ignored %q and any flags after it; put flags before positional arguments\n", strings.Join(rest, " "))
  636. }
  637. if reset {
  638. if err = resetSetting(); err != nil {
  639. return
  640. }
  641. } else {
  642. if err = updateSetting(port, username, password, webBasePath, listenIP, resetTwoFactor); err != nil {
  643. return
  644. }
  645. }
  646. if webCertFile != "" || webKeyFile != "" {
  647. updateCert(webCertFile, webKeyFile)
  648. }
  649. if show {
  650. showSetting(show)
  651. }
  652. if getListen {
  653. GetListenIP(getListen)
  654. }
  655. if getCert {
  656. GetCertificate(getCert)
  657. }
  658. if getApiToken {
  659. GetApiToken(getApiToken, tokenName)
  660. }
  661. if (tgbottoken != "") || (tgbotchatid != "") || (tgbotRuntime != "") {
  662. updateTgbotSetting(tgbottoken, tgbotchatid, tgbotRuntime)
  663. }
  664. if enabletgbot {
  665. updateTgbotEnableSts(enabletgbot)
  666. }
  667. case "cert":
  668. err := settingCmd.Parse(os.Args[2:])
  669. if err != nil {
  670. fmt.Println(err)
  671. return
  672. }
  673. if reset {
  674. updateCert("", "")
  675. } else {
  676. updateCert(webCertFile, webKeyFile)
  677. }
  678. default:
  679. fmt.Println("Invalid subcommands")
  680. fmt.Println()
  681. runCmd.Usage()
  682. fmt.Println()
  683. settingCmd.Usage()
  684. }
  685. }
  686. func commandHelp() string {
  687. return `
  688. Commands:
  689. run run web panel
  690. migrate migrate from other/old x-ui
  691. migrate-db SQLite <-> .dump (--dump/--restore) or copy into PostgreSQL (--dsn)
  692. encrypt-tokens encrypt node bearer tokens with the configured active key
  693. setting set settings
  694. `
  695. }