setting.go 25 KB

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