setting.go 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608
  1. package service
  2. import (
  3. _ "embed"
  4. "encoding/json"
  5. "errors"
  6. "fmt"
  7. "reflect"
  8. "strconv"
  9. "strings"
  10. "time"
  11. "x-ui/database"
  12. "x-ui/database/model"
  13. "x-ui/logger"
  14. "x-ui/util/common"
  15. "x-ui/util/random"
  16. "x-ui/util/reflect_util"
  17. "x-ui/web/entity"
  18. "x-ui/xray"
  19. )
  20. //go:embed config.json
  21. var xrayTemplateConfig string
  22. var defaultValueMap = map[string]string{
  23. "xrayTemplateConfig": xrayTemplateConfig,
  24. "webListen": "",
  25. "webDomain": "",
  26. "webPort": "2053",
  27. "webCertFile": "",
  28. "webKeyFile": "",
  29. "secret": random.Seq(32),
  30. "webBasePath": "/",
  31. "sessionMaxAge": "60",
  32. "pageSize": "50",
  33. "expireDiff": "0",
  34. "trafficDiff": "0",
  35. "remarkModel": "-ieo",
  36. "timeLocation": "Local",
  37. "tgBotEnable": "false",
  38. "tgBotToken": "",
  39. "tgBotProxy": "",
  40. "tgBotAPIServer": "",
  41. "tgBotChatId": "",
  42. "tgRunTime": "@daily",
  43. "tgBotBackup": "false",
  44. "tgBotLoginNotify": "true",
  45. "tgCpu": "80",
  46. "tgLang": "en-US",
  47. "secretEnable": "false",
  48. "subEnable": "false",
  49. "subSyncEnable": "true",
  50. "subListen": "",
  51. "subPort": "2096",
  52. "subPath": "/sub/",
  53. "subDomain": "",
  54. "subCertFile": "",
  55. "subKeyFile": "",
  56. "subUpdates": "12",
  57. "subEncrypt": "true",
  58. "subShowInfo": "true",
  59. "subURI": "",
  60. "subJsonPath": "/json/",
  61. "subJsonURI": "",
  62. "subJsonFragment": "",
  63. "subJsonNoises": "",
  64. "subJsonMux": "",
  65. "subJsonRules": "",
  66. "datepicker": "gregorian",
  67. "warp": "",
  68. }
  69. type SettingService struct{}
  70. func (s *SettingService) GetDefaultJsonConfig() (interface{}, error) {
  71. var jsonData interface{}
  72. err := json.Unmarshal([]byte(xrayTemplateConfig), &jsonData)
  73. if err != nil {
  74. return nil, err
  75. }
  76. return jsonData, nil
  77. }
  78. func (s *SettingService) GetAllSetting() (*entity.AllSetting, error) {
  79. db := database.GetDB()
  80. settings := make([]*model.Setting, 0)
  81. err := db.Model(model.Setting{}).Not("key = ?", "xrayTemplateConfig").Find(&settings).Error
  82. if err != nil {
  83. return nil, err
  84. }
  85. allSetting := &entity.AllSetting{}
  86. t := reflect.TypeOf(allSetting).Elem()
  87. v := reflect.ValueOf(allSetting).Elem()
  88. fields := reflect_util.GetFields(t)
  89. setSetting := func(key, value string) (err error) {
  90. defer func() {
  91. panicErr := recover()
  92. if panicErr != nil {
  93. err = errors.New(fmt.Sprint(panicErr))
  94. }
  95. }()
  96. var found bool
  97. var field reflect.StructField
  98. for _, f := range fields {
  99. if f.Tag.Get("json") == key {
  100. field = f
  101. found = true
  102. break
  103. }
  104. }
  105. if !found {
  106. // Some settings are automatically generated, no need to return to the front end to modify the user
  107. return nil
  108. }
  109. fieldV := v.FieldByName(field.Name)
  110. switch t := fieldV.Interface().(type) {
  111. case int:
  112. n, err := strconv.ParseInt(value, 10, 64)
  113. if err != nil {
  114. return err
  115. }
  116. fieldV.SetInt(n)
  117. case string:
  118. fieldV.SetString(value)
  119. case bool:
  120. fieldV.SetBool(value == "true")
  121. default:
  122. return common.NewErrorf("unknown field %v type %v", key, t)
  123. }
  124. return
  125. }
  126. keyMap := map[string]bool{}
  127. for _, setting := range settings {
  128. err := setSetting(setting.Key, setting.Value)
  129. if err != nil {
  130. return nil, err
  131. }
  132. keyMap[setting.Key] = true
  133. }
  134. for key, value := range defaultValueMap {
  135. if keyMap[key] {
  136. continue
  137. }
  138. err := setSetting(key, value)
  139. if err != nil {
  140. return nil, err
  141. }
  142. }
  143. return allSetting, nil
  144. }
  145. func (s *SettingService) ResetSettings() error {
  146. db := database.GetDB()
  147. err := db.Where("1 = 1").Delete(model.Setting{}).Error
  148. if err != nil {
  149. return err
  150. }
  151. return db.Model(model.User{}).
  152. Where("1 = 1").
  153. Update("login_secret", "").Error
  154. }
  155. func (s *SettingService) getSetting(key string) (*model.Setting, error) {
  156. db := database.GetDB()
  157. setting := &model.Setting{}
  158. err := db.Model(model.Setting{}).Where("key = ?", key).First(setting).Error
  159. if err != nil {
  160. return nil, err
  161. }
  162. return setting, nil
  163. }
  164. func (s *SettingService) saveSetting(key string, value string) error {
  165. setting, err := s.getSetting(key)
  166. db := database.GetDB()
  167. if database.IsNotFound(err) {
  168. return db.Create(&model.Setting{
  169. Key: key,
  170. Value: value,
  171. }).Error
  172. } else if err != nil {
  173. return err
  174. }
  175. setting.Key = key
  176. setting.Value = value
  177. return db.Save(setting).Error
  178. }
  179. func (s *SettingService) getString(key string) (string, error) {
  180. setting, err := s.getSetting(key)
  181. if database.IsNotFound(err) {
  182. value, ok := defaultValueMap[key]
  183. if !ok {
  184. return "", common.NewErrorf("key <%v> not in defaultValueMap", key)
  185. }
  186. return value, nil
  187. } else if err != nil {
  188. return "", err
  189. }
  190. return setting.Value, nil
  191. }
  192. func (s *SettingService) setString(key string, value string) error {
  193. return s.saveSetting(key, value)
  194. }
  195. func (s *SettingService) getBool(key string) (bool, error) {
  196. str, err := s.getString(key)
  197. if err != nil {
  198. return false, err
  199. }
  200. return strconv.ParseBool(str)
  201. }
  202. func (s *SettingService) setBool(key string, value bool) error {
  203. return s.setString(key, strconv.FormatBool(value))
  204. }
  205. func (s *SettingService) getInt(key string) (int, error) {
  206. str, err := s.getString(key)
  207. if err != nil {
  208. return 0, err
  209. }
  210. return strconv.Atoi(str)
  211. }
  212. func (s *SettingService) setInt(key string, value int) error {
  213. return s.setString(key, strconv.Itoa(value))
  214. }
  215. func (s *SettingService) GetXrayConfigTemplate() (string, error) {
  216. return s.getString("xrayTemplateConfig")
  217. }
  218. func (s *SettingService) GetListen() (string, error) {
  219. return s.getString("webListen")
  220. }
  221. func (s *SettingService) SetListen(ip string) error {
  222. return s.setString("webListen", ip)
  223. }
  224. func (s *SettingService) GetWebDomain() (string, error) {
  225. return s.getString("webDomain")
  226. }
  227. func (s *SettingService) GetTgBotToken() (string, error) {
  228. return s.getString("tgBotToken")
  229. }
  230. func (s *SettingService) SetTgBotToken(token string) error {
  231. return s.setString("tgBotToken", token)
  232. }
  233. func (s *SettingService) GetTgBotProxy() (string, error) {
  234. return s.getString("tgBotProxy")
  235. }
  236. func (s *SettingService) SetTgBotProxy(token string) error {
  237. return s.setString("tgBotProxy", token)
  238. }
  239. func (s *SettingService) GetTgBotAPIServer() (string, error) {
  240. return s.getString("tgBotAPIServer")
  241. }
  242. func (s *SettingService) SetTgBotAPIServer(token string) error {
  243. return s.setString("tgBotAPIServer", token)
  244. }
  245. func (s *SettingService) GetTgBotChatId() (string, error) {
  246. return s.getString("tgBotChatId")
  247. }
  248. func (s *SettingService) SetTgBotChatId(chatIds string) error {
  249. return s.setString("tgBotChatId", chatIds)
  250. }
  251. func (s *SettingService) GetTgbotEnabled() (bool, error) {
  252. return s.getBool("tgBotEnable")
  253. }
  254. func (s *SettingService) SetTgbotEnabled(value bool) error {
  255. return s.setBool("tgBotEnable", value)
  256. }
  257. func (s *SettingService) GetTgbotRuntime() (string, error) {
  258. return s.getString("tgRunTime")
  259. }
  260. func (s *SettingService) SetTgbotRuntime(time string) error {
  261. return s.setString("tgRunTime", time)
  262. }
  263. func (s *SettingService) GetTgBotBackup() (bool, error) {
  264. return s.getBool("tgBotBackup")
  265. }
  266. func (s *SettingService) GetTgBotLoginNotify() (bool, error) {
  267. return s.getBool("tgBotLoginNotify")
  268. }
  269. func (s *SettingService) GetTgCpu() (int, error) {
  270. return s.getInt("tgCpu")
  271. }
  272. func (s *SettingService) GetTgLang() (string, error) {
  273. return s.getString("tgLang")
  274. }
  275. func (s *SettingService) GetPort() (int, error) {
  276. return s.getInt("webPort")
  277. }
  278. func (s *SettingService) SetPort(port int) error {
  279. return s.setInt("webPort", port)
  280. }
  281. func (s *SettingService) SetCertFile(webCertFile string) error {
  282. return s.setString("webCertFile", webCertFile)
  283. }
  284. func (s *SettingService) GetCertFile() (string, error) {
  285. return s.getString("webCertFile")
  286. }
  287. func (s *SettingService) SetKeyFile(webKeyFile string) error {
  288. return s.setString("webKeyFile", webKeyFile)
  289. }
  290. func (s *SettingService) GetKeyFile() (string, error) {
  291. return s.getString("webKeyFile")
  292. }
  293. func (s *SettingService) GetExpireDiff() (int, error) {
  294. return s.getInt("expireDiff")
  295. }
  296. func (s *SettingService) GetTrafficDiff() (int, error) {
  297. return s.getInt("trafficDiff")
  298. }
  299. func (s *SettingService) GetSessionMaxAge() (int, error) {
  300. return s.getInt("sessionMaxAge")
  301. }
  302. func (s *SettingService) GetRemarkModel() (string, error) {
  303. return s.getString("remarkModel")
  304. }
  305. func (s *SettingService) GetSecretStatus() (bool, error) {
  306. return s.getBool("secretEnable")
  307. }
  308. func (s *SettingService) SetSecretStatus(value bool) error {
  309. return s.setBool("secretEnable", value)
  310. }
  311. func (s *SettingService) GetSecret() ([]byte, error) {
  312. secret, err := s.getString("secret")
  313. if secret == defaultValueMap["secret"] {
  314. err := s.saveSetting("secret", secret)
  315. if err != nil {
  316. logger.Warning("save secret failed:", err)
  317. }
  318. }
  319. return []byte(secret), err
  320. }
  321. func (s *SettingService) SetBasePath(basePath string) error {
  322. if !strings.HasPrefix(basePath, "/") {
  323. basePath = "/" + basePath
  324. }
  325. if !strings.HasSuffix(basePath, "/") {
  326. basePath += "/"
  327. }
  328. return s.setString("webBasePath", basePath)
  329. }
  330. func (s *SettingService) GetBasePath() (string, error) {
  331. basePath, err := s.getString("webBasePath")
  332. if err != nil {
  333. return "", err
  334. }
  335. if !strings.HasPrefix(basePath, "/") {
  336. basePath = "/" + basePath
  337. }
  338. if !strings.HasSuffix(basePath, "/") {
  339. basePath += "/"
  340. }
  341. return basePath, nil
  342. }
  343. func (s *SettingService) GetTimeLocation() (*time.Location, error) {
  344. l, err := s.getString("timeLocation")
  345. if err != nil {
  346. return nil, err
  347. }
  348. location, err := time.LoadLocation(l)
  349. if err != nil {
  350. defaultLocation := defaultValueMap["timeLocation"]
  351. logger.Errorf("location <%v> not exist, using default location: %v", l, defaultLocation)
  352. return time.LoadLocation(defaultLocation)
  353. }
  354. return location, nil
  355. }
  356. func (s *SettingService) GetSubEnable() (bool, error) {
  357. return s.getBool("subEnable")
  358. }
  359. func (s *SettingService) GetSubSyncEnable() (bool, error) {
  360. return s.getBool("subSyncEnable")
  361. }
  362. func (s *SettingService) SetSubSyncEnable(value bool) error {
  363. return s.setBool("subSyncEnable", value)
  364. }
  365. func (s *SettingService) GetSubListen() (string, error) {
  366. return s.getString("subListen")
  367. }
  368. func (s *SettingService) GetSubPort() (int, error) {
  369. return s.getInt("subPort")
  370. }
  371. func (s *SettingService) GetSubPath() (string, error) {
  372. return s.getString("subPath")
  373. }
  374. func (s *SettingService) GetSubJsonPath() (string, error) {
  375. return s.getString("subJsonPath")
  376. }
  377. func (s *SettingService) GetSubDomain() (string, error) {
  378. return s.getString("subDomain")
  379. }
  380. func (s *SettingService) GetSubCertFile() (string, error) {
  381. return s.getString("subCertFile")
  382. }
  383. func (s *SettingService) GetSubKeyFile() (string, error) {
  384. return s.getString("subKeyFile")
  385. }
  386. func (s *SettingService) GetSubUpdates() (string, error) {
  387. return s.getString("subUpdates")
  388. }
  389. func (s *SettingService) GetSubEncrypt() (bool, error) {
  390. return s.getBool("subEncrypt")
  391. }
  392. func (s *SettingService) GetSubShowInfo() (bool, error) {
  393. return s.getBool("subShowInfo")
  394. }
  395. func (s *SettingService) GetPageSize() (int, error) {
  396. return s.getInt("pageSize")
  397. }
  398. func (s *SettingService) GetSubURI() (string, error) {
  399. return s.getString("subURI")
  400. }
  401. func (s *SettingService) GetSubJsonURI() (string, error) {
  402. return s.getString("subJsonURI")
  403. }
  404. func (s *SettingService) GetSubJsonFragment() (string, error) {
  405. return s.getString("subJsonFragment")
  406. }
  407. func (s *SettingService) GetSubJsonNoises() (string, error) {
  408. return s.getString("subJsonNoises")
  409. }
  410. func (s *SettingService) GetSubJsonMux() (string, error) {
  411. return s.getString("subJsonMux")
  412. }
  413. func (s *SettingService) GetSubJsonRules() (string, error) {
  414. return s.getString("subJsonRules")
  415. }
  416. func (s *SettingService) GetDatepicker() (string, error) {
  417. return s.getString("datepicker")
  418. }
  419. func (s *SettingService) GetWarp() (string, error) {
  420. return s.getString("warp")
  421. }
  422. func (s *SettingService) SetWarp(data string) error {
  423. return s.setString("warp", data)
  424. }
  425. func (s *SettingService) GetIpLimitEnable() (bool, error) {
  426. accessLogPath, err := xray.GetAccessLogPath()
  427. if err != nil {
  428. return false, err
  429. }
  430. return (accessLogPath != "none" && accessLogPath != ""), nil
  431. }
  432. func (s *SettingService) UpdateAllSetting(allSetting *entity.AllSetting) error {
  433. if err := allSetting.CheckValid(); err != nil {
  434. return err
  435. }
  436. v := reflect.ValueOf(allSetting).Elem()
  437. t := reflect.TypeOf(allSetting).Elem()
  438. fields := reflect_util.GetFields(t)
  439. errs := make([]error, 0)
  440. for _, field := range fields {
  441. key := field.Tag.Get("json")
  442. fieldV := v.FieldByName(field.Name)
  443. value := fmt.Sprint(fieldV.Interface())
  444. err := s.saveSetting(key, value)
  445. if err != nil {
  446. errs = append(errs, err)
  447. }
  448. }
  449. return common.Combine(errs...)
  450. }
  451. func (s *SettingService) GetDefaultXrayConfig() (interface{}, error) {
  452. var jsonData interface{}
  453. err := json.Unmarshal([]byte(xrayTemplateConfig), &jsonData)
  454. if err != nil {
  455. return nil, err
  456. }
  457. return jsonData, nil
  458. }
  459. func (s *SettingService) GetDefaultSettings(host string) (interface{}, error) {
  460. type settingFunc func() (interface{}, error)
  461. settings := map[string]settingFunc{
  462. "expireDiff": func() (interface{}, error) { return s.GetExpireDiff() },
  463. "trafficDiff": func() (interface{}, error) { return s.GetTrafficDiff() },
  464. "pageSize": func() (interface{}, error) { return s.GetPageSize() },
  465. "defaultCert": func() (interface{}, error) { return s.GetCertFile() },
  466. "defaultKey": func() (interface{}, error) { return s.GetKeyFile() },
  467. "tgBotEnable": func() (interface{}, error) { return s.GetTgbotEnabled() },
  468. "subEnable": func() (interface{}, error) { return s.GetSubEnable() },
  469. "subSyncEnable": func() (interface{}, error) { return s.GetSubSyncEnable() },
  470. "subURI": func() (interface{}, error) { return s.GetSubURI() },
  471. "subJsonURI": func() (interface{}, error) { return s.GetSubJsonURI() },
  472. "remarkModel": func() (interface{}, error) { return s.GetRemarkModel() },
  473. "datepicker": func() (interface{}, error) { return s.GetDatepicker() },
  474. "ipLimitEnable": func() (interface{}, error) { return s.GetIpLimitEnable() },
  475. }
  476. result := make(map[string]interface{})
  477. for key, fn := range settings {
  478. value, err := fn()
  479. if err != nil {
  480. return "", err
  481. }
  482. result[key] = value
  483. }
  484. if result["subEnable"].(bool) && (result["subURI"].(string) == "" || result["subJsonURI"].(string) == "") {
  485. subURI := ""
  486. subPort, _ := s.GetSubPort()
  487. subPath, _ := s.GetSubPath()
  488. subJsonPath, _ := s.GetSubJsonPath()
  489. subDomain, _ := s.GetSubDomain()
  490. subKeyFile, _ := s.GetSubKeyFile()
  491. subCertFile, _ := s.GetSubCertFile()
  492. subTLS := false
  493. if subKeyFile != "" && subCertFile != "" {
  494. subTLS = true
  495. }
  496. if subDomain == "" {
  497. subDomain = strings.Split(host, ":")[0]
  498. }
  499. if subTLS {
  500. subURI = "https://"
  501. } else {
  502. subURI = "http://"
  503. }
  504. if (subPort == 443 && subTLS) || (subPort == 80 && !subTLS) {
  505. subURI += subDomain
  506. } else {
  507. subURI += fmt.Sprintf("%s:%d", subDomain, subPort)
  508. }
  509. if result["subURI"].(string) == "" {
  510. result["subURI"] = subURI + subPath
  511. }
  512. if result["subJsonURI"].(string) == "" {
  513. result["subJsonURI"] = subURI + subJsonPath
  514. }
  515. }
  516. return result, nil
  517. }