setting.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735
  1. package user
  2. import (
  3. "bytes"
  4. gocontext "context"
  5. "encoding/base64"
  6. "fmt"
  7. "html/template"
  8. "image/png"
  9. "io"
  10. "net/http"
  11. "github.com/cockroachdb/errors"
  12. "github.com/pquerna/otp"
  13. "github.com/pquerna/otp/totp"
  14. "gopkg.in/macaron.v1"
  15. log "unknwon.dev/clog/v2"
  16. "gogs.io/gogs/internal/auth"
  17. "gogs.io/gogs/internal/conf"
  18. "gogs.io/gogs/internal/context"
  19. "gogs.io/gogs/internal/cryptoutil"
  20. "gogs.io/gogs/internal/database"
  21. "gogs.io/gogs/internal/email"
  22. "gogs.io/gogs/internal/form"
  23. "gogs.io/gogs/internal/tool"
  24. "gogs.io/gogs/internal/userutil"
  25. )
  26. // SettingsHandler is the handler for users settings endpoints.
  27. type SettingsHandler struct {
  28. store SettingsStore
  29. }
  30. // NewSettingsHandler returns a new SettingsHandler for users settings endpoints.
  31. func NewSettingsHandler(s SettingsStore) *SettingsHandler {
  32. return &SettingsHandler{
  33. store: s,
  34. }
  35. }
  36. const (
  37. tmplUserSettingsProfile = "user/settings/profile"
  38. tmplUserSettingsAvatar = "user/settings/avatar"
  39. tmplUserSettingsPassword = "user/settings/password"
  40. tmplUserSettingsEmail = "user/settings/email"
  41. tmplUserSettingsSSHKeys = "user/settings/sshkeys"
  42. tmplUserSettingsSecurity = "user/settings/security"
  43. tmplUserSettingsTwoFactorEnable = "user/settings/two_factor_enable"
  44. tmplUserSettingsTwoFactorRecoveryCodes = "user/settings/two_factor_recovery_codes"
  45. tmplUserSettingsRepositories = "user/settings/repositories"
  46. tmplUserSettingsOrganizations = "user/settings/organizations"
  47. tmplUserSettingsApplications = "user/settings/applications"
  48. tmplUserSettingsDelete = "user/settings/delete"
  49. tmplUserNotification = "user/notification"
  50. )
  51. func Settings(c *context.Context) {
  52. c.Title("settings.profile")
  53. c.PageIs("SettingsProfile")
  54. c.Data["origin_name"] = c.User.Name
  55. c.Data["name"] = c.User.Name
  56. c.Data["full_name"] = c.User.FullName
  57. c.Data["email"] = c.User.Email
  58. c.Data["website"] = c.User.Website
  59. c.Data["location"] = c.User.Location
  60. c.Success(tmplUserSettingsProfile)
  61. }
  62. func SettingsPost(c *context.Context, f form.UpdateProfile) {
  63. c.Title("settings.profile")
  64. c.PageIs("SettingsProfile")
  65. c.Data["origin_name"] = c.User.Name
  66. if c.HasError() {
  67. c.HTML(http.StatusBadRequest, tmplUserSettingsProfile)
  68. return
  69. }
  70. // Non-local users are not allowed to change their username
  71. if c.User.IsLocal() {
  72. // Check if the username (including cases) had been changed
  73. if c.User.Name != f.Name {
  74. err := database.Handle.Users().ChangeUsername(c.Req.Context(), c.User.ID, f.Name)
  75. if err != nil {
  76. c.FormErr("Name")
  77. switch {
  78. case database.IsErrUserAlreadyExist(errors.Cause(err)):
  79. c.RenderWithErr(c.Tr("form.username_been_taken"), http.StatusUnprocessableEntity, tmplUserSettingsProfile, &f)
  80. case database.IsErrNameNotAllowed(errors.Cause(err)):
  81. c.RenderWithErr(c.Tr("user.form.name_not_allowed", err.(database.ErrNameNotAllowed).Value()), http.StatusBadRequest, tmplUserSettingsProfile, &f)
  82. default:
  83. c.Error(err, "change user name")
  84. }
  85. return
  86. }
  87. log.Trace("Username changed: %s -> %s", c.User.Name, f.Name)
  88. }
  89. }
  90. err := database.Handle.Users().Update(
  91. c.Req.Context(),
  92. c.User.ID,
  93. database.UpdateUserOptions{
  94. FullName: &f.FullName,
  95. Website: &f.Website,
  96. Location: &f.Location,
  97. },
  98. )
  99. if err != nil {
  100. c.Error(err, "update user")
  101. return
  102. }
  103. c.Flash.Success(c.Tr("settings.update_profile_success"))
  104. c.RedirectSubpath("/user/settings")
  105. }
  106. // FIXME: limit upload size
  107. func UpdateAvatarSetting(c *context.Context, f form.Avatar, ctxUser *database.User) error {
  108. if f.Source == form.AvatarLookup && f.Gravatar != "" {
  109. avatar := cryptoutil.MD5(f.Gravatar)
  110. err := database.Handle.Users().Update(
  111. c.Req.Context(),
  112. ctxUser.ID,
  113. database.UpdateUserOptions{
  114. Avatar: &avatar,
  115. AvatarEmail: &f.Gravatar,
  116. },
  117. )
  118. if err != nil {
  119. return errors.Wrap(err, "update user")
  120. }
  121. err = database.Handle.Users().DeleteCustomAvatar(c.Req.Context(), c.User.ID)
  122. if err != nil {
  123. return errors.Wrap(err, "delete custom avatar")
  124. }
  125. return nil
  126. }
  127. if f.Avatar != nil && f.Avatar.Filename != "" {
  128. r, err := f.Avatar.Open()
  129. if err != nil {
  130. return errors.Newf("open avatar reader: %v", err)
  131. }
  132. defer func() { _ = r.Close() }()
  133. data, err := io.ReadAll(r)
  134. if err != nil {
  135. return errors.Newf("read avatar content: %v", err)
  136. }
  137. if !tool.IsImageFile(data) {
  138. return errors.New(c.Tr("settings.uploaded_avatar_not_a_image"))
  139. }
  140. err = database.Handle.Users().UseCustomAvatar(c.Req.Context(), ctxUser.ID, data)
  141. if err != nil {
  142. return errors.Wrap(err, "save avatar")
  143. }
  144. return nil
  145. }
  146. return nil
  147. }
  148. func SettingsAvatar(c *context.Context) {
  149. c.Title("settings.avatar")
  150. c.PageIs("SettingsAvatar")
  151. c.Success(tmplUserSettingsAvatar)
  152. }
  153. func SettingsAvatarPost(c *context.Context, f form.Avatar) {
  154. if err := UpdateAvatarSetting(c, f, c.User); err != nil {
  155. c.Flash.Error(err.Error())
  156. } else {
  157. c.Flash.Success(c.Tr("settings.update_avatar_success"))
  158. }
  159. c.RedirectSubpath("/user/settings/avatar")
  160. }
  161. func SettingsDeleteAvatar(c *context.Context) {
  162. err := database.Handle.Users().DeleteCustomAvatar(c.Req.Context(), c.User.ID)
  163. if err != nil {
  164. c.Flash.Error(fmt.Sprintf("Failed to delete avatar: %v", err))
  165. }
  166. c.RedirectSubpath("/user/settings/avatar")
  167. }
  168. func SettingsPassword(c *context.Context) {
  169. c.Title("settings.password")
  170. c.PageIs("SettingsPassword")
  171. c.Success(tmplUserSettingsPassword)
  172. }
  173. func SettingsPasswordPost(c *context.Context, f form.ChangePassword) {
  174. c.Title("settings.password")
  175. c.PageIs("SettingsPassword")
  176. if c.HasError() {
  177. c.HTML(http.StatusBadRequest, tmplUserSettingsPassword)
  178. return
  179. }
  180. if !userutil.ValidatePassword(c.User.Password, c.User.Salt, f.OldPassword) {
  181. c.Flash.Error(c.Tr("settings.password_incorrect"))
  182. } else if f.Password != f.Retype {
  183. c.Flash.Error(c.Tr("form.password_not_match"))
  184. } else {
  185. err := database.Handle.Users().Update(
  186. c.Req.Context(),
  187. c.User.ID,
  188. database.UpdateUserOptions{
  189. Password: &f.Password,
  190. },
  191. )
  192. if err != nil {
  193. c.Errorf(err, "update user")
  194. return
  195. }
  196. c.Flash.Success(c.Tr("settings.change_password_success"))
  197. }
  198. c.RedirectSubpath("/user/settings/password")
  199. }
  200. func SettingsEmails(c *context.Context) {
  201. c.Title("settings.emails")
  202. c.PageIs("SettingsEmails")
  203. emails, err := database.Handle.Users().ListEmails(c.Req.Context(), c.User.ID)
  204. if err != nil {
  205. c.Errorf(err, "get email addresses")
  206. return
  207. }
  208. c.Data["Emails"] = emails
  209. c.Success(tmplUserSettingsEmail)
  210. }
  211. func SettingsEmailPost(c *context.Context, f form.AddEmail) {
  212. c.Title("settings.emails")
  213. c.PageIs("SettingsEmails")
  214. if c.Query("_method") == "PRIMARY" {
  215. err := database.Handle.Users().MarkEmailPrimary(c.Req.Context(), c.User.ID, c.Query("email"))
  216. if err != nil {
  217. c.Errorf(err, "make email primary")
  218. return
  219. }
  220. c.RedirectSubpath("/user/settings/email")
  221. return
  222. }
  223. // Add Email address.
  224. emails, err := database.Handle.Users().ListEmails(c.Req.Context(), c.User.ID)
  225. if err != nil {
  226. c.Errorf(err, "get email addresses")
  227. return
  228. }
  229. c.Data["Emails"] = emails
  230. if c.HasError() {
  231. c.HTML(http.StatusBadRequest, tmplUserSettingsEmail)
  232. return
  233. }
  234. err = database.Handle.Users().AddEmail(c.Req.Context(), c.User.ID, f.Email, !conf.Auth.RequireEmailConfirmation)
  235. if err != nil {
  236. if database.IsErrEmailAlreadyUsed(err) {
  237. c.RenderWithErr(c.Tr("form.email_been_used"), http.StatusUnprocessableEntity, tmplUserSettingsEmail, &f)
  238. } else {
  239. c.Errorf(err, "add email address")
  240. }
  241. return
  242. }
  243. // Send confirmation email
  244. if conf.Auth.RequireEmailConfirmation {
  245. email.SendActivateEmailMail(c.Context, database.NewMailerUser(c.User), f.Email)
  246. if err := c.Cache.Put("MailResendLimit_"+c.User.LowerName, c.User.LowerName, 180); err != nil {
  247. log.Error("Set cache 'MailResendLimit' failed: %v", err)
  248. }
  249. c.Flash.Info(c.Tr("settings.add_email_confirmation_sent", f.Email, conf.Auth.ActivateCodeLives/60))
  250. } else {
  251. c.Flash.Success(c.Tr("settings.add_email_success"))
  252. }
  253. c.RedirectSubpath("/user/settings/email")
  254. }
  255. func DeleteEmail(c *context.Context) {
  256. email := c.Query("id") // The "id" here is the actual email address
  257. if c.User.Email == email {
  258. c.Flash.Error(c.Tr("settings.email_deletion_primary"))
  259. c.JSONSuccess(map[string]any{
  260. "redirect": conf.Server.Subpath + "/user/settings/email",
  261. })
  262. return
  263. }
  264. err := database.Handle.Users().DeleteEmail(c.Req.Context(), c.User.ID, email)
  265. if err != nil {
  266. c.Error(err, "delete email address")
  267. return
  268. }
  269. c.Flash.Success(c.Tr("settings.email_deletion_success"))
  270. c.JSONSuccess(map[string]any{
  271. "redirect": conf.Server.Subpath + "/user/settings/email",
  272. })
  273. }
  274. func SettingsSSHKeys(c *context.Context) {
  275. c.Title("settings.ssh_keys")
  276. c.PageIs("SettingsSSHKeys")
  277. keys, err := database.ListPublicKeys(c.User.ID)
  278. if err != nil {
  279. c.Errorf(err, "list public keys")
  280. return
  281. }
  282. c.Data["Keys"] = keys
  283. c.Success(tmplUserSettingsSSHKeys)
  284. }
  285. func SettingsSSHKeysPost(c *context.Context, f form.AddSSHKey) {
  286. c.Title("settings.ssh_keys")
  287. c.PageIs("SettingsSSHKeys")
  288. keys, err := database.ListPublicKeys(c.User.ID)
  289. if err != nil {
  290. c.Errorf(err, "list public keys")
  291. return
  292. }
  293. c.Data["Keys"] = keys
  294. if c.HasError() {
  295. c.HTML(http.StatusBadRequest, tmplUserSettingsSSHKeys)
  296. return
  297. }
  298. content, err := database.CheckPublicKeyString(f.Content)
  299. if err != nil {
  300. if database.IsErrKeyUnableVerify(err) {
  301. c.Flash.Info(c.Tr("form.unable_verify_ssh_key"))
  302. } else {
  303. c.Flash.Error(c.Tr("form.invalid_ssh_key", err.Error()))
  304. c.RedirectSubpath("/user/settings/ssh")
  305. return
  306. }
  307. }
  308. if _, err = database.AddPublicKey(c.User.ID, f.Title, content); err != nil {
  309. c.Data["HasError"] = true
  310. switch {
  311. case database.IsErrKeyAlreadyExist(err):
  312. c.FormErr("Content")
  313. c.RenderWithErr(c.Tr("settings.ssh_key_been_used"), http.StatusUnprocessableEntity, tmplUserSettingsSSHKeys, &f)
  314. case database.IsErrKeyNameAlreadyUsed(err):
  315. c.FormErr("Title")
  316. c.RenderWithErr(c.Tr("settings.ssh_key_name_used"), http.StatusUnprocessableEntity, tmplUserSettingsSSHKeys, &f)
  317. default:
  318. c.Errorf(err, "add public key")
  319. }
  320. return
  321. }
  322. c.Flash.Success(c.Tr("settings.add_key_success", f.Title))
  323. c.RedirectSubpath("/user/settings/ssh")
  324. }
  325. func DeleteSSHKey(c *context.Context) {
  326. if err := database.DeletePublicKey(c.User, c.QueryInt64("id")); err != nil {
  327. c.Flash.Error("DeletePublicKey: " + err.Error())
  328. } else {
  329. c.Flash.Success(c.Tr("settings.ssh_key_deletion_success"))
  330. }
  331. c.JSONSuccess(map[string]any{
  332. "redirect": conf.Server.Subpath + "/user/settings/ssh",
  333. })
  334. }
  335. func SettingsSecurity(c *context.Context) {
  336. c.Title("settings.security")
  337. c.PageIs("SettingsSecurity")
  338. t, err := database.Handle.TwoFactors().GetByUserID(c.Req.Context(), c.UserID())
  339. if err != nil && !database.IsErrTwoFactorNotFound(err) {
  340. c.Errorf(err, "get two factor by user ID")
  341. return
  342. }
  343. c.Data["TwoFactor"] = t
  344. c.Success(tmplUserSettingsSecurity)
  345. }
  346. func SettingsTwoFactorEnable(c *context.Context) {
  347. if database.Handle.TwoFactors().IsEnabled(c.Req.Context(), c.User.ID) {
  348. c.NotFound()
  349. return
  350. }
  351. c.Title("settings.two_factor_enable_title")
  352. c.PageIs("SettingsSecurity")
  353. var key *otp.Key
  354. var err error
  355. keyURL := c.Session.Get("twoFactorURL")
  356. if keyURL != nil {
  357. key, _ = otp.NewKeyFromURL(keyURL.(string))
  358. }
  359. if key == nil {
  360. key, err = totp.Generate(totp.GenerateOpts{
  361. Issuer: conf.App.BrandName,
  362. AccountName: c.User.Email,
  363. })
  364. if err != nil {
  365. c.Errorf(err, "generate TOTP")
  366. return
  367. }
  368. }
  369. c.Data["TwoFactorSecret"] = key.Secret()
  370. img, err := key.Image(240, 240)
  371. if err != nil {
  372. c.Errorf(err, "generate image")
  373. return
  374. }
  375. var buf bytes.Buffer
  376. if err = png.Encode(&buf, img); err != nil {
  377. c.Errorf(err, "encode image")
  378. return
  379. }
  380. c.Data["QRCode"] = template.URL("data:image/png;base64," + base64.StdEncoding.EncodeToString(buf.Bytes()))
  381. _ = c.Session.Set("twoFactorSecret", c.Data["TwoFactorSecret"])
  382. _ = c.Session.Set("twoFactorURL", key.String())
  383. c.Success(tmplUserSettingsTwoFactorEnable)
  384. }
  385. func SettingsTwoFactorEnablePost(c *context.Context) {
  386. secret, ok := c.Session.Get("twoFactorSecret").(string)
  387. if !ok {
  388. c.NotFound()
  389. return
  390. }
  391. if !totp.Validate(c.Query("passcode"), secret) {
  392. c.Flash.Error(c.Tr("settings.two_factor_invalid_passcode"))
  393. c.RedirectSubpath("/user/settings/security/two_factor_enable")
  394. return
  395. }
  396. if err := database.Handle.TwoFactors().Create(c.Req.Context(), c.UserID(), conf.Security.SecretKey, secret); err != nil {
  397. c.Flash.Error(c.Tr("settings.two_factor_enable_error", err))
  398. c.RedirectSubpath("/user/settings/security/two_factor_enable")
  399. return
  400. }
  401. _ = c.Session.Delete("twoFactorSecret")
  402. _ = c.Session.Delete("twoFactorURL")
  403. c.Flash.Success(c.Tr("settings.two_factor_enable_success"))
  404. c.RedirectSubpath("/user/settings/security/two_factor_recovery_codes")
  405. }
  406. func SettingsTwoFactorRecoveryCodes(c *context.Context) {
  407. if !database.Handle.TwoFactors().IsEnabled(c.Req.Context(), c.User.ID) {
  408. c.NotFound()
  409. return
  410. }
  411. c.Title("settings.two_factor_recovery_codes_title")
  412. c.PageIs("SettingsSecurity")
  413. recoveryCodes, err := database.GetRecoveryCodesByUserID(c.UserID())
  414. if err != nil {
  415. c.Errorf(err, "get recovery codes by user ID")
  416. return
  417. }
  418. c.Data["RecoveryCodes"] = recoveryCodes
  419. c.Success(tmplUserSettingsTwoFactorRecoveryCodes)
  420. }
  421. func SettingsTwoFactorRecoveryCodesPost(c *context.Context) {
  422. if !database.Handle.TwoFactors().IsEnabled(c.Req.Context(), c.User.ID) {
  423. c.NotFound()
  424. return
  425. }
  426. if err := database.RegenerateRecoveryCodes(c.UserID()); err != nil {
  427. c.Flash.Error(c.Tr("settings.two_factor_regenerate_recovery_codes_error", err))
  428. } else {
  429. c.Flash.Success(c.Tr("settings.two_factor_regenerate_recovery_codes_success"))
  430. }
  431. c.RedirectSubpath("/user/settings/security/two_factor_recovery_codes")
  432. }
  433. func SettingsTwoFactorDisable(c *context.Context) {
  434. if !database.Handle.TwoFactors().IsEnabled(c.Req.Context(), c.User.ID) {
  435. c.NotFound()
  436. return
  437. }
  438. if err := database.DeleteTwoFactor(c.UserID()); err != nil {
  439. c.Errorf(err, "delete two factor")
  440. return
  441. }
  442. c.Flash.Success(c.Tr("settings.two_factor_disable_success"))
  443. c.JSONSuccess(map[string]any{
  444. "redirect": conf.Server.Subpath + "/user/settings/security",
  445. })
  446. }
  447. func SettingsRepos(c *context.Context) {
  448. c.Title("settings.repos")
  449. c.PageIs("SettingsRepositories")
  450. repos, err := database.GetUserAndCollaborativeRepositories(c.User.ID)
  451. if err != nil {
  452. c.Errorf(err, "get user and collaborative repositories")
  453. return
  454. }
  455. if err = database.RepositoryList(repos).LoadAttributes(); err != nil {
  456. c.Errorf(err, "load attributes")
  457. return
  458. }
  459. c.Data["Repos"] = repos
  460. c.Success(tmplUserSettingsRepositories)
  461. }
  462. func SettingsLeaveRepo(c *context.Context) {
  463. repo, err := database.GetRepositoryByID(c.QueryInt64("id"))
  464. if err != nil {
  465. c.NotFoundOrError(err, "get repository by ID")
  466. return
  467. }
  468. if err = repo.DeleteCollaboration(c.User.ID); err != nil {
  469. c.Errorf(err, "delete collaboration")
  470. return
  471. }
  472. c.Flash.Success(c.Tr("settings.repos.leave_success", repo.FullName()))
  473. c.JSONSuccess(map[string]any{
  474. "redirect": conf.Server.Subpath + "/user/settings/repositories",
  475. })
  476. }
  477. func SettingsOrganizations(c *context.Context) {
  478. c.Title("settings.orgs")
  479. c.PageIs("SettingsOrganizations")
  480. orgs, err := database.GetOrgsByUserID(c.User.ID, true)
  481. if err != nil {
  482. c.Errorf(err, "get organizations by user ID")
  483. return
  484. }
  485. c.Data["Orgs"] = orgs
  486. c.Success(tmplUserSettingsOrganizations)
  487. }
  488. func SettingsLeaveOrganization(c *context.Context) {
  489. if err := database.RemoveOrgUser(c.QueryInt64("id"), c.User.ID); err != nil {
  490. if database.IsErrLastOrgOwner(err) {
  491. c.Flash.Error(c.Tr("form.last_org_owner"))
  492. } else {
  493. c.Errorf(err, "remove organization user")
  494. return
  495. }
  496. }
  497. c.JSONSuccess(map[string]any{
  498. "redirect": conf.Server.Subpath + "/user/settings/organizations",
  499. })
  500. }
  501. func (h *SettingsHandler) Applications() macaron.Handler {
  502. return func(c *context.Context) {
  503. c.Title("settings.applications")
  504. c.PageIs("SettingsApplications")
  505. tokens, err := h.store.ListAccessTokens(c.Req.Context(), c.User.ID)
  506. if err != nil {
  507. c.Errorf(err, "list access tokens")
  508. return
  509. }
  510. c.Data["Tokens"] = tokens
  511. c.Success(tmplUserSettingsApplications)
  512. }
  513. }
  514. func (h *SettingsHandler) ApplicationsPost() macaron.Handler {
  515. return func(c *context.Context, f form.NewAccessToken) {
  516. c.Title("settings.applications")
  517. c.PageIs("SettingsApplications")
  518. if c.HasError() {
  519. tokens, err := h.store.ListAccessTokens(c.Req.Context(), c.User.ID)
  520. if err != nil {
  521. c.Errorf(err, "list access tokens")
  522. return
  523. }
  524. c.Data["Tokens"] = tokens
  525. c.HTML(http.StatusBadRequest, tmplUserSettingsApplications)
  526. return
  527. }
  528. t, err := h.store.CreateAccessToken(c.Req.Context(), c.User.ID, f.Name)
  529. if err != nil {
  530. if database.IsErrAccessTokenAlreadyExist(err) {
  531. c.Flash.Error(c.Tr("settings.token_name_exists"))
  532. c.RedirectSubpath("/user/settings/applications")
  533. } else {
  534. c.Errorf(err, "new access token")
  535. }
  536. return
  537. }
  538. c.Flash.Success(c.Tr("settings.generate_token_succees"))
  539. c.Flash.Info(t.Sha1)
  540. c.RedirectSubpath("/user/settings/applications")
  541. }
  542. }
  543. func (h *SettingsHandler) DeleteApplication() macaron.Handler {
  544. return func(c *context.Context) {
  545. if err := h.store.DeleteAccessTokenByID(c.Req.Context(), c.User.ID, c.QueryInt64("id")); err != nil {
  546. c.Flash.Error("DeleteAccessTokenByID: " + err.Error())
  547. } else {
  548. c.Flash.Success(c.Tr("settings.delete_token_success"))
  549. }
  550. c.JSONSuccess(map[string]any{
  551. "redirect": conf.Server.Subpath + "/user/settings/applications",
  552. })
  553. }
  554. }
  555. func SettingsDelete(c *context.Context) {
  556. c.Title("settings.delete")
  557. c.PageIs("SettingsDelete")
  558. if c.Req.Method == "POST" {
  559. if _, err := database.Handle.Users().Authenticate(c.Req.Context(), c.User.Name, c.Query("password"), c.User.LoginSource); err != nil {
  560. if auth.IsErrBadCredentials(err) {
  561. c.RenderWithErr(c.Tr("form.enterred_invalid_password"), http.StatusUnauthorized, tmplUserSettingsDelete, nil)
  562. } else {
  563. c.Errorf(err, "authenticate user")
  564. }
  565. return
  566. }
  567. if err := database.Handle.Users().DeleteByID(c.Req.Context(), c.User.ID, false); err != nil {
  568. switch {
  569. case database.IsErrUserOwnRepos(err):
  570. c.Flash.Error(c.Tr("form.still_own_repo"))
  571. c.Redirect(conf.Server.Subpath + "/user/settings/delete")
  572. case database.IsErrUserHasOrgs(err):
  573. c.Flash.Error(c.Tr("form.still_has_org"))
  574. c.Redirect(conf.Server.Subpath + "/user/settings/delete")
  575. default:
  576. c.Errorf(err, "delete user")
  577. }
  578. } else {
  579. log.Trace("Account deleted: %s", c.User.Name)
  580. c.Redirect(conf.Server.Subpath + "/")
  581. }
  582. return
  583. }
  584. c.Success(tmplUserSettingsDelete)
  585. }
  586. // SettingsStore is the data layer carrier for user settings endpoints. This
  587. // interface is meant to abstract away and limit the exposure of the underlying
  588. // data layer to the handler through a thin-wrapper.
  589. type SettingsStore interface {
  590. // CreateAccessToken creates a new access token and persist to database. It
  591. // returns database.ErrAccessTokenAlreadyExist when an access token with same
  592. // name already exists for the user.
  593. CreateAccessToken(ctx gocontext.Context, userID int64, name string) (*database.AccessToken, error)
  594. // GetAccessTokenBySHA1 returns the access token with given SHA1. It returns
  595. // database.ErrAccessTokenNotExist when not found.
  596. GetAccessTokenBySHA1(ctx gocontext.Context, sha1 string) (*database.AccessToken, error)
  597. // TouchAccessTokenByID updates the updated time of the given access token to
  598. // the current time.
  599. TouchAccessTokenByID(ctx gocontext.Context, id int64) error
  600. // ListAccessTokens returns all access tokens belongs to given user.
  601. ListAccessTokens(ctx gocontext.Context, userID int64) ([]*database.AccessToken, error)
  602. // DeleteAccessTokenByID deletes the access token by given ID.
  603. DeleteAccessTokenByID(ctx gocontext.Context, userID, id int64) error
  604. }
  605. type settingsStore struct{}
  606. // NewSettingsStore returns a new SettingsStore using the global database
  607. // handle.
  608. func NewSettingsStore() SettingsStore {
  609. return &settingsStore{}
  610. }
  611. func (*settingsStore) CreateAccessToken(ctx gocontext.Context, userID int64, name string) (*database.AccessToken, error) {
  612. return database.Handle.AccessTokens().Create(ctx, userID, name)
  613. }
  614. func (*settingsStore) GetAccessTokenBySHA1(ctx gocontext.Context, sha1 string) (*database.AccessToken, error) {
  615. return database.Handle.AccessTokens().GetBySHA1(ctx, sha1)
  616. }
  617. func (*settingsStore) TouchAccessTokenByID(ctx gocontext.Context, id int64) error {
  618. return database.Handle.AccessTokens().Touch(ctx, id)
  619. }
  620. func (*settingsStore) ListAccessTokens(ctx gocontext.Context, userID int64) ([]*database.AccessToken, error) {
  621. return database.Handle.AccessTokens().List(ctx, userID)
  622. }
  623. func (*settingsStore) DeleteAccessTokenByID(ctx gocontext.Context, userID, id int64) error {
  624. return database.Handle.AccessTokens().DeleteByID(ctx, userID, id)
  625. }