setting.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737
  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. if err := email.SendActivateEmailMail(c.Context, database.NewMailerUser(c.User), f.Email); err != nil {
  246. log.Error("Failed to send activate email mail: %v", err)
  247. }
  248. if err := c.Cache.Put("MailResendLimit_"+c.User.LowerName, c.User.LowerName, 180); err != nil {
  249. log.Error("Set cache 'MailResendLimit' failed: %v", err)
  250. }
  251. c.Flash.Info(c.Tr("settings.add_email_confirmation_sent", f.Email, conf.Auth.ActivateCodeLives/60))
  252. } else {
  253. c.Flash.Success(c.Tr("settings.add_email_success"))
  254. }
  255. c.RedirectSubpath("/user/settings/email")
  256. }
  257. func DeleteEmail(c *context.Context) {
  258. email := c.Query("id") // The "id" here is the actual email address
  259. if c.User.Email == email {
  260. c.Flash.Error(c.Tr("settings.email_deletion_primary"))
  261. c.JSONSuccess(map[string]any{
  262. "redirect": conf.Server.Subpath + "/user/settings/email",
  263. })
  264. return
  265. }
  266. err := database.Handle.Users().DeleteEmail(c.Req.Context(), c.User.ID, email)
  267. if err != nil {
  268. c.Error(err, "delete email address")
  269. return
  270. }
  271. c.Flash.Success(c.Tr("settings.email_deletion_success"))
  272. c.JSONSuccess(map[string]any{
  273. "redirect": conf.Server.Subpath + "/user/settings/email",
  274. })
  275. }
  276. func SettingsSSHKeys(c *context.Context) {
  277. c.Title("settings.ssh_keys")
  278. c.PageIs("SettingsSSHKeys")
  279. keys, err := database.ListPublicKeys(c.User.ID)
  280. if err != nil {
  281. c.Errorf(err, "list public keys")
  282. return
  283. }
  284. c.Data["Keys"] = keys
  285. c.Success(tmplUserSettingsSSHKeys)
  286. }
  287. func SettingsSSHKeysPost(c *context.Context, f form.AddSSHKey) {
  288. c.Title("settings.ssh_keys")
  289. c.PageIs("SettingsSSHKeys")
  290. keys, err := database.ListPublicKeys(c.User.ID)
  291. if err != nil {
  292. c.Errorf(err, "list public keys")
  293. return
  294. }
  295. c.Data["Keys"] = keys
  296. if c.HasError() {
  297. c.HTML(http.StatusBadRequest, tmplUserSettingsSSHKeys)
  298. return
  299. }
  300. content, err := database.CheckPublicKeyString(f.Content)
  301. if err != nil {
  302. if database.IsErrKeyUnableVerify(err) {
  303. c.Flash.Info(c.Tr("form.unable_verify_ssh_key"))
  304. } else {
  305. c.Flash.Error(c.Tr("form.invalid_ssh_key", err.Error()))
  306. c.RedirectSubpath("/user/settings/ssh")
  307. return
  308. }
  309. }
  310. if _, err = database.AddPublicKey(c.User.ID, f.Title, content); err != nil {
  311. c.Data["HasError"] = true
  312. switch {
  313. case database.IsErrKeyAlreadyExist(err):
  314. c.FormErr("Content")
  315. c.RenderWithErr(c.Tr("settings.ssh_key_been_used"), http.StatusUnprocessableEntity, tmplUserSettingsSSHKeys, &f)
  316. case database.IsErrKeyNameAlreadyUsed(err):
  317. c.FormErr("Title")
  318. c.RenderWithErr(c.Tr("settings.ssh_key_name_used"), http.StatusUnprocessableEntity, tmplUserSettingsSSHKeys, &f)
  319. default:
  320. c.Errorf(err, "add public key")
  321. }
  322. return
  323. }
  324. c.Flash.Success(c.Tr("settings.add_key_success", f.Title))
  325. c.RedirectSubpath("/user/settings/ssh")
  326. }
  327. func DeleteSSHKey(c *context.Context) {
  328. if err := database.DeletePublicKey(c.User, c.QueryInt64("id")); err != nil {
  329. c.Flash.Error("DeletePublicKey: " + err.Error())
  330. } else {
  331. c.Flash.Success(c.Tr("settings.ssh_key_deletion_success"))
  332. }
  333. c.JSONSuccess(map[string]any{
  334. "redirect": conf.Server.Subpath + "/user/settings/ssh",
  335. })
  336. }
  337. func SettingsSecurity(c *context.Context) {
  338. c.Title("settings.security")
  339. c.PageIs("SettingsSecurity")
  340. t, err := database.Handle.TwoFactors().GetByUserID(c.Req.Context(), c.UserID())
  341. if err != nil && !database.IsErrTwoFactorNotFound(err) {
  342. c.Errorf(err, "get two factor by user ID")
  343. return
  344. }
  345. c.Data["TwoFactor"] = t
  346. c.Success(tmplUserSettingsSecurity)
  347. }
  348. func SettingsTwoFactorEnable(c *context.Context) {
  349. if database.Handle.TwoFactors().IsEnabled(c.Req.Context(), c.User.ID) {
  350. c.NotFound()
  351. return
  352. }
  353. c.Title("settings.two_factor_enable_title")
  354. c.PageIs("SettingsSecurity")
  355. var key *otp.Key
  356. var err error
  357. keyURL := c.Session.Get("twoFactorURL")
  358. if keyURL != nil {
  359. key, _ = otp.NewKeyFromURL(keyURL.(string))
  360. }
  361. if key == nil {
  362. key, err = totp.Generate(totp.GenerateOpts{
  363. Issuer: conf.App.BrandName,
  364. AccountName: c.User.Email,
  365. })
  366. if err != nil {
  367. c.Errorf(err, "generate TOTP")
  368. return
  369. }
  370. }
  371. c.Data["TwoFactorSecret"] = key.Secret()
  372. img, err := key.Image(240, 240)
  373. if err != nil {
  374. c.Errorf(err, "generate image")
  375. return
  376. }
  377. var buf bytes.Buffer
  378. if err = png.Encode(&buf, img); err != nil {
  379. c.Errorf(err, "encode image")
  380. return
  381. }
  382. c.Data["QRCode"] = template.URL("data:image/png;base64," + base64.StdEncoding.EncodeToString(buf.Bytes()))
  383. _ = c.Session.Set("twoFactorSecret", c.Data["TwoFactorSecret"])
  384. _ = c.Session.Set("twoFactorURL", key.String())
  385. c.Success(tmplUserSettingsTwoFactorEnable)
  386. }
  387. func SettingsTwoFactorEnablePost(c *context.Context) {
  388. secret, ok := c.Session.Get("twoFactorSecret").(string)
  389. if !ok {
  390. c.NotFound()
  391. return
  392. }
  393. if !totp.Validate(c.Query("passcode"), secret) {
  394. c.Flash.Error(c.Tr("settings.two_factor_invalid_passcode"))
  395. c.RedirectSubpath("/user/settings/security/two_factor_enable")
  396. return
  397. }
  398. if err := database.Handle.TwoFactors().Create(c.Req.Context(), c.UserID(), conf.Security.SecretKey, secret); err != nil {
  399. c.Flash.Error(c.Tr("settings.two_factor_enable_error", err))
  400. c.RedirectSubpath("/user/settings/security/two_factor_enable")
  401. return
  402. }
  403. _ = c.Session.Delete("twoFactorSecret")
  404. _ = c.Session.Delete("twoFactorURL")
  405. c.Flash.Success(c.Tr("settings.two_factor_enable_success"))
  406. c.RedirectSubpath("/user/settings/security/two_factor_recovery_codes")
  407. }
  408. func SettingsTwoFactorRecoveryCodes(c *context.Context) {
  409. if !database.Handle.TwoFactors().IsEnabled(c.Req.Context(), c.User.ID) {
  410. c.NotFound()
  411. return
  412. }
  413. c.Title("settings.two_factor_recovery_codes_title")
  414. c.PageIs("SettingsSecurity")
  415. recoveryCodes, err := database.GetRecoveryCodesByUserID(c.UserID())
  416. if err != nil {
  417. c.Errorf(err, "get recovery codes by user ID")
  418. return
  419. }
  420. c.Data["RecoveryCodes"] = recoveryCodes
  421. c.Success(tmplUserSettingsTwoFactorRecoveryCodes)
  422. }
  423. func SettingsTwoFactorRecoveryCodesPost(c *context.Context) {
  424. if !database.Handle.TwoFactors().IsEnabled(c.Req.Context(), c.User.ID) {
  425. c.NotFound()
  426. return
  427. }
  428. if err := database.RegenerateRecoveryCodes(c.UserID()); err != nil {
  429. c.Flash.Error(c.Tr("settings.two_factor_regenerate_recovery_codes_error", err))
  430. } else {
  431. c.Flash.Success(c.Tr("settings.two_factor_regenerate_recovery_codes_success"))
  432. }
  433. c.RedirectSubpath("/user/settings/security/two_factor_recovery_codes")
  434. }
  435. func SettingsTwoFactorDisable(c *context.Context) {
  436. if !database.Handle.TwoFactors().IsEnabled(c.Req.Context(), c.User.ID) {
  437. c.NotFound()
  438. return
  439. }
  440. if err := database.DeleteTwoFactor(c.UserID()); err != nil {
  441. c.Errorf(err, "delete two factor")
  442. return
  443. }
  444. c.Flash.Success(c.Tr("settings.two_factor_disable_success"))
  445. c.JSONSuccess(map[string]any{
  446. "redirect": conf.Server.Subpath + "/user/settings/security",
  447. })
  448. }
  449. func SettingsRepos(c *context.Context) {
  450. c.Title("settings.repos")
  451. c.PageIs("SettingsRepositories")
  452. repos, err := database.GetUserAndCollaborativeRepositories(c.User.ID)
  453. if err != nil {
  454. c.Errorf(err, "get user and collaborative repositories")
  455. return
  456. }
  457. if err = database.RepositoryList(repos).LoadAttributes(); err != nil {
  458. c.Errorf(err, "load attributes")
  459. return
  460. }
  461. c.Data["Repos"] = repos
  462. c.Success(tmplUserSettingsRepositories)
  463. }
  464. func SettingsLeaveRepo(c *context.Context) {
  465. repo, err := database.GetRepositoryByID(c.QueryInt64("id"))
  466. if err != nil {
  467. c.NotFoundOrError(err, "get repository by ID")
  468. return
  469. }
  470. if err = repo.DeleteCollaboration(c.User.ID); err != nil {
  471. c.Errorf(err, "delete collaboration")
  472. return
  473. }
  474. c.Flash.Success(c.Tr("settings.repos.leave_success", repo.FullName()))
  475. c.JSONSuccess(map[string]any{
  476. "redirect": conf.Server.Subpath + "/user/settings/repositories",
  477. })
  478. }
  479. func SettingsOrganizations(c *context.Context) {
  480. c.Title("settings.orgs")
  481. c.PageIs("SettingsOrganizations")
  482. orgs, err := database.GetOrgsByUserID(c.User.ID, true)
  483. if err != nil {
  484. c.Errorf(err, "get organizations by user ID")
  485. return
  486. }
  487. c.Data["Orgs"] = orgs
  488. c.Success(tmplUserSettingsOrganizations)
  489. }
  490. func SettingsLeaveOrganization(c *context.Context) {
  491. if err := database.RemoveOrgUser(c.QueryInt64("id"), c.User.ID); err != nil {
  492. if database.IsErrLastOrgOwner(err) {
  493. c.Flash.Error(c.Tr("form.last_org_owner"))
  494. } else {
  495. c.Errorf(err, "remove organization user")
  496. return
  497. }
  498. }
  499. c.JSONSuccess(map[string]any{
  500. "redirect": conf.Server.Subpath + "/user/settings/organizations",
  501. })
  502. }
  503. func (h *SettingsHandler) Applications() macaron.Handler {
  504. return func(c *context.Context) {
  505. c.Title("settings.applications")
  506. c.PageIs("SettingsApplications")
  507. tokens, err := h.store.ListAccessTokens(c.Req.Context(), c.User.ID)
  508. if err != nil {
  509. c.Errorf(err, "list access tokens")
  510. return
  511. }
  512. c.Data["Tokens"] = tokens
  513. c.Success(tmplUserSettingsApplications)
  514. }
  515. }
  516. func (h *SettingsHandler) ApplicationsPost() macaron.Handler {
  517. return func(c *context.Context, f form.NewAccessToken) {
  518. c.Title("settings.applications")
  519. c.PageIs("SettingsApplications")
  520. if c.HasError() {
  521. tokens, err := h.store.ListAccessTokens(c.Req.Context(), c.User.ID)
  522. if err != nil {
  523. c.Errorf(err, "list access tokens")
  524. return
  525. }
  526. c.Data["Tokens"] = tokens
  527. c.HTML(http.StatusBadRequest, tmplUserSettingsApplications)
  528. return
  529. }
  530. t, err := h.store.CreateAccessToken(c.Req.Context(), c.User.ID, f.Name)
  531. if err != nil {
  532. if database.IsErrAccessTokenAlreadyExist(err) {
  533. c.Flash.Error(c.Tr("settings.token_name_exists"))
  534. c.RedirectSubpath("/user/settings/applications")
  535. } else {
  536. c.Errorf(err, "new access token")
  537. }
  538. return
  539. }
  540. c.Flash.Success(c.Tr("settings.generate_token_succees"))
  541. c.Flash.Info(t.Sha1)
  542. c.RedirectSubpath("/user/settings/applications")
  543. }
  544. }
  545. func (h *SettingsHandler) DeleteApplication() macaron.Handler {
  546. return func(c *context.Context) {
  547. if err := h.store.DeleteAccessTokenByID(c.Req.Context(), c.User.ID, c.QueryInt64("id")); err != nil {
  548. c.Flash.Error("DeleteAccessTokenByID: " + err.Error())
  549. } else {
  550. c.Flash.Success(c.Tr("settings.delete_token_success"))
  551. }
  552. c.JSONSuccess(map[string]any{
  553. "redirect": conf.Server.Subpath + "/user/settings/applications",
  554. })
  555. }
  556. }
  557. func SettingsDelete(c *context.Context) {
  558. c.Title("settings.delete")
  559. c.PageIs("SettingsDelete")
  560. if c.Req.Method == "POST" {
  561. if _, err := database.Handle.Users().Authenticate(c.Req.Context(), c.User.Name, c.Query("password"), c.User.LoginSource); err != nil {
  562. if auth.IsErrBadCredentials(err) {
  563. c.RenderWithErr(c.Tr("form.enterred_invalid_password"), http.StatusUnauthorized, tmplUserSettingsDelete, nil)
  564. } else {
  565. c.Errorf(err, "authenticate user")
  566. }
  567. return
  568. }
  569. if err := database.Handle.Users().DeleteByID(c.Req.Context(), c.User.ID, false); err != nil {
  570. switch {
  571. case database.IsErrUserOwnRepos(err):
  572. c.Flash.Error(c.Tr("form.still_own_repo"))
  573. c.Redirect(conf.Server.Subpath + "/user/settings/delete")
  574. case database.IsErrUserHasOrgs(err):
  575. c.Flash.Error(c.Tr("form.still_has_org"))
  576. c.Redirect(conf.Server.Subpath + "/user/settings/delete")
  577. default:
  578. c.Errorf(err, "delete user")
  579. }
  580. } else {
  581. log.Trace("Account deleted: %s", c.User.Name)
  582. c.Redirect(conf.Server.Subpath + "/")
  583. }
  584. return
  585. }
  586. c.Success(tmplUserSettingsDelete)
  587. }
  588. // SettingsStore is the data layer carrier for user settings endpoints. This
  589. // interface is meant to abstract away and limit the exposure of the underlying
  590. // data layer to the handler through a thin-wrapper.
  591. type SettingsStore interface {
  592. // CreateAccessToken creates a new access token and persist to database. It
  593. // returns database.ErrAccessTokenAlreadyExist when an access token with same
  594. // name already exists for the user.
  595. CreateAccessToken(ctx gocontext.Context, userID int64, name string) (*database.AccessToken, error)
  596. // GetAccessTokenBySHA1 returns the access token with given SHA1. It returns
  597. // database.ErrAccessTokenNotExist when not found.
  598. GetAccessTokenBySHA1(ctx gocontext.Context, sha1 string) (*database.AccessToken, error)
  599. // TouchAccessTokenByID updates the updated time of the given access token to
  600. // the current time.
  601. TouchAccessTokenByID(ctx gocontext.Context, id int64) error
  602. // ListAccessTokens returns all access tokens belongs to given user.
  603. ListAccessTokens(ctx gocontext.Context, userID int64) ([]*database.AccessToken, error)
  604. // DeleteAccessTokenByID deletes the access token by given ID.
  605. DeleteAccessTokenByID(ctx gocontext.Context, userID, id int64) error
  606. }
  607. type settingsStore struct{}
  608. // NewSettingsStore returns a new SettingsStore using the global database
  609. // handle.
  610. func NewSettingsStore() SettingsStore {
  611. return &settingsStore{}
  612. }
  613. func (*settingsStore) CreateAccessToken(ctx gocontext.Context, userID int64, name string) (*database.AccessToken, error) {
  614. return database.Handle.AccessTokens().Create(ctx, userID, name)
  615. }
  616. func (*settingsStore) GetAccessTokenBySHA1(ctx gocontext.Context, sha1 string) (*database.AccessToken, error) {
  617. return database.Handle.AccessTokens().GetBySHA1(ctx, sha1)
  618. }
  619. func (*settingsStore) TouchAccessTokenByID(ctx gocontext.Context, id int64) error {
  620. return database.Handle.AccessTokens().Touch(ctx, id)
  621. }
  622. func (*settingsStore) ListAccessTokens(ctx gocontext.Context, userID int64) ([]*database.AccessToken, error) {
  623. return database.Handle.AccessTokens().List(ctx, userID)
  624. }
  625. func (*settingsStore) DeleteAccessTokenByID(ctx gocontext.Context, userID, id int64) error {
  626. return database.Handle.AccessTokens().DeleteByID(ctx, userID, id)
  627. }