You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

repo_transfer.go 12KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411
  1. // Copyright 2021 The Gitea Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package models
  5. import (
  6. "fmt"
  7. "os"
  8. "code.gitea.io/gitea/models/db"
  9. "code.gitea.io/gitea/modules/log"
  10. "code.gitea.io/gitea/modules/timeutil"
  11. "code.gitea.io/gitea/modules/util"
  12. )
  13. // RepoTransfer is used to manage repository transfers
  14. type RepoTransfer struct {
  15. ID int64 `xorm:"pk autoincr"`
  16. DoerID int64
  17. Doer *User `xorm:"-"`
  18. RecipientID int64
  19. Recipient *User `xorm:"-"`
  20. RepoID int64
  21. TeamIDs []int64
  22. Teams []*Team `xorm:"-"`
  23. CreatedUnix timeutil.TimeStamp `xorm:"INDEX NOT NULL created"`
  24. UpdatedUnix timeutil.TimeStamp `xorm:"INDEX NOT NULL updated"`
  25. }
  26. func init() {
  27. db.RegisterModel(new(RepoTransfer))
  28. }
  29. // LoadAttributes fetches the transfer recipient from the database
  30. func (r *RepoTransfer) LoadAttributes() error {
  31. if r.Recipient == nil {
  32. u, err := GetUserByID(r.RecipientID)
  33. if err != nil {
  34. return err
  35. }
  36. r.Recipient = u
  37. }
  38. if r.Recipient.IsOrganization() && len(r.TeamIDs) != len(r.Teams) {
  39. for _, v := range r.TeamIDs {
  40. team, err := GetTeamByID(v)
  41. if err != nil {
  42. return err
  43. }
  44. if team.OrgID != r.Recipient.ID {
  45. return fmt.Errorf("team %d belongs not to org %d", v, r.Recipient.ID)
  46. }
  47. r.Teams = append(r.Teams, team)
  48. }
  49. }
  50. if r.Doer == nil {
  51. u, err := GetUserByID(r.DoerID)
  52. if err != nil {
  53. return err
  54. }
  55. r.Doer = u
  56. }
  57. return nil
  58. }
  59. // CanUserAcceptTransfer checks if the user has the rights to accept/decline a repo transfer.
  60. // For user, it checks if it's himself
  61. // For organizations, it checks if the user is able to create repos
  62. func (r *RepoTransfer) CanUserAcceptTransfer(u *User) bool {
  63. if err := r.LoadAttributes(); err != nil {
  64. log.Error("LoadAttributes: %v", err)
  65. return false
  66. }
  67. if !r.Recipient.IsOrganization() {
  68. return r.RecipientID == u.ID
  69. }
  70. allowed, err := CanCreateOrgRepo(r.RecipientID, u.ID)
  71. if err != nil {
  72. log.Error("CanCreateOrgRepo: %v", err)
  73. return false
  74. }
  75. return allowed
  76. }
  77. // GetPendingRepositoryTransfer fetches the most recent and ongoing transfer
  78. // process for the repository
  79. func GetPendingRepositoryTransfer(repo *Repository) (*RepoTransfer, error) {
  80. transfer := new(RepoTransfer)
  81. has, err := db.GetEngine(db.DefaultContext).Where("repo_id = ? ", repo.ID).Get(transfer)
  82. if err != nil {
  83. return nil, err
  84. }
  85. if !has {
  86. return nil, ErrNoPendingRepoTransfer{RepoID: repo.ID}
  87. }
  88. return transfer, nil
  89. }
  90. func deleteRepositoryTransfer(e db.Engine, repoID int64) error {
  91. _, err := e.Where("repo_id = ?", repoID).Delete(&RepoTransfer{})
  92. return err
  93. }
  94. // CancelRepositoryTransfer marks the repository as ready and remove pending transfer entry,
  95. // thus cancel the transfer process.
  96. func CancelRepositoryTransfer(repo *Repository) error {
  97. sess := db.NewSession(db.DefaultContext)
  98. defer sess.Close()
  99. if err := sess.Begin(); err != nil {
  100. return err
  101. }
  102. repo.Status = RepositoryReady
  103. if err := updateRepositoryCols(sess, repo, "status"); err != nil {
  104. return err
  105. }
  106. if err := deleteRepositoryTransfer(sess, repo.ID); err != nil {
  107. return err
  108. }
  109. return sess.Commit()
  110. }
  111. // TestRepositoryReadyForTransfer make sure repo is ready to transfer
  112. func TestRepositoryReadyForTransfer(status RepositoryStatus) error {
  113. switch status {
  114. case RepositoryBeingMigrated:
  115. return fmt.Errorf("repo is not ready, currently migrating")
  116. case RepositoryPendingTransfer:
  117. return ErrRepoTransferInProgress{}
  118. }
  119. return nil
  120. }
  121. // CreatePendingRepositoryTransfer transfer a repo from one owner to a new one.
  122. // it marks the repository transfer as "pending"
  123. func CreatePendingRepositoryTransfer(doer, newOwner *User, repoID int64, teams []*Team) error {
  124. sess := db.NewSession(db.DefaultContext)
  125. defer sess.Close()
  126. if err := sess.Begin(); err != nil {
  127. return err
  128. }
  129. repo, err := getRepositoryByID(sess, repoID)
  130. if err != nil {
  131. return err
  132. }
  133. // Make sure repo is ready to transfer
  134. if err := TestRepositoryReadyForTransfer(repo.Status); err != nil {
  135. return err
  136. }
  137. repo.Status = RepositoryPendingTransfer
  138. if err := updateRepositoryCols(sess, repo, "status"); err != nil {
  139. return err
  140. }
  141. // Check if new owner has repository with same name.
  142. if has, err := isRepositoryExist(sess, newOwner, repo.Name); err != nil {
  143. return fmt.Errorf("IsRepositoryExist: %v", err)
  144. } else if has {
  145. return ErrRepoAlreadyExist{newOwner.LowerName, repo.Name}
  146. }
  147. transfer := &RepoTransfer{
  148. RepoID: repo.ID,
  149. RecipientID: newOwner.ID,
  150. CreatedUnix: timeutil.TimeStampNow(),
  151. UpdatedUnix: timeutil.TimeStampNow(),
  152. DoerID: doer.ID,
  153. TeamIDs: make([]int64, 0, len(teams)),
  154. }
  155. for k := range teams {
  156. transfer.TeamIDs = append(transfer.TeamIDs, teams[k].ID)
  157. }
  158. if _, err := sess.Insert(transfer); err != nil {
  159. return err
  160. }
  161. return sess.Commit()
  162. }
  163. // TransferOwnership transfers all corresponding repository items from old user to new one.
  164. func TransferOwnership(doer *User, newOwnerName string, repo *Repository) (err error) {
  165. repoRenamed := false
  166. wikiRenamed := false
  167. oldOwnerName := doer.Name
  168. defer func() {
  169. if !repoRenamed && !wikiRenamed {
  170. return
  171. }
  172. recoverErr := recover()
  173. if err == nil && recoverErr == nil {
  174. return
  175. }
  176. if repoRenamed {
  177. if err := util.Rename(RepoPath(newOwnerName, repo.Name), RepoPath(oldOwnerName, repo.Name)); err != nil {
  178. log.Critical("Unable to move repository %s/%s directory from %s back to correct place %s: %v", oldOwnerName, repo.Name, RepoPath(newOwnerName, repo.Name), RepoPath(oldOwnerName, repo.Name), err)
  179. }
  180. }
  181. if wikiRenamed {
  182. if err := util.Rename(WikiPath(newOwnerName, repo.Name), WikiPath(oldOwnerName, repo.Name)); err != nil {
  183. log.Critical("Unable to move wiki for repository %s/%s directory from %s back to correct place %s: %v", oldOwnerName, repo.Name, WikiPath(newOwnerName, repo.Name), WikiPath(oldOwnerName, repo.Name), err)
  184. }
  185. }
  186. if recoverErr != nil {
  187. log.Error("Panic within TransferOwnership: %v\n%s", recoverErr, log.Stack(2))
  188. panic(recoverErr)
  189. }
  190. }()
  191. sess := db.NewSession(db.DefaultContext)
  192. defer sess.Close()
  193. if err := sess.Begin(); err != nil {
  194. return fmt.Errorf("sess.Begin: %v", err)
  195. }
  196. newOwner, err := getUserByName(sess, newOwnerName)
  197. if err != nil {
  198. return fmt.Errorf("get new owner '%s': %v", newOwnerName, err)
  199. }
  200. newOwnerName = newOwner.Name // ensure capitalisation matches
  201. // Check if new owner has repository with same name.
  202. if has, err := isRepositoryExist(sess, newOwner, repo.Name); err != nil {
  203. return fmt.Errorf("IsRepositoryExist: %v", err)
  204. } else if has {
  205. return ErrRepoAlreadyExist{newOwnerName, repo.Name}
  206. }
  207. oldOwner := repo.Owner
  208. oldOwnerName = oldOwner.Name
  209. // Note: we have to set value here to make sure recalculate accesses is based on
  210. // new owner.
  211. repo.OwnerID = newOwner.ID
  212. repo.Owner = newOwner
  213. repo.OwnerName = newOwner.Name
  214. // Update repository.
  215. if _, err := sess.ID(repo.ID).Update(repo); err != nil {
  216. return fmt.Errorf("update owner: %v", err)
  217. }
  218. // Remove redundant collaborators.
  219. collaborators, err := repo.getCollaborators(sess, db.ListOptions{})
  220. if err != nil {
  221. return fmt.Errorf("getCollaborators: %v", err)
  222. }
  223. // Dummy object.
  224. collaboration := &Collaboration{RepoID: repo.ID}
  225. for _, c := range collaborators {
  226. if c.IsGhost() {
  227. collaboration.ID = c.Collaboration.ID
  228. if _, err := sess.Delete(collaboration); err != nil {
  229. return fmt.Errorf("remove collaborator '%d': %v", c.ID, err)
  230. }
  231. collaboration.ID = 0
  232. }
  233. if c.ID != newOwner.ID {
  234. isMember, err := isOrganizationMember(sess, newOwner.ID, c.ID)
  235. if err != nil {
  236. return fmt.Errorf("IsOrgMember: %v", err)
  237. } else if !isMember {
  238. continue
  239. }
  240. }
  241. collaboration.UserID = c.ID
  242. if _, err := sess.Delete(collaboration); err != nil {
  243. return fmt.Errorf("remove collaborator '%d': %v", c.ID, err)
  244. }
  245. collaboration.UserID = 0
  246. }
  247. // Remove old team-repository relations.
  248. if oldOwner.IsOrganization() {
  249. if err := oldOwner.removeOrgRepo(sess, repo.ID); err != nil {
  250. return fmt.Errorf("removeOrgRepo: %v", err)
  251. }
  252. }
  253. if newOwner.IsOrganization() {
  254. if err := newOwner.loadTeams(sess); err != nil {
  255. return fmt.Errorf("LoadTeams: %v", err)
  256. }
  257. for _, t := range newOwner.Teams {
  258. if t.IncludesAllRepositories {
  259. if err := t.addRepository(sess, repo); err != nil {
  260. return fmt.Errorf("addRepository: %v", err)
  261. }
  262. }
  263. }
  264. } else if err := repo.recalculateAccesses(sess); err != nil {
  265. // Organization called this in addRepository method.
  266. return fmt.Errorf("recalculateAccesses: %v", err)
  267. }
  268. // Update repository count.
  269. if _, err := sess.Exec("UPDATE `user` SET num_repos=num_repos+1 WHERE id=?", newOwner.ID); err != nil {
  270. return fmt.Errorf("increase new owner repository count: %v", err)
  271. } else if _, err := sess.Exec("UPDATE `user` SET num_repos=num_repos-1 WHERE id=?", oldOwner.ID); err != nil {
  272. return fmt.Errorf("decrease old owner repository count: %v", err)
  273. }
  274. if err := watchRepo(sess, doer.ID, repo.ID, true); err != nil {
  275. return fmt.Errorf("watchRepo: %v", err)
  276. }
  277. // Remove watch for organization.
  278. if oldOwner.IsOrganization() {
  279. if err := watchRepo(sess, oldOwner.ID, repo.ID, false); err != nil {
  280. return fmt.Errorf("watchRepo [false]: %v", err)
  281. }
  282. }
  283. // Delete labels that belong to the old organization and comments that added these labels
  284. if oldOwner.IsOrganization() {
  285. if _, err := sess.Exec(`DELETE FROM issue_label WHERE issue_label.id IN (
  286. SELECT il_too.id FROM (
  287. SELECT il_too_too.id
  288. FROM issue_label AS il_too_too
  289. INNER JOIN label ON il_too_too.label_id = label.id
  290. INNER JOIN issue on issue.id = il_too_too.issue_id
  291. WHERE
  292. issue.repo_id = ? AND ((label.org_id = 0 AND issue.repo_id != label.repo_id) OR (label.repo_id = 0 AND label.org_id != ?))
  293. ) AS il_too )`, repo.ID, newOwner.ID); err != nil {
  294. return fmt.Errorf("Unable to remove old org labels: %v", err)
  295. }
  296. if _, err := sess.Exec(`DELETE FROM comment WHERE comment.id IN (
  297. SELECT il_too.id FROM (
  298. SELECT com.id
  299. FROM comment AS com
  300. INNER JOIN label ON com.label_id = label.id
  301. INNER JOIN issue ON issue.id = com.issue_id
  302. WHERE
  303. com.type = ? AND issue.repo_id = ? AND ((label.org_id = 0 AND issue.repo_id != label.repo_id) OR (label.repo_id = 0 AND label.org_id != ?))
  304. ) AS il_too)`, CommentTypeLabel, repo.ID, newOwner.ID); err != nil {
  305. return fmt.Errorf("Unable to remove old org label comments: %v", err)
  306. }
  307. }
  308. // Rename remote repository to new path and delete local copy.
  309. dir := UserPath(newOwner.Name)
  310. if err := os.MkdirAll(dir, os.ModePerm); err != nil {
  311. return fmt.Errorf("Failed to create dir %s: %v", dir, err)
  312. }
  313. if err := util.Rename(RepoPath(oldOwner.Name, repo.Name), RepoPath(newOwner.Name, repo.Name)); err != nil {
  314. return fmt.Errorf("rename repository directory: %v", err)
  315. }
  316. repoRenamed = true
  317. // Rename remote wiki repository to new path and delete local copy.
  318. wikiPath := WikiPath(oldOwner.Name, repo.Name)
  319. if isExist, err := util.IsExist(wikiPath); err != nil {
  320. log.Error("Unable to check if %s exists. Error: %v", wikiPath, err)
  321. return err
  322. } else if isExist {
  323. if err := util.Rename(wikiPath, WikiPath(newOwner.Name, repo.Name)); err != nil {
  324. return fmt.Errorf("rename repository wiki: %v", err)
  325. }
  326. wikiRenamed = true
  327. }
  328. if err := deleteRepositoryTransfer(sess, repo.ID); err != nil {
  329. return fmt.Errorf("deleteRepositoryTransfer: %v", err)
  330. }
  331. repo.Status = RepositoryReady
  332. if err := updateRepositoryCols(sess, repo, "status"); err != nil {
  333. return err
  334. }
  335. // If there was previously a redirect at this location, remove it.
  336. if err := deleteRepoRedirect(sess, newOwner.ID, repo.Name); err != nil {
  337. return fmt.Errorf("delete repo redirect: %v", err)
  338. }
  339. if err := newRepoRedirect(sess, oldOwner.ID, repo.ID, repo.Name, repo.Name); err != nil {
  340. return fmt.Errorf("newRepoRedirect: %v", err)
  341. }
  342. return sess.Commit()
  343. }