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.

org_team.go 27KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070
  1. // Copyright 2018 The Gitea Authors. All rights reserved.
  2. // Copyright 2016 The Gogs Authors. All rights reserved.
  3. // Use of this source code is governed by a MIT-style
  4. // license that can be found in the LICENSE file.
  5. package models
  6. import (
  7. "errors"
  8. "fmt"
  9. "sort"
  10. "strings"
  11. "code.gitea.io/gitea/models/db"
  12. "code.gitea.io/gitea/modules/log"
  13. "code.gitea.io/gitea/modules/setting"
  14. "xorm.io/builder"
  15. "xorm.io/xorm"
  16. )
  17. const ownerTeamName = "Owners"
  18. // Team represents a organization team.
  19. type Team struct {
  20. ID int64 `xorm:"pk autoincr"`
  21. OrgID int64 `xorm:"INDEX"`
  22. LowerName string
  23. Name string
  24. Description string
  25. Authorize AccessMode
  26. Repos []*Repository `xorm:"-"`
  27. Members []*User `xorm:"-"`
  28. NumRepos int
  29. NumMembers int
  30. Units []*TeamUnit `xorm:"-"`
  31. IncludesAllRepositories bool `xorm:"NOT NULL DEFAULT false"`
  32. CanCreateOrgRepo bool `xorm:"NOT NULL DEFAULT false"`
  33. }
  34. func init() {
  35. db.RegisterModel(new(Team))
  36. db.RegisterModel(new(TeamUser))
  37. db.RegisterModel(new(TeamRepo))
  38. db.RegisterModel(new(TeamUnit))
  39. }
  40. // SearchTeamOptions holds the search options
  41. type SearchTeamOptions struct {
  42. db.ListOptions
  43. UserID int64
  44. Keyword string
  45. OrgID int64
  46. IncludeDesc bool
  47. }
  48. // SearchMembersOptions holds the search options
  49. type SearchMembersOptions struct {
  50. db.ListOptions
  51. }
  52. // SearchTeam search for teams. Caller is responsible to check permissions.
  53. func SearchTeam(opts *SearchTeamOptions) ([]*Team, int64, error) {
  54. if opts.Page <= 0 {
  55. opts.Page = 1
  56. }
  57. if opts.PageSize == 0 {
  58. // Default limit
  59. opts.PageSize = 10
  60. }
  61. cond := builder.NewCond()
  62. if len(opts.Keyword) > 0 {
  63. lowerKeyword := strings.ToLower(opts.Keyword)
  64. var keywordCond builder.Cond = builder.Like{"lower_name", lowerKeyword}
  65. if opts.IncludeDesc {
  66. keywordCond = keywordCond.Or(builder.Like{"LOWER(description)", lowerKeyword})
  67. }
  68. cond = cond.And(keywordCond)
  69. }
  70. cond = cond.And(builder.Eq{"org_id": opts.OrgID})
  71. sess := db.NewSession(db.DefaultContext)
  72. defer sess.Close()
  73. count, err := sess.
  74. Where(cond).
  75. Count(new(Team))
  76. if err != nil {
  77. return nil, 0, err
  78. }
  79. sess = sess.Where(cond)
  80. if opts.PageSize == -1 {
  81. opts.PageSize = int(count)
  82. } else {
  83. sess = sess.Limit(opts.PageSize, (opts.Page-1)*opts.PageSize)
  84. }
  85. teams := make([]*Team, 0, opts.PageSize)
  86. if err = sess.
  87. OrderBy("lower_name").
  88. Find(&teams); err != nil {
  89. return nil, 0, err
  90. }
  91. return teams, count, nil
  92. }
  93. // ColorFormat provides a basic color format for a Team
  94. func (t *Team) ColorFormat(s fmt.State) {
  95. log.ColorFprintf(s, "%d:%s (OrgID: %d) %-v",
  96. log.NewColoredIDValue(t.ID),
  97. t.Name,
  98. log.NewColoredIDValue(t.OrgID),
  99. t.Authorize)
  100. }
  101. // GetUnits return a list of available units for a team
  102. func (t *Team) GetUnits() error {
  103. return t.getUnits(db.GetEngine(db.DefaultContext))
  104. }
  105. func (t *Team) getUnits(e db.Engine) (err error) {
  106. if t.Units != nil {
  107. return nil
  108. }
  109. t.Units, err = getUnitsByTeamID(e, t.ID)
  110. return err
  111. }
  112. // GetUnitNames returns the team units names
  113. func (t *Team) GetUnitNames() (res []string) {
  114. for _, u := range t.Units {
  115. res = append(res, Units[u.Type].NameKey)
  116. }
  117. return
  118. }
  119. // HasWriteAccess returns true if team has at least write level access mode.
  120. func (t *Team) HasWriteAccess() bool {
  121. return t.Authorize >= AccessModeWrite
  122. }
  123. // IsOwnerTeam returns true if team is owner team.
  124. func (t *Team) IsOwnerTeam() bool {
  125. return t.Name == ownerTeamName
  126. }
  127. // IsMember returns true if given user is a member of team.
  128. func (t *Team) IsMember(userID int64) bool {
  129. isMember, err := IsTeamMember(t.OrgID, t.ID, userID)
  130. if err != nil {
  131. log.Error("IsMember: %v", err)
  132. return false
  133. }
  134. return isMember
  135. }
  136. func (t *Team) getRepositories(e db.Engine) error {
  137. if t.Repos != nil {
  138. return nil
  139. }
  140. return e.Join("INNER", "team_repo", "repository.id = team_repo.repo_id").
  141. Where("team_repo.team_id=?", t.ID).
  142. OrderBy("repository.name").
  143. Find(&t.Repos)
  144. }
  145. // GetRepositories returns paginated repositories in team of organization.
  146. func (t *Team) GetRepositories(opts *SearchTeamOptions) error {
  147. if opts.Page == 0 {
  148. return t.getRepositories(db.GetEngine(db.DefaultContext))
  149. }
  150. return t.getRepositories(db.GetPaginatedSession(opts))
  151. }
  152. func (t *Team) getMembers(e db.Engine) (err error) {
  153. t.Members, err = getTeamMembers(e, t.ID)
  154. return err
  155. }
  156. // GetMembers returns paginated members in team of organization.
  157. func (t *Team) GetMembers(opts *SearchMembersOptions) (err error) {
  158. if opts.Page == 0 {
  159. return t.getMembers(db.GetEngine(db.DefaultContext))
  160. }
  161. return t.getMembers(db.GetPaginatedSession(opts))
  162. }
  163. // AddMember adds new membership of the team to the organization,
  164. // the user will have membership to the organization automatically when needed.
  165. func (t *Team) AddMember(userID int64) error {
  166. return AddTeamMember(t, userID)
  167. }
  168. // RemoveMember removes member from team of organization.
  169. func (t *Team) RemoveMember(userID int64) error {
  170. return RemoveTeamMember(t, userID)
  171. }
  172. func (t *Team) hasRepository(e db.Engine, repoID int64) bool {
  173. return hasTeamRepo(e, t.OrgID, t.ID, repoID)
  174. }
  175. // HasRepository returns true if given repository belong to team.
  176. func (t *Team) HasRepository(repoID int64) bool {
  177. return t.hasRepository(db.GetEngine(db.DefaultContext), repoID)
  178. }
  179. func (t *Team) addRepository(e db.Engine, repo *Repository) (err error) {
  180. if err = addTeamRepo(e, t.OrgID, t.ID, repo.ID); err != nil {
  181. return err
  182. }
  183. if _, err = e.Incr("num_repos").ID(t.ID).Update(new(Team)); err != nil {
  184. return fmt.Errorf("update team: %v", err)
  185. }
  186. t.NumRepos++
  187. if err = repo.recalculateTeamAccesses(e, 0); err != nil {
  188. return fmt.Errorf("recalculateAccesses: %v", err)
  189. }
  190. // Make all team members watch this repo if enabled in global settings
  191. if setting.Service.AutoWatchNewRepos {
  192. if err = t.getMembers(e); err != nil {
  193. return fmt.Errorf("getMembers: %v", err)
  194. }
  195. for _, u := range t.Members {
  196. if err = watchRepo(e, u.ID, repo.ID, true); err != nil {
  197. return fmt.Errorf("watchRepo: %v", err)
  198. }
  199. }
  200. }
  201. return nil
  202. }
  203. // addAllRepositories adds all repositories to the team.
  204. // If the team already has some repositories they will be left unchanged.
  205. func (t *Team) addAllRepositories(e db.Engine) error {
  206. var orgRepos []Repository
  207. if err := e.Where("owner_id = ?", t.OrgID).Find(&orgRepos); err != nil {
  208. return fmt.Errorf("get org repos: %v", err)
  209. }
  210. for _, repo := range orgRepos {
  211. if !t.hasRepository(e, repo.ID) {
  212. if err := t.addRepository(e, &repo); err != nil {
  213. return fmt.Errorf("addRepository: %v", err)
  214. }
  215. }
  216. }
  217. return nil
  218. }
  219. // AddAllRepositories adds all repositories to the team
  220. func (t *Team) AddAllRepositories() (err error) {
  221. sess := db.NewSession(db.DefaultContext)
  222. defer sess.Close()
  223. if err = sess.Begin(); err != nil {
  224. return err
  225. }
  226. if err = t.addAllRepositories(sess); err != nil {
  227. return err
  228. }
  229. return sess.Commit()
  230. }
  231. // AddRepository adds new repository to team of organization.
  232. func (t *Team) AddRepository(repo *Repository) (err error) {
  233. if repo.OwnerID != t.OrgID {
  234. return errors.New("Repository does not belong to organization")
  235. } else if t.HasRepository(repo.ID) {
  236. return nil
  237. }
  238. sess := db.NewSession(db.DefaultContext)
  239. defer sess.Close()
  240. if err = sess.Begin(); err != nil {
  241. return err
  242. }
  243. if err = t.addRepository(sess, repo); err != nil {
  244. return err
  245. }
  246. return sess.Commit()
  247. }
  248. // RemoveAllRepositories removes all repositories from team and recalculates access
  249. func (t *Team) RemoveAllRepositories() (err error) {
  250. if t.IncludesAllRepositories {
  251. return nil
  252. }
  253. sess := db.NewSession(db.DefaultContext)
  254. defer sess.Close()
  255. if err = sess.Begin(); err != nil {
  256. return err
  257. }
  258. if err = t.removeAllRepositories(sess); err != nil {
  259. return err
  260. }
  261. return sess.Commit()
  262. }
  263. // removeAllRepositories removes all repositories from team and recalculates access
  264. // Note: Shall not be called if team includes all repositories
  265. func (t *Team) removeAllRepositories(e db.Engine) (err error) {
  266. // Delete all accesses.
  267. for _, repo := range t.Repos {
  268. if err := repo.recalculateTeamAccesses(e, t.ID); err != nil {
  269. return err
  270. }
  271. // Remove watches from all users and now unaccessible repos
  272. for _, user := range t.Members {
  273. has, err := hasAccess(e, user.ID, repo)
  274. if err != nil {
  275. return err
  276. } else if has {
  277. continue
  278. }
  279. if err = watchRepo(e, user.ID, repo.ID, false); err != nil {
  280. return err
  281. }
  282. // Remove all IssueWatches a user has subscribed to in the repositories
  283. if err = removeIssueWatchersByRepoID(e, user.ID, repo.ID); err != nil {
  284. return err
  285. }
  286. }
  287. }
  288. // Delete team-repo
  289. if _, err := e.
  290. Where("team_id=?", t.ID).
  291. Delete(new(TeamRepo)); err != nil {
  292. return err
  293. }
  294. t.NumRepos = 0
  295. if _, err = e.ID(t.ID).Cols("num_repos").Update(t); err != nil {
  296. return err
  297. }
  298. return nil
  299. }
  300. // removeRepository removes a repository from a team and recalculates access
  301. // Note: Repository shall not be removed from team if it includes all repositories (unless the repository is deleted)
  302. func (t *Team) removeRepository(e db.Engine, repo *Repository, recalculate bool) (err error) {
  303. if err = removeTeamRepo(e, t.ID, repo.ID); err != nil {
  304. return err
  305. }
  306. t.NumRepos--
  307. if _, err = e.ID(t.ID).Cols("num_repos").Update(t); err != nil {
  308. return err
  309. }
  310. // Don't need to recalculate when delete a repository from organization.
  311. if recalculate {
  312. if err = repo.recalculateTeamAccesses(e, t.ID); err != nil {
  313. return err
  314. }
  315. }
  316. teamUsers, err := getTeamUsersByTeamID(e, t.ID)
  317. if err != nil {
  318. return fmt.Errorf("getTeamUsersByTeamID: %v", err)
  319. }
  320. for _, teamUser := range teamUsers {
  321. has, err := hasAccess(e, teamUser.UID, repo)
  322. if err != nil {
  323. return err
  324. } else if has {
  325. continue
  326. }
  327. if err = watchRepo(e, teamUser.UID, repo.ID, false); err != nil {
  328. return err
  329. }
  330. // Remove all IssueWatches a user has subscribed to in the repositories
  331. if err := removeIssueWatchersByRepoID(e, teamUser.UID, repo.ID); err != nil {
  332. return err
  333. }
  334. }
  335. return nil
  336. }
  337. // RemoveRepository removes repository from team of organization.
  338. // If the team shall include all repositories the request is ignored.
  339. func (t *Team) RemoveRepository(repoID int64) error {
  340. if !t.HasRepository(repoID) {
  341. return nil
  342. }
  343. if t.IncludesAllRepositories {
  344. return nil
  345. }
  346. repo, err := GetRepositoryByID(repoID)
  347. if err != nil {
  348. return err
  349. }
  350. sess := db.NewSession(db.DefaultContext)
  351. defer sess.Close()
  352. if err = sess.Begin(); err != nil {
  353. return err
  354. }
  355. if err = t.removeRepository(sess, repo, true); err != nil {
  356. return err
  357. }
  358. return sess.Commit()
  359. }
  360. // UnitEnabled returns if the team has the given unit type enabled
  361. func (t *Team) UnitEnabled(tp UnitType) bool {
  362. return t.unitEnabled(db.GetEngine(db.DefaultContext), tp)
  363. }
  364. func (t *Team) unitEnabled(e db.Engine, tp UnitType) bool {
  365. if err := t.getUnits(e); err != nil {
  366. log.Warn("Error loading team (ID: %d) units: %s", t.ID, err.Error())
  367. }
  368. for _, unit := range t.Units {
  369. if unit.Type == tp {
  370. return true
  371. }
  372. }
  373. return false
  374. }
  375. // IsUsableTeamName tests if a name could be as team name
  376. func IsUsableTeamName(name string) error {
  377. switch name {
  378. case "new":
  379. return ErrNameReserved{name}
  380. default:
  381. return nil
  382. }
  383. }
  384. // NewTeam creates a record of new team.
  385. // It's caller's responsibility to assign organization ID.
  386. func NewTeam(t *Team) (err error) {
  387. if len(t.Name) == 0 {
  388. return errors.New("empty team name")
  389. }
  390. if err = IsUsableTeamName(t.Name); err != nil {
  391. return err
  392. }
  393. has, err := db.GetEngine(db.DefaultContext).ID(t.OrgID).Get(new(User))
  394. if err != nil {
  395. return err
  396. }
  397. if !has {
  398. return ErrOrgNotExist{t.OrgID, ""}
  399. }
  400. t.LowerName = strings.ToLower(t.Name)
  401. has, err = db.GetEngine(db.DefaultContext).
  402. Where("org_id=?", t.OrgID).
  403. And("lower_name=?", t.LowerName).
  404. Get(new(Team))
  405. if err != nil {
  406. return err
  407. }
  408. if has {
  409. return ErrTeamAlreadyExist{t.OrgID, t.LowerName}
  410. }
  411. sess := db.NewSession(db.DefaultContext)
  412. defer sess.Close()
  413. if err = sess.Begin(); err != nil {
  414. return err
  415. }
  416. if _, err = sess.Insert(t); err != nil {
  417. errRollback := sess.Rollback()
  418. if errRollback != nil {
  419. log.Error("NewTeam sess.Rollback: %v", errRollback)
  420. }
  421. return err
  422. }
  423. // insert units for team
  424. if len(t.Units) > 0 {
  425. for _, unit := range t.Units {
  426. unit.TeamID = t.ID
  427. }
  428. if _, err = sess.Insert(&t.Units); err != nil {
  429. errRollback := sess.Rollback()
  430. if errRollback != nil {
  431. log.Error("NewTeam sess.Rollback: %v", errRollback)
  432. }
  433. return err
  434. }
  435. }
  436. // Add all repositories to the team if it has access to all of them.
  437. if t.IncludesAllRepositories {
  438. err = t.addAllRepositories(sess)
  439. if err != nil {
  440. return fmt.Errorf("addAllRepositories: %v", err)
  441. }
  442. }
  443. // Update organization number of teams.
  444. if _, err = sess.Exec("UPDATE `user` SET num_teams=num_teams+1 WHERE id = ?", t.OrgID); err != nil {
  445. errRollback := sess.Rollback()
  446. if errRollback != nil {
  447. log.Error("NewTeam sess.Rollback: %v", errRollback)
  448. }
  449. return err
  450. }
  451. return sess.Commit()
  452. }
  453. func getTeam(e db.Engine, orgID int64, name string) (*Team, error) {
  454. t := &Team{
  455. OrgID: orgID,
  456. LowerName: strings.ToLower(name),
  457. }
  458. has, err := e.Get(t)
  459. if err != nil {
  460. return nil, err
  461. } else if !has {
  462. return nil, ErrTeamNotExist{orgID, 0, name}
  463. }
  464. return t, nil
  465. }
  466. // GetTeam returns team by given team name and organization.
  467. func GetTeam(orgID int64, name string) (*Team, error) {
  468. return getTeam(db.GetEngine(db.DefaultContext), orgID, name)
  469. }
  470. // GetTeamIDsByNames returns a slice of team ids corresponds to names.
  471. func GetTeamIDsByNames(orgID int64, names []string, ignoreNonExistent bool) ([]int64, error) {
  472. ids := make([]int64, 0, len(names))
  473. for _, name := range names {
  474. u, err := GetTeam(orgID, name)
  475. if err != nil {
  476. if ignoreNonExistent {
  477. continue
  478. } else {
  479. return nil, err
  480. }
  481. }
  482. ids = append(ids, u.ID)
  483. }
  484. return ids, nil
  485. }
  486. // getOwnerTeam returns team by given team name and organization.
  487. func getOwnerTeam(e db.Engine, orgID int64) (*Team, error) {
  488. return getTeam(e, orgID, ownerTeamName)
  489. }
  490. func getTeamByID(e db.Engine, teamID int64) (*Team, error) {
  491. t := new(Team)
  492. has, err := e.ID(teamID).Get(t)
  493. if err != nil {
  494. return nil, err
  495. } else if !has {
  496. return nil, ErrTeamNotExist{0, teamID, ""}
  497. }
  498. return t, nil
  499. }
  500. // GetTeamByID returns team by given ID.
  501. func GetTeamByID(teamID int64) (*Team, error) {
  502. return getTeamByID(db.GetEngine(db.DefaultContext), teamID)
  503. }
  504. // GetTeamNamesByID returns team's lower name from a list of team ids.
  505. func GetTeamNamesByID(teamIDs []int64) ([]string, error) {
  506. if len(teamIDs) == 0 {
  507. return []string{}, nil
  508. }
  509. var teamNames []string
  510. err := db.GetEngine(db.DefaultContext).Table("team").
  511. Select("lower_name").
  512. In("id", teamIDs).
  513. Asc("name").
  514. Find(&teamNames)
  515. return teamNames, err
  516. }
  517. // UpdateTeam updates information of team.
  518. func UpdateTeam(t *Team, authChanged, includeAllChanged bool) (err error) {
  519. if len(t.Name) == 0 {
  520. return errors.New("empty team name")
  521. }
  522. if len(t.Description) > 255 {
  523. t.Description = t.Description[:255]
  524. }
  525. sess := db.NewSession(db.DefaultContext)
  526. defer sess.Close()
  527. if err = sess.Begin(); err != nil {
  528. return err
  529. }
  530. t.LowerName = strings.ToLower(t.Name)
  531. has, err := sess.
  532. Where("org_id=?", t.OrgID).
  533. And("lower_name=?", t.LowerName).
  534. And("id!=?", t.ID).
  535. Get(new(Team))
  536. if err != nil {
  537. return err
  538. } else if has {
  539. return ErrTeamAlreadyExist{t.OrgID, t.LowerName}
  540. }
  541. if _, err = sess.ID(t.ID).Cols("name", "lower_name", "description",
  542. "can_create_org_repo", "authorize", "includes_all_repositories").Update(t); err != nil {
  543. return fmt.Errorf("update: %v", err)
  544. }
  545. // update units for team
  546. if len(t.Units) > 0 {
  547. for _, unit := range t.Units {
  548. unit.TeamID = t.ID
  549. }
  550. // Delete team-unit.
  551. if _, err := sess.
  552. Where("team_id=?", t.ID).
  553. Delete(new(TeamUnit)); err != nil {
  554. return err
  555. }
  556. if _, err = sess.Cols("org_id", "team_id", "type").Insert(&t.Units); err != nil {
  557. errRollback := sess.Rollback()
  558. if errRollback != nil {
  559. log.Error("UpdateTeam sess.Rollback: %v", errRollback)
  560. }
  561. return err
  562. }
  563. }
  564. // Update access for team members if needed.
  565. if authChanged {
  566. if err = t.getRepositories(sess); err != nil {
  567. return fmt.Errorf("getRepositories: %v", err)
  568. }
  569. for _, repo := range t.Repos {
  570. if err = repo.recalculateTeamAccesses(sess, 0); err != nil {
  571. return fmt.Errorf("recalculateTeamAccesses: %v", err)
  572. }
  573. }
  574. }
  575. // Add all repositories to the team if it has access to all of them.
  576. if includeAllChanged && t.IncludesAllRepositories {
  577. err = t.addAllRepositories(sess)
  578. if err != nil {
  579. return fmt.Errorf("addAllRepositories: %v", err)
  580. }
  581. }
  582. return sess.Commit()
  583. }
  584. // DeleteTeam deletes given team.
  585. // It's caller's responsibility to assign organization ID.
  586. func DeleteTeam(t *Team) error {
  587. if err := t.GetRepositories(&SearchTeamOptions{}); err != nil {
  588. return err
  589. }
  590. sess := db.NewSession(db.DefaultContext)
  591. defer sess.Close()
  592. if err := sess.Begin(); err != nil {
  593. return err
  594. }
  595. if err := t.getMembers(sess); err != nil {
  596. return err
  597. }
  598. if err := t.removeAllRepositories(sess); err != nil {
  599. return err
  600. }
  601. // Delete team-user.
  602. if _, err := sess.
  603. Where("org_id=?", t.OrgID).
  604. Where("team_id=?", t.ID).
  605. Delete(new(TeamUser)); err != nil {
  606. return err
  607. }
  608. // Delete team-unit.
  609. if _, err := sess.
  610. Where("team_id=?", t.ID).
  611. Delete(new(TeamUnit)); err != nil {
  612. return err
  613. }
  614. // Delete team.
  615. if _, err := sess.ID(t.ID).Delete(new(Team)); err != nil {
  616. return err
  617. }
  618. // Update organization number of teams.
  619. if _, err := sess.Exec("UPDATE `user` SET num_teams=num_teams-1 WHERE id=?", t.OrgID); err != nil {
  620. return err
  621. }
  622. return sess.Commit()
  623. }
  624. // ___________ ____ ___
  625. // \__ ___/___ _____ _____ | | \______ ___________
  626. // | |_/ __ \\__ \ / \| | / ___// __ \_ __ \
  627. // | |\ ___/ / __ \| Y Y \ | /\___ \\ ___/| | \/
  628. // |____| \___ >____ /__|_| /______//____ >\___ >__|
  629. // \/ \/ \/ \/ \/
  630. // TeamUser represents an team-user relation.
  631. type TeamUser struct {
  632. ID int64 `xorm:"pk autoincr"`
  633. OrgID int64 `xorm:"INDEX"`
  634. TeamID int64 `xorm:"UNIQUE(s)"`
  635. UID int64 `xorm:"UNIQUE(s)"`
  636. }
  637. func isTeamMember(e db.Engine, orgID, teamID, userID int64) (bool, error) {
  638. return e.
  639. Where("org_id=?", orgID).
  640. And("team_id=?", teamID).
  641. And("uid=?", userID).
  642. Table("team_user").
  643. Exist()
  644. }
  645. // IsTeamMember returns true if given user is a member of team.
  646. func IsTeamMember(orgID, teamID, userID int64) (bool, error) {
  647. return isTeamMember(db.GetEngine(db.DefaultContext), orgID, teamID, userID)
  648. }
  649. func getTeamUsersByTeamID(e db.Engine, teamID int64) ([]*TeamUser, error) {
  650. teamUsers := make([]*TeamUser, 0, 10)
  651. return teamUsers, e.
  652. Where("team_id=?", teamID).
  653. Find(&teamUsers)
  654. }
  655. func getTeamMembers(e db.Engine, teamID int64) (_ []*User, err error) {
  656. teamUsers, err := getTeamUsersByTeamID(e, teamID)
  657. if err != nil {
  658. return nil, fmt.Errorf("get team-users: %v", err)
  659. }
  660. members := make([]*User, len(teamUsers))
  661. for i, teamUser := range teamUsers {
  662. member, err := getUserByID(e, teamUser.UID)
  663. if err != nil {
  664. return nil, fmt.Errorf("get user '%d': %v", teamUser.UID, err)
  665. }
  666. members[i] = member
  667. }
  668. sort.Slice(members, func(i, j int) bool {
  669. return members[i].DisplayName() < members[j].DisplayName()
  670. })
  671. return members, nil
  672. }
  673. // GetTeamMembers returns all members in given team of organization.
  674. func GetTeamMembers(teamID int64) ([]*User, error) {
  675. return getTeamMembers(db.GetEngine(db.DefaultContext), teamID)
  676. }
  677. func getUserOrgTeams(e db.Engine, orgID, userID int64) (teams []*Team, err error) {
  678. return teams, e.
  679. Join("INNER", "team_user", "team_user.team_id = team.id").
  680. Where("team.org_id = ?", orgID).
  681. And("team_user.uid=?", userID).
  682. Find(&teams)
  683. }
  684. func getUserRepoTeams(e db.Engine, orgID, userID, repoID int64) (teams []*Team, err error) {
  685. return teams, e.
  686. Join("INNER", "team_user", "team_user.team_id = team.id").
  687. Join("INNER", "team_repo", "team_repo.team_id = team.id").
  688. Where("team.org_id = ?", orgID).
  689. And("team_user.uid=?", userID).
  690. And("team_repo.repo_id=?", repoID).
  691. Find(&teams)
  692. }
  693. // GetUserOrgTeams returns all teams that user belongs to in given organization.
  694. func GetUserOrgTeams(orgID, userID int64) ([]*Team, error) {
  695. return getUserOrgTeams(db.GetEngine(db.DefaultContext), orgID, userID)
  696. }
  697. // AddTeamMember adds new membership of given team to given organization,
  698. // the user will have membership to given organization automatically when needed.
  699. func AddTeamMember(team *Team, userID int64) error {
  700. isAlreadyMember, err := IsTeamMember(team.OrgID, team.ID, userID)
  701. if err != nil || isAlreadyMember {
  702. return err
  703. }
  704. if err := AddOrgUser(team.OrgID, userID); err != nil {
  705. return err
  706. }
  707. // Get team and its repositories.
  708. if err := team.GetRepositories(&SearchTeamOptions{}); err != nil {
  709. return err
  710. }
  711. sess := db.NewSession(db.DefaultContext)
  712. defer sess.Close()
  713. if err := sess.Begin(); err != nil {
  714. return err
  715. }
  716. if _, err := sess.Insert(&TeamUser{
  717. UID: userID,
  718. OrgID: team.OrgID,
  719. TeamID: team.ID,
  720. }); err != nil {
  721. return err
  722. } else if _, err := sess.Incr("num_members").ID(team.ID).Update(new(Team)); err != nil {
  723. return err
  724. }
  725. team.NumMembers++
  726. // Give access to team repositories.
  727. for _, repo := range team.Repos {
  728. if err := repo.recalculateUserAccess(sess, userID); err != nil {
  729. return err
  730. }
  731. if setting.Service.AutoWatchNewRepos {
  732. if err = watchRepo(sess, userID, repo.ID, true); err != nil {
  733. return err
  734. }
  735. }
  736. }
  737. return sess.Commit()
  738. }
  739. func removeTeamMember(e *xorm.Session, team *Team, userID int64) error {
  740. isMember, err := isTeamMember(e, team.OrgID, team.ID, userID)
  741. if err != nil || !isMember {
  742. return err
  743. }
  744. // Check if the user to delete is the last member in owner team.
  745. if team.IsOwnerTeam() && team.NumMembers == 1 {
  746. return ErrLastOrgOwner{UID: userID}
  747. }
  748. team.NumMembers--
  749. if err := team.getRepositories(e); err != nil {
  750. return err
  751. }
  752. if _, err := e.Delete(&TeamUser{
  753. UID: userID,
  754. OrgID: team.OrgID,
  755. TeamID: team.ID,
  756. }); err != nil {
  757. return err
  758. } else if _, err = e.
  759. ID(team.ID).
  760. Cols("num_members").
  761. Update(team); err != nil {
  762. return err
  763. }
  764. // Delete access to team repositories.
  765. for _, repo := range team.Repos {
  766. if err := repo.recalculateUserAccess(e, userID); err != nil {
  767. return err
  768. }
  769. // Remove watches from now unaccessible
  770. if err := repo.reconsiderWatches(e, userID); err != nil {
  771. return err
  772. }
  773. // Remove issue assignments from now unaccessible
  774. if err := repo.reconsiderIssueAssignees(e, userID); err != nil {
  775. return err
  776. }
  777. }
  778. // Check if the user is a member of any team in the organization.
  779. if count, err := e.Count(&TeamUser{
  780. UID: userID,
  781. OrgID: team.OrgID,
  782. }); err != nil {
  783. return err
  784. } else if count == 0 {
  785. return removeOrgUser(e, team.OrgID, userID)
  786. }
  787. return nil
  788. }
  789. // RemoveTeamMember removes member from given team of given organization.
  790. func RemoveTeamMember(team *Team, userID int64) error {
  791. sess := db.NewSession(db.DefaultContext)
  792. defer sess.Close()
  793. if err := sess.Begin(); err != nil {
  794. return err
  795. }
  796. if err := removeTeamMember(sess, team, userID); err != nil {
  797. return err
  798. }
  799. return sess.Commit()
  800. }
  801. // IsUserInTeams returns if a user in some teams
  802. func IsUserInTeams(userID int64, teamIDs []int64) (bool, error) {
  803. return isUserInTeams(db.GetEngine(db.DefaultContext), userID, teamIDs)
  804. }
  805. func isUserInTeams(e db.Engine, userID int64, teamIDs []int64) (bool, error) {
  806. return e.Where("uid=?", userID).In("team_id", teamIDs).Exist(new(TeamUser))
  807. }
  808. // UsersInTeamsCount counts the number of users which are in userIDs and teamIDs
  809. func UsersInTeamsCount(userIDs, teamIDs []int64) (int64, error) {
  810. var ids []int64
  811. if err := db.GetEngine(db.DefaultContext).In("uid", userIDs).In("team_id", teamIDs).
  812. Table("team_user").
  813. Cols("uid").GroupBy("uid").Find(&ids); err != nil {
  814. return 0, err
  815. }
  816. return int64(len(ids)), nil
  817. }
  818. // ___________ __________
  819. // \__ ___/___ _____ _____\______ \ ____ ______ ____
  820. // | |_/ __ \\__ \ / \| _// __ \\____ \ / _ \
  821. // | |\ ___/ / __ \| Y Y \ | \ ___/| |_> > <_> )
  822. // |____| \___ >____ /__|_| /____|_ /\___ > __/ \____/
  823. // \/ \/ \/ \/ \/|__|
  824. // TeamRepo represents an team-repository relation.
  825. type TeamRepo struct {
  826. ID int64 `xorm:"pk autoincr"`
  827. OrgID int64 `xorm:"INDEX"`
  828. TeamID int64 `xorm:"UNIQUE(s)"`
  829. RepoID int64 `xorm:"UNIQUE(s)"`
  830. }
  831. func hasTeamRepo(e db.Engine, orgID, teamID, repoID int64) bool {
  832. has, _ := e.
  833. Where("org_id=?", orgID).
  834. And("team_id=?", teamID).
  835. And("repo_id=?", repoID).
  836. Get(new(TeamRepo))
  837. return has
  838. }
  839. // HasTeamRepo returns true if given repository belongs to team.
  840. func HasTeamRepo(orgID, teamID, repoID int64) bool {
  841. return hasTeamRepo(db.GetEngine(db.DefaultContext), orgID, teamID, repoID)
  842. }
  843. func addTeamRepo(e db.Engine, orgID, teamID, repoID int64) error {
  844. _, err := e.InsertOne(&TeamRepo{
  845. OrgID: orgID,
  846. TeamID: teamID,
  847. RepoID: repoID,
  848. })
  849. return err
  850. }
  851. func removeTeamRepo(e db.Engine, teamID, repoID int64) error {
  852. _, err := e.Delete(&TeamRepo{
  853. TeamID: teamID,
  854. RepoID: repoID,
  855. })
  856. return err
  857. }
  858. // GetTeamsWithAccessToRepo returns all teams in an organization that have given access level to the repository.
  859. func GetTeamsWithAccessToRepo(orgID, repoID int64, mode AccessMode) ([]*Team, error) {
  860. teams := make([]*Team, 0, 5)
  861. return teams, db.GetEngine(db.DefaultContext).Where("team.authorize >= ?", mode).
  862. Join("INNER", "team_repo", "team_repo.team_id = team.id").
  863. And("team_repo.org_id = ?", orgID).
  864. And("team_repo.repo_id = ?", repoID).
  865. Find(&teams)
  866. }
  867. // ___________ ____ ___ .__ __
  868. // \__ ___/___ _____ _____ | | \____ |__|/ |_
  869. // | |_/ __ \\__ \ / \| | / \| \ __\
  870. // | |\ ___/ / __ \| Y Y \ | / | \ || |
  871. // |____| \___ >____ /__|_| /______/|___| /__||__|
  872. // \/ \/ \/ \/
  873. // TeamUnit describes all units of a repository
  874. type TeamUnit struct {
  875. ID int64 `xorm:"pk autoincr"`
  876. OrgID int64 `xorm:"INDEX"`
  877. TeamID int64 `xorm:"UNIQUE(s)"`
  878. Type UnitType `xorm:"UNIQUE(s)"`
  879. }
  880. // Unit returns Unit
  881. func (t *TeamUnit) Unit() Unit {
  882. return Units[t.Type]
  883. }
  884. func getUnitsByTeamID(e db.Engine, teamID int64) (units []*TeamUnit, err error) {
  885. return units, e.Where("team_id = ?", teamID).Find(&units)
  886. }
  887. // UpdateTeamUnits updates a teams's units
  888. func UpdateTeamUnits(team *Team, units []TeamUnit) (err error) {
  889. sess := db.NewSession(db.DefaultContext)
  890. defer sess.Close()
  891. if err = sess.Begin(); err != nil {
  892. return err
  893. }
  894. if _, err = sess.Where("team_id = ?", team.ID).Delete(new(TeamUnit)); err != nil {
  895. return err
  896. }
  897. if len(units) > 0 {
  898. if _, err = sess.Insert(units); err != nil {
  899. errRollback := sess.Rollback()
  900. if errRollback != nil {
  901. log.Error("UpdateTeamUnits sess.Rollback: %v", errRollback)
  902. }
  903. return err
  904. }
  905. }
  906. return sess.Commit()
  907. }