setting.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997
  1. package service
  2. import (
  3. _ "embed"
  4. "encoding/json"
  5. "errors"
  6. "fmt"
  7. "net"
  8. "net/http"
  9. "reflect"
  10. "strconv"
  11. "strings"
  12. "time"
  13. "github.com/mhsanaei/3x-ui/v3/database"
  14. "github.com/mhsanaei/3x-ui/v3/database/model"
  15. "github.com/mhsanaei/3x-ui/v3/logger"
  16. "github.com/mhsanaei/3x-ui/v3/util/common"
  17. "github.com/mhsanaei/3x-ui/v3/util/netproxy"
  18. "github.com/mhsanaei/3x-ui/v3/util/random"
  19. "github.com/mhsanaei/3x-ui/v3/util/reflect_util"
  20. "github.com/mhsanaei/3x-ui/v3/web/entity"
  21. "github.com/mhsanaei/3x-ui/v3/xray"
  22. )
  23. //go:embed config.json
  24. var xrayTemplateConfig string
  25. var defaultValueMap = map[string]string{
  26. "xrayTemplateConfig": xrayTemplateConfig,
  27. "webListen": "",
  28. "webDomain": "",
  29. "webPort": "2053",
  30. "webCertFile": "",
  31. "webKeyFile": "",
  32. "secret": random.Seq(32),
  33. "apiToken": "",
  34. "webBasePath": "/",
  35. "sessionMaxAge": "360",
  36. "trustedProxyCIDRs": "127.0.0.1/32,::1/128",
  37. "pageSize": "25",
  38. "expireDiff": "0",
  39. "trafficDiff": "0",
  40. "remarkModel": "-io",
  41. "timeLocation": "Local",
  42. "tgBotEnable": "false",
  43. "tgBotToken": "",
  44. "tgBotProxy": "",
  45. "tgBotAPIServer": "",
  46. "tgBotChatId": "",
  47. "tgRunTime": "@daily",
  48. "tgBotBackup": "false",
  49. "tgBotLoginNotify": "true",
  50. "tgCpu": "80",
  51. "tgLang": "en-US",
  52. "twoFactorEnable": "false",
  53. "twoFactorToken": "",
  54. "subEnable": "true",
  55. "subJsonEnable": "false",
  56. "subTitle": "",
  57. "subSupportUrl": "",
  58. "subProfileUrl": "",
  59. "subAnnounce": "",
  60. "subEnableRouting": "true",
  61. "subRoutingRules": "",
  62. "subListen": "",
  63. "subPort": "2096",
  64. "subPath": "/sub/",
  65. "subDomain": "",
  66. "subCertFile": "",
  67. "subKeyFile": "",
  68. "subUpdates": "12",
  69. "subEncrypt": "true",
  70. "subShowInfo": "true",
  71. "subEmailInRemark": "true",
  72. "subURI": "",
  73. "subJsonPath": "/json/",
  74. "subJsonURI": "",
  75. "subClashEnable": "true",
  76. "subClashPath": "/clash/",
  77. "subClashURI": "",
  78. "subJsonFragment": "",
  79. "subJsonNoises": "",
  80. "subJsonMux": "",
  81. "subJsonRules": "",
  82. "datepicker": "gregorian",
  83. "warp": "",
  84. "nord": "",
  85. "externalTrafficInformEnable": "false",
  86. "externalTrafficInformURI": "",
  87. "restartXrayOnClientDisable": "true",
  88. "xrayOutboundTestUrl": "https://www.google.com/generate_204",
  89. "panelProxy": "",
  90. // LDAP defaults
  91. "ldapEnable": "false",
  92. "ldapHost": "",
  93. "ldapPort": "389",
  94. "ldapUseTLS": "false",
  95. "ldapBindDN": "",
  96. "ldapPassword": "",
  97. "ldapBaseDN": "",
  98. "ldapUserFilter": "(objectClass=person)",
  99. "ldapUserAttr": "mail",
  100. "ldapVlessField": "vless_enabled",
  101. "ldapSyncCron": "@every 1m",
  102. "ldapFlagField": "",
  103. "ldapTruthyValues": "true,1,yes,on",
  104. "ldapInvertFlag": "false",
  105. "ldapInboundTags": "",
  106. "ldapAutoCreate": "false",
  107. "ldapAutoDelete": "false",
  108. "ldapDefaultTotalGB": "0",
  109. "ldapDefaultExpiryDays": "0",
  110. "ldapDefaultLimitIP": "0",
  111. }
  112. // SettingService provides business logic for application settings management.
  113. // It handles configuration storage, retrieval, and validation for all system settings.
  114. type SettingService struct{}
  115. func (s *SettingService) GetDefaultJSONConfig() (any, error) {
  116. var jsonData any
  117. err := json.Unmarshal([]byte(xrayTemplateConfig), &jsonData)
  118. if err != nil {
  119. return nil, err
  120. }
  121. return jsonData, nil
  122. }
  123. func (s *SettingService) GetAllSetting() (*entity.AllSetting, error) {
  124. db := database.GetDB()
  125. settings := make([]*model.Setting, 0)
  126. err := db.Model(model.Setting{}).Not("key = ?", "xrayTemplateConfig").Find(&settings).Error
  127. if err != nil {
  128. return nil, err
  129. }
  130. allSetting := &entity.AllSetting{}
  131. t := reflect.TypeFor[entity.AllSetting]()
  132. v := reflect.ValueOf(allSetting).Elem()
  133. fields := reflect_util.GetFields(t)
  134. setSetting := func(key, value string) (err error) {
  135. defer func() {
  136. panicErr := recover()
  137. if panicErr != nil {
  138. err = errors.New(fmt.Sprint(panicErr))
  139. }
  140. }()
  141. var found bool
  142. var field reflect.StructField
  143. for _, f := range fields {
  144. if f.Tag.Get("json") == key {
  145. field = f
  146. found = true
  147. break
  148. }
  149. }
  150. if !found {
  151. // Some settings are automatically generated, no need to return to the front end to modify the user
  152. return nil
  153. }
  154. fieldV := v.FieldByName(field.Name)
  155. switch t := fieldV.Interface().(type) {
  156. case int:
  157. n, err := strconv.ParseInt(value, 10, 64)
  158. if err != nil {
  159. return err
  160. }
  161. fieldV.SetInt(n)
  162. case string:
  163. fieldV.SetString(value)
  164. case bool:
  165. fieldV.SetBool(value == "true")
  166. default:
  167. return common.NewErrorf("unknown field %v type %v", key, t)
  168. }
  169. return
  170. }
  171. keyMap := map[string]bool{}
  172. for _, setting := range settings {
  173. err := setSetting(setting.Key, setting.Value)
  174. if err != nil {
  175. return nil, err
  176. }
  177. keyMap[setting.Key] = true
  178. }
  179. for key, value := range defaultValueMap {
  180. if keyMap[key] {
  181. continue
  182. }
  183. err := setSetting(key, value)
  184. if err != nil {
  185. return nil, err
  186. }
  187. }
  188. return allSetting, nil
  189. }
  190. func (s *SettingService) GetAllSettingView() (*entity.AllSettingView, error) {
  191. allSetting, err := s.GetAllSetting()
  192. if err != nil {
  193. return nil, err
  194. }
  195. view := &entity.AllSettingView{AllSetting: *allSetting}
  196. view.HasTgBotToken = secretConfigured(allSetting.TgBotToken)
  197. view.HasTwoFactorToken = secretConfigured(allSetting.TwoFactorToken)
  198. view.HasLdapPassword = secretConfigured(allSetting.LdapPassword)
  199. view.HasWarpSecret = secretConfigured(mustString(s.GetWarp()))
  200. view.HasNordSecret = secretConfigured(mustString(s.GetNord()))
  201. var apiTokenCount int64
  202. if err := database.GetDB().Model(model.ApiToken{}).Where("enabled = ?", true).Count(&apiTokenCount).Error; err == nil {
  203. view.HasApiToken = apiTokenCount > 0
  204. }
  205. view.TgBotToken = ""
  206. view.TwoFactorToken = ""
  207. view.LdapPassword = ""
  208. return view, nil
  209. }
  210. func secretConfigured(value string) bool {
  211. return strings.TrimSpace(value) != ""
  212. }
  213. func mustString(value string, _ error) string {
  214. return value
  215. }
  216. func (s *SettingService) ResetSettings() error {
  217. db := database.GetDB()
  218. err := db.Where("1 = 1").Delete(model.Setting{}).Error
  219. if err != nil {
  220. return err
  221. }
  222. return db.Model(model.User{}).
  223. Where("1 = 1").Error
  224. }
  225. func (s *SettingService) getSetting(key string) (*model.Setting, error) {
  226. db := database.GetDB()
  227. setting := &model.Setting{}
  228. err := db.Model(model.Setting{}).Where("key = ?", key).First(setting).Error
  229. if err != nil {
  230. return nil, err
  231. }
  232. return setting, nil
  233. }
  234. func (s *SettingService) saveSetting(key string, value string) error {
  235. setting, err := s.getSetting(key)
  236. db := database.GetDB()
  237. if database.IsNotFound(err) {
  238. return db.Create(&model.Setting{
  239. Key: key,
  240. Value: value,
  241. }).Error
  242. } else if err != nil {
  243. return err
  244. }
  245. setting.Key = key
  246. setting.Value = value
  247. return db.Save(setting).Error
  248. }
  249. func (s *SettingService) getString(key string) (string, error) {
  250. setting, err := s.getSetting(key)
  251. if database.IsNotFound(err) {
  252. value, ok := defaultValueMap[key]
  253. if !ok {
  254. return "", common.NewErrorf("key <%v> not in defaultValueMap", key)
  255. }
  256. return value, nil
  257. } else if err != nil {
  258. return "", err
  259. }
  260. return setting.Value, nil
  261. }
  262. func (s *SettingService) setString(key string, value string) error {
  263. return s.saveSetting(key, value)
  264. }
  265. func (s *SettingService) getBool(key string) (bool, error) {
  266. str, err := s.getString(key)
  267. if err != nil {
  268. return false, err
  269. }
  270. return strconv.ParseBool(str)
  271. }
  272. func (s *SettingService) setBool(key string, value bool) error {
  273. return s.setString(key, strconv.FormatBool(value))
  274. }
  275. func (s *SettingService) getInt(key string) (int, error) {
  276. str, err := s.getString(key)
  277. if err != nil {
  278. return 0, err
  279. }
  280. return strconv.Atoi(str)
  281. }
  282. func (s *SettingService) setInt(key string, value int) error {
  283. return s.setString(key, strconv.Itoa(value))
  284. }
  285. func (s *SettingService) GetXrayConfigTemplate() (string, error) {
  286. return s.getString("xrayTemplateConfig")
  287. }
  288. func (s *SettingService) GetXrayOutboundTestUrl() (string, error) {
  289. return s.getString("xrayOutboundTestUrl")
  290. }
  291. func (s *SettingService) SetXrayOutboundTestUrl(url string) error {
  292. clean, err := SanitizeHTTPURL(url)
  293. if err != nil {
  294. return err
  295. }
  296. return s.setString("xrayOutboundTestUrl", clean)
  297. }
  298. func (s *SettingService) GetListen() (string, error) {
  299. return s.getString("webListen")
  300. }
  301. func (s *SettingService) SetListen(ip string) error {
  302. return s.setString("webListen", ip)
  303. }
  304. func (s *SettingService) GetWebDomain() (string, error) {
  305. return s.getString("webDomain")
  306. }
  307. func (s *SettingService) GetTgBotToken() (string, error) {
  308. return s.getString("tgBotToken")
  309. }
  310. func (s *SettingService) SetTgBotToken(token string) error {
  311. return s.setString("tgBotToken", token)
  312. }
  313. func (s *SettingService) GetTgBotProxy() (string, error) {
  314. return s.getString("tgBotProxy")
  315. }
  316. func (s *SettingService) SetTgBotProxy(token string) error {
  317. return s.setString("tgBotProxy", token)
  318. }
  319. func (s *SettingService) GetPanelProxy() (string, error) {
  320. return s.getString("panelProxy")
  321. }
  322. func (s *SettingService) SetPanelProxy(proxyUrl string) error {
  323. return s.setString("panelProxy", proxyUrl)
  324. }
  325. // NewProxiedHTTPClient returns an HTTP client that routes the panel's own
  326. // outbound requests through the configured panelProxy setting. An invalid or
  327. // missing proxy falls back to a direct client so existing behavior is preserved.
  328. func (s *SettingService) NewProxiedHTTPClient(timeout time.Duration) *http.Client {
  329. proxyUrl, err := s.GetPanelProxy()
  330. if err != nil {
  331. logger.Warning("Failed to read panel proxy setting:", err)
  332. proxyUrl = ""
  333. }
  334. client, err := netproxy.NewHTTPClient(proxyUrl, timeout)
  335. if err != nil {
  336. logger.Warningf("Invalid panel proxy %q, using direct connection: %v", proxyUrl, err)
  337. return &http.Client{Timeout: timeout}
  338. }
  339. return client
  340. }
  341. func (s *SettingService) GetTgBotAPIServer() (string, error) {
  342. return s.getString("tgBotAPIServer")
  343. }
  344. func (s *SettingService) SetTgBotAPIServer(token string) error {
  345. return s.setString("tgBotAPIServer", token)
  346. }
  347. func (s *SettingService) GetTgBotChatId() (string, error) {
  348. return s.getString("tgBotChatId")
  349. }
  350. func (s *SettingService) SetTgBotChatId(chatIds string) error {
  351. return s.setString("tgBotChatId", chatIds)
  352. }
  353. func (s *SettingService) GetTgbotEnabled() (bool, error) {
  354. return s.getBool("tgBotEnable")
  355. }
  356. func (s *SettingService) SetTgbotEnabled(value bool) error {
  357. return s.setBool("tgBotEnable", value)
  358. }
  359. func (s *SettingService) GetTgbotRuntime() (string, error) {
  360. return s.getString("tgRunTime")
  361. }
  362. func (s *SettingService) SetTgbotRuntime(time string) error {
  363. return s.setString("tgRunTime", time)
  364. }
  365. func (s *SettingService) GetTgBotBackup() (bool, error) {
  366. return s.getBool("tgBotBackup")
  367. }
  368. func (s *SettingService) GetTgBotLoginNotify() (bool, error) {
  369. return s.getBool("tgBotLoginNotify")
  370. }
  371. func (s *SettingService) GetTgCpu() (int, error) {
  372. return s.getInt("tgCpu")
  373. }
  374. func (s *SettingService) GetTgLang() (string, error) {
  375. return s.getString("tgLang")
  376. }
  377. func (s *SettingService) GetTwoFactorEnable() (bool, error) {
  378. return s.getBool("twoFactorEnable")
  379. }
  380. func (s *SettingService) SetTwoFactorEnable(value bool) error {
  381. return s.setBool("twoFactorEnable", value)
  382. }
  383. func (s *SettingService) GetTwoFactorToken() (string, error) {
  384. return s.getString("twoFactorToken")
  385. }
  386. func (s *SettingService) SetTwoFactorToken(value string) error {
  387. return s.setString("twoFactorToken", value)
  388. }
  389. func (s *SettingService) GetPort() (int, error) {
  390. return s.getInt("webPort")
  391. }
  392. func (s *SettingService) SetPort(port int) error {
  393. return s.setInt("webPort", port)
  394. }
  395. func (s *SettingService) SetCertFile(webCertFile string) error {
  396. return s.setString("webCertFile", webCertFile)
  397. }
  398. func (s *SettingService) GetCertFile() (string, error) {
  399. return s.getString("webCertFile")
  400. }
  401. func (s *SettingService) SetKeyFile(webKeyFile string) error {
  402. return s.setString("webKeyFile", webKeyFile)
  403. }
  404. func (s *SettingService) GetKeyFile() (string, error) {
  405. return s.getString("webKeyFile")
  406. }
  407. func (s *SettingService) GetExpireDiff() (int, error) {
  408. return s.getInt("expireDiff")
  409. }
  410. func (s *SettingService) GetTrafficDiff() (int, error) {
  411. return s.getInt("trafficDiff")
  412. }
  413. func (s *SettingService) GetSessionMaxAge() (int, error) {
  414. return s.getInt("sessionMaxAge")
  415. }
  416. func (s *SettingService) GetTrustedProxyCIDRs() (string, error) {
  417. return s.getString("trustedProxyCIDRs")
  418. }
  419. func (s *SettingService) GetRemarkModel() (string, error) {
  420. return s.getString("remarkModel")
  421. }
  422. func (s *SettingService) GetSecret() ([]byte, error) {
  423. secret, err := s.getString("secret")
  424. if secret == defaultValueMap["secret"] {
  425. err := s.saveSetting("secret", secret)
  426. if err != nil {
  427. logger.Warning("save secret failed:", err)
  428. }
  429. }
  430. return []byte(secret), err
  431. }
  432. func (s *SettingService) SetBasePath(basePath string) error {
  433. if !strings.HasPrefix(basePath, "/") {
  434. basePath = "/" + basePath
  435. }
  436. if !strings.HasSuffix(basePath, "/") {
  437. basePath += "/"
  438. }
  439. return s.setString("webBasePath", basePath)
  440. }
  441. func (s *SettingService) GetBasePath() (string, error) {
  442. basePath, err := s.getString("webBasePath")
  443. if err != nil {
  444. return "", err
  445. }
  446. if !strings.HasPrefix(basePath, "/") {
  447. basePath = "/" + basePath
  448. }
  449. if !strings.HasSuffix(basePath, "/") {
  450. basePath += "/"
  451. }
  452. return basePath, nil
  453. }
  454. func (s *SettingService) GetTimeLocation() (*time.Location, error) {
  455. l, err := s.getString("timeLocation")
  456. if err != nil {
  457. return nil, err
  458. }
  459. location, err := time.LoadLocation(l)
  460. if err != nil {
  461. defaultLocation := defaultValueMap["timeLocation"]
  462. logger.Errorf("location <%v> not exist, using default location: %v", l, defaultLocation)
  463. location, err = time.LoadLocation(defaultLocation)
  464. if err != nil {
  465. logger.Errorf("failed to load default location, using UTC: %v", err)
  466. return time.UTC, nil
  467. }
  468. return location, nil
  469. }
  470. return location, nil
  471. }
  472. func (s *SettingService) GetSubEnable() (bool, error) {
  473. return s.getBool("subEnable")
  474. }
  475. func (s *SettingService) GetSubJsonEnable() (bool, error) {
  476. return s.getBool("subJsonEnable")
  477. }
  478. func (s *SettingService) GetSubTitle() (string, error) {
  479. return s.getString("subTitle")
  480. }
  481. func (s *SettingService) GetSubSupportUrl() (string, error) {
  482. return s.getString("subSupportUrl")
  483. }
  484. func (s *SettingService) GetSubProfileUrl() (string, error) {
  485. return s.getString("subProfileUrl")
  486. }
  487. func (s *SettingService) GetSubAnnounce() (string, error) {
  488. return s.getString("subAnnounce")
  489. }
  490. func (s *SettingService) GetSubEnableRouting() (bool, error) {
  491. return s.getBool("subEnableRouting")
  492. }
  493. func (s *SettingService) GetSubRoutingRules() (string, error) {
  494. return s.getString("subRoutingRules")
  495. }
  496. func (s *SettingService) GetSubListen() (string, error) {
  497. return s.getString("subListen")
  498. }
  499. func (s *SettingService) GetSubPort() (int, error) {
  500. return s.getInt("subPort")
  501. }
  502. func (s *SettingService) GetSubPath() (string, error) {
  503. return s.getString("subPath")
  504. }
  505. func (s *SettingService) GetSubJsonPath() (string, error) {
  506. return s.getString("subJsonPath")
  507. }
  508. func (s *SettingService) GetSubDomain() (string, error) {
  509. return s.getString("subDomain")
  510. }
  511. func (s *SettingService) SetSubCertFile(subCertFile string) error {
  512. return s.setString("subCertFile", subCertFile)
  513. }
  514. func (s *SettingService) GetSubCertFile() (string, error) {
  515. return s.getString("subCertFile")
  516. }
  517. func (s *SettingService) SetSubKeyFile(subKeyFile string) error {
  518. return s.setString("subKeyFile", subKeyFile)
  519. }
  520. func (s *SettingService) GetSubKeyFile() (string, error) {
  521. return s.getString("subKeyFile")
  522. }
  523. func (s *SettingService) GetSubUpdates() (string, error) {
  524. return s.getString("subUpdates")
  525. }
  526. func (s *SettingService) GetSubEncrypt() (bool, error) {
  527. return s.getBool("subEncrypt")
  528. }
  529. func (s *SettingService) GetSubShowInfo() (bool, error) {
  530. return s.getBool("subShowInfo")
  531. }
  532. func (s *SettingService) GetSubEmailInRemark() (bool, error) {
  533. return s.getBool("subEmailInRemark")
  534. }
  535. func (s *SettingService) GetPageSize() (int, error) {
  536. return s.getInt("pageSize")
  537. }
  538. func (s *SettingService) GetSubURI() (string, error) {
  539. return s.getString("subURI")
  540. }
  541. func (s *SettingService) GetSubJsonURI() (string, error) {
  542. return s.getString("subJsonURI")
  543. }
  544. func (s *SettingService) GetSubClashEnable() (bool, error) {
  545. return s.getBool("subClashEnable")
  546. }
  547. func (s *SettingService) GetSubClashPath() (string, error) {
  548. return s.getString("subClashPath")
  549. }
  550. func (s *SettingService) GetSubClashURI() (string, error) {
  551. return s.getString("subClashURI")
  552. }
  553. func (s *SettingService) GetSubJsonFragment() (string, error) {
  554. return s.getString("subJsonFragment")
  555. }
  556. func (s *SettingService) GetSubJsonNoises() (string, error) {
  557. return s.getString("subJsonNoises")
  558. }
  559. func (s *SettingService) GetSubJsonMux() (string, error) {
  560. return s.getString("subJsonMux")
  561. }
  562. func (s *SettingService) GetSubJsonRules() (string, error) {
  563. return s.getString("subJsonRules")
  564. }
  565. func (s *SettingService) GetDatepicker() (string, error) {
  566. return s.getString("datepicker")
  567. }
  568. func (s *SettingService) GetWarp() (string, error) {
  569. return s.getString("warp")
  570. }
  571. func (s *SettingService) SetWarp(data string) error {
  572. return s.setString("warp", data)
  573. }
  574. func (s *SettingService) GetNord() (string, error) {
  575. return s.getString("nord")
  576. }
  577. func (s *SettingService) SetNord(data string) error {
  578. return s.setString("nord", data)
  579. }
  580. func (s *SettingService) GetExternalTrafficInformEnable() (bool, error) {
  581. return s.getBool("externalTrafficInformEnable")
  582. }
  583. func (s *SettingService) SetExternalTrafficInformEnable(value bool) error {
  584. return s.setBool("externalTrafficInformEnable", value)
  585. }
  586. func (s *SettingService) GetExternalTrafficInformURI() (string, error) {
  587. return s.getString("externalTrafficInformURI")
  588. }
  589. func (s *SettingService) SetExternalTrafficInformURI(InformURI string) error {
  590. return s.setString("externalTrafficInformURI", InformURI)
  591. }
  592. func (s *SettingService) GetRestartXrayOnClientDisable() (bool, error) {
  593. return s.getBool("restartXrayOnClientDisable")
  594. }
  595. func (s *SettingService) SetRestartXrayOnClientDisable(value bool) error {
  596. return s.setBool("restartXrayOnClientDisable", value)
  597. }
  598. func (s *SettingService) GetIpLimitEnable() (bool, error) {
  599. accessLogPath, err := xray.GetAccessLogPath()
  600. if err != nil {
  601. return false, err
  602. }
  603. return (accessLogPath != "none" && accessLogPath != ""), nil
  604. }
  605. // GetLdapEnable returns whether LDAP is enabled.
  606. func (s *SettingService) GetLdapEnable() (bool, error) {
  607. return s.getBool("ldapEnable")
  608. }
  609. func (s *SettingService) GetLdapHost() (string, error) {
  610. return s.getString("ldapHost")
  611. }
  612. func (s *SettingService) GetLdapPort() (int, error) {
  613. return s.getInt("ldapPort")
  614. }
  615. func (s *SettingService) GetLdapUseTLS() (bool, error) {
  616. return s.getBool("ldapUseTLS")
  617. }
  618. func (s *SettingService) GetLdapBindDN() (string, error) {
  619. return s.getString("ldapBindDN")
  620. }
  621. func (s *SettingService) GetLdapPassword() (string, error) {
  622. return s.getString("ldapPassword")
  623. }
  624. func (s *SettingService) GetLdapBaseDN() (string, error) {
  625. return s.getString("ldapBaseDN")
  626. }
  627. func (s *SettingService) GetLdapUserFilter() (string, error) {
  628. return s.getString("ldapUserFilter")
  629. }
  630. func (s *SettingService) GetLdapUserAttr() (string, error) {
  631. return s.getString("ldapUserAttr")
  632. }
  633. func (s *SettingService) GetLdapVlessField() (string, error) {
  634. return s.getString("ldapVlessField")
  635. }
  636. func (s *SettingService) GetLdapSyncCron() (string, error) {
  637. return s.getString("ldapSyncCron")
  638. }
  639. func (s *SettingService) GetLdapFlagField() (string, error) {
  640. return s.getString("ldapFlagField")
  641. }
  642. func (s *SettingService) GetLdapTruthyValues() (string, error) {
  643. return s.getString("ldapTruthyValues")
  644. }
  645. func (s *SettingService) GetLdapInvertFlag() (bool, error) {
  646. return s.getBool("ldapInvertFlag")
  647. }
  648. func (s *SettingService) GetLdapInboundTags() (string, error) {
  649. return s.getString("ldapInboundTags")
  650. }
  651. func (s *SettingService) GetLdapAutoCreate() (bool, error) {
  652. return s.getBool("ldapAutoCreate")
  653. }
  654. func (s *SettingService) GetLdapAutoDelete() (bool, error) {
  655. return s.getBool("ldapAutoDelete")
  656. }
  657. func (s *SettingService) GetLdapDefaultTotalGB() (int, error) {
  658. return s.getInt("ldapDefaultTotalGB")
  659. }
  660. func (s *SettingService) GetLdapDefaultExpiryDays() (int, error) {
  661. return s.getInt("ldapDefaultExpiryDays")
  662. }
  663. func (s *SettingService) GetLdapDefaultLimitIP() (int, error) {
  664. return s.getInt("ldapDefaultLimitIP")
  665. }
  666. func (s *SettingService) UpdateAllSetting(allSetting *entity.AllSetting) error {
  667. if err := s.preserveRedactedSecrets(allSetting); err != nil {
  668. return err
  669. }
  670. if err := validateSettingsURLs(allSetting); err != nil {
  671. return err
  672. }
  673. if err := allSetting.CheckValid(); err != nil {
  674. return err
  675. }
  676. v := reflect.ValueOf(allSetting).Elem()
  677. t := reflect.TypeFor[entity.AllSetting]()
  678. fields := reflect_util.GetFields(t)
  679. errs := make([]error, 0)
  680. for _, field := range fields {
  681. key := field.Tag.Get("json")
  682. fieldV := v.FieldByName(field.Name)
  683. value := fmt.Sprint(fieldV.Interface())
  684. err := s.saveSetting(key, value)
  685. if err != nil {
  686. errs = append(errs, err)
  687. }
  688. }
  689. return common.Combine(errs...)
  690. }
  691. func (s *SettingService) preserveRedactedSecrets(allSetting *entity.AllSetting) error {
  692. if strings.TrimSpace(allSetting.TgBotToken) == "" {
  693. value, err := s.GetTgBotToken()
  694. if err != nil {
  695. return err
  696. }
  697. allSetting.TgBotToken = value
  698. }
  699. if strings.TrimSpace(allSetting.LdapPassword) == "" {
  700. value, err := s.GetLdapPassword()
  701. if err != nil {
  702. return err
  703. }
  704. allSetting.LdapPassword = value
  705. }
  706. if allSetting.TwoFactorEnable && strings.TrimSpace(allSetting.TwoFactorToken) == "" {
  707. value, err := s.GetTwoFactorToken()
  708. if err != nil {
  709. return err
  710. }
  711. allSetting.TwoFactorToken = value
  712. }
  713. return nil
  714. }
  715. func validateSettingsURLs(allSetting *entity.AllSetting) error {
  716. if allSetting.ExternalTrafficInformURI != "" {
  717. u, err := SanitizeHTTPURL(allSetting.ExternalTrafficInformURI)
  718. if err != nil {
  719. return common.NewError("external traffic inform URI is invalid:", err)
  720. }
  721. allSetting.ExternalTrafficInformURI = u
  722. }
  723. if allSetting.TgBotAPIServer != "" {
  724. u, err := SanitizeHTTPURL(allSetting.TgBotAPIServer)
  725. if err != nil {
  726. return common.NewError("telegram API server URL is invalid:", err)
  727. }
  728. allSetting.TgBotAPIServer = u
  729. }
  730. return nil
  731. }
  732. func (s *SettingService) UpdateSecret(key string, value string) error {
  733. switch key {
  734. case "tgBotToken", "ldapPassword", "twoFactorToken":
  735. return s.saveSetting(key, strings.TrimSpace(value))
  736. default:
  737. return common.NewError("secret key is not replaceable:", key)
  738. }
  739. }
  740. func (s *SettingService) GetDefaultXrayConfig() (any, error) {
  741. var jsonData any
  742. err := json.Unmarshal([]byte(xrayTemplateConfig), &jsonData)
  743. if err != nil {
  744. return nil, err
  745. }
  746. return jsonData, nil
  747. }
  748. func extractHostname(host string) string {
  749. h, _, err := net.SplitHostPort(host)
  750. // Err is not nil means host does not contain port
  751. if err != nil {
  752. h = host
  753. }
  754. ip := net.ParseIP(h)
  755. // If it's not an IP, return as is
  756. if ip == nil {
  757. return h
  758. }
  759. // If it's an IPv4, return as is
  760. if ip.To4() != nil {
  761. return h
  762. }
  763. // IPv6 needs bracketing
  764. return "[" + h + "]"
  765. }
  766. func (s *SettingService) GetDefaultSettings(host string) (any, error) {
  767. type settingFunc func() (any, error)
  768. settings := map[string]settingFunc{
  769. "expireDiff": func() (any, error) { return s.GetExpireDiff() },
  770. "trafficDiff": func() (any, error) { return s.GetTrafficDiff() },
  771. "pageSize": func() (any, error) { return s.GetPageSize() },
  772. "defaultCert": func() (any, error) { return s.GetCertFile() },
  773. "defaultKey": func() (any, error) { return s.GetKeyFile() },
  774. "tgBotEnable": func() (any, error) { return s.GetTgbotEnabled() },
  775. "subEnable": func() (any, error) { return s.GetSubEnable() },
  776. "subJsonEnable": func() (any, error) { return s.GetSubJsonEnable() },
  777. "subClashEnable": func() (any, error) { return s.GetSubClashEnable() },
  778. "subTitle": func() (any, error) { return s.GetSubTitle() },
  779. "subURI": func() (any, error) { return s.GetSubURI() },
  780. "subJsonURI": func() (any, error) { return s.GetSubJsonURI() },
  781. "subClashURI": func() (any, error) { return s.GetSubClashURI() },
  782. "remarkModel": func() (any, error) { return s.GetRemarkModel() },
  783. "datepicker": func() (any, error) { return s.GetDatepicker() },
  784. "ipLimitEnable": func() (any, error) { return s.GetIpLimitEnable() },
  785. }
  786. result := make(map[string]any)
  787. for key, fn := range settings {
  788. value, err := fn()
  789. if err != nil {
  790. return "", err
  791. }
  792. result[key] = value
  793. }
  794. subEnable := result["subEnable"].(bool)
  795. subJsonEnable := false
  796. if v, ok := result["subJsonEnable"]; ok {
  797. if b, ok2 := v.(bool); ok2 {
  798. subJsonEnable = b
  799. }
  800. }
  801. subClashEnable := false
  802. if v, ok := result["subClashEnable"]; ok {
  803. if b, ok2 := v.(bool); ok2 {
  804. subClashEnable = b
  805. }
  806. }
  807. if (subEnable && result["subURI"].(string) == "") || (subJsonEnable && result["subJsonURI"].(string) == "") || (subClashEnable && result["subClashURI"].(string) == "") {
  808. subURI := ""
  809. subTitle, _ := s.GetSubTitle()
  810. subPort, _ := s.GetSubPort()
  811. subPath, _ := s.GetSubPath()
  812. subJsonPath, _ := s.GetSubJsonPath()
  813. subClashPath, _ := s.GetSubClashPath()
  814. subDomain, _ := s.GetSubDomain()
  815. subKeyFile, _ := s.GetSubKeyFile()
  816. subCertFile, _ := s.GetSubCertFile()
  817. subTLS := false
  818. if subKeyFile != "" && subCertFile != "" {
  819. subTLS = true
  820. }
  821. if subDomain == "" {
  822. subDomain = extractHostname(host)
  823. }
  824. if subTLS {
  825. subURI = "https://"
  826. } else {
  827. subURI = "http://"
  828. }
  829. if (subPort == 443 && subTLS) || (subPort == 80 && !subTLS) {
  830. subURI += subDomain
  831. } else {
  832. subURI += fmt.Sprintf("%s:%d", subDomain, subPort)
  833. }
  834. if subEnable && result["subURI"].(string) == "" {
  835. result["subURI"] = subURI + subPath
  836. }
  837. if result["subTitle"].(string) == "" {
  838. result["subTitle"] = subTitle
  839. }
  840. if subJsonEnable && result["subJsonURI"].(string) == "" {
  841. result["subJsonURI"] = subURI + subJsonPath
  842. }
  843. if subClashEnable && result["subClashURI"].(string) == "" {
  844. result["subClashURI"] = subURI + subClashPath
  845. }
  846. }
  847. return result, nil
  848. }