org.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951
  1. // Copyright 2014 The Gogs 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. "errors"
  7. "fmt"
  8. "os"
  9. "path"
  10. "strings"
  11. "github.com/Unknwon/com"
  12. "github.com/go-xorm/xorm"
  13. "github.com/gogits/gogs/modules/base"
  14. )
  15. var (
  16. ErrOrgNotExist = errors.New("Organization does not exist")
  17. ErrTeamAlreadyExist = errors.New("Team already exist")
  18. ErrTeamNotExist = errors.New("Team does not exist")
  19. ErrTeamNameIllegal = errors.New("Team name contains illegal characters")
  20. ErrLastOrgOwner = errors.New("The user to remove is the last member in owner team")
  21. )
  22. // IsOrgOwner returns true if given user is in the owner team.
  23. func (org *User) IsOrgOwner(uid int64) bool {
  24. return IsOrganizationOwner(org.Id, uid)
  25. }
  26. // IsOrgMember returns true if given user is member of organization.
  27. func (org *User) IsOrgMember(uid int64) bool {
  28. return IsOrganizationMember(org.Id, uid)
  29. }
  30. // GetTeam returns named team of organization.
  31. func (org *User) GetTeam(name string) (*Team, error) {
  32. return GetTeam(org.Id, name)
  33. }
  34. // GetOwnerTeam returns owner team of organization.
  35. func (org *User) GetOwnerTeam() (*Team, error) {
  36. return org.GetTeam(OWNER_TEAM)
  37. }
  38. // GetTeams returns all teams that belong to organization.
  39. func (org *User) GetTeams() error {
  40. return x.Where("org_id=?", org.Id).Find(&org.Teams)
  41. }
  42. // GetMembers returns all members of organization.
  43. func (org *User) GetMembers() error {
  44. ous, err := GetOrgUsersByOrgId(org.Id)
  45. if err != nil {
  46. return err
  47. }
  48. org.Members = make([]*User, len(ous))
  49. for i, ou := range ous {
  50. org.Members[i], err = GetUserById(ou.Uid)
  51. if err != nil {
  52. return err
  53. }
  54. }
  55. return nil
  56. }
  57. // AddMember adds new member to organization.
  58. func (org *User) AddMember(uid int64) error {
  59. return AddOrgUser(org.Id, uid)
  60. }
  61. // RemoveMember removes member from organization.
  62. func (org *User) RemoveMember(uid int64) error {
  63. return RemoveOrgUser(org.Id, uid)
  64. }
  65. // CreateOrganization creates record of a new organization.
  66. func CreateOrganization(org, owner *User) (*User, error) {
  67. if !IsLegalName(org.Name) {
  68. return nil, ErrUserNameIllegal
  69. }
  70. isExist, err := IsUserExist(org.Name)
  71. if err != nil {
  72. return nil, err
  73. } else if isExist {
  74. return nil, ErrUserAlreadyExist
  75. }
  76. isExist, err = IsEmailUsed(org.Email)
  77. if err != nil {
  78. return nil, err
  79. } else if isExist {
  80. return nil, ErrEmailAlreadyUsed
  81. }
  82. org.LowerName = strings.ToLower(org.Name)
  83. org.FullName = org.Name
  84. org.Avatar = base.EncodeMd5(org.Email)
  85. org.AvatarEmail = org.Email
  86. // No password for organization.
  87. org.NumTeams = 1
  88. org.NumMembers = 1
  89. sess := x.NewSession()
  90. defer sess.Close()
  91. if err = sess.Begin(); err != nil {
  92. return nil, err
  93. }
  94. if _, err = sess.Insert(org); err != nil {
  95. sess.Rollback()
  96. return nil, err
  97. }
  98. if err = os.MkdirAll(UserPath(org.Name), os.ModePerm); err != nil {
  99. sess.Rollback()
  100. return nil, err
  101. }
  102. // Create default owner team.
  103. t := &Team{
  104. OrgId: org.Id,
  105. LowerName: strings.ToLower(OWNER_TEAM),
  106. Name: OWNER_TEAM,
  107. Authorize: ORG_ADMIN,
  108. NumMembers: 1,
  109. }
  110. if _, err = sess.Insert(t); err != nil {
  111. sess.Rollback()
  112. return nil, err
  113. }
  114. // Add initial creator to organization and owner team.
  115. ou := &OrgUser{
  116. Uid: owner.Id,
  117. OrgId: org.Id,
  118. IsOwner: true,
  119. NumTeams: 1,
  120. }
  121. if _, err = sess.Insert(ou); err != nil {
  122. sess.Rollback()
  123. return nil, err
  124. }
  125. tu := &TeamUser{
  126. Uid: owner.Id,
  127. OrgId: org.Id,
  128. TeamId: t.Id,
  129. }
  130. if _, err = sess.Insert(tu); err != nil {
  131. sess.Rollback()
  132. return nil, err
  133. }
  134. return org, sess.Commit()
  135. }
  136. // TODO: need some kind of mechanism to record failure.
  137. // DeleteOrganization completely and permanently deletes everything of organization.
  138. func DeleteOrganization(org *User) (err error) {
  139. if err := DeleteUser(org); err != nil {
  140. return err
  141. }
  142. sess := x.NewSession()
  143. defer sess.Close()
  144. if err = sess.Begin(); err != nil {
  145. return err
  146. }
  147. if _, err = sess.Delete(&Team{OrgId: org.Id}); err != nil {
  148. sess.Rollback()
  149. return err
  150. }
  151. if _, err = sess.Delete(&OrgUser{OrgId: org.Id}); err != nil {
  152. sess.Rollback()
  153. return err
  154. }
  155. if _, err = sess.Delete(&TeamUser{OrgId: org.Id}); err != nil {
  156. sess.Rollback()
  157. return err
  158. }
  159. return sess.Commit()
  160. }
  161. // ________ ____ ___
  162. // \_____ \_______ ____ | | \______ ___________
  163. // / | \_ __ \/ ___\| | / ___// __ \_ __ \
  164. // / | \ | \/ /_/ > | /\___ \\ ___/| | \/
  165. // \_______ /__| \___ /|______//____ >\___ >__|
  166. // \/ /_____/ \/ \/
  167. // OrgUser represents an organization-user relation.
  168. type OrgUser struct {
  169. Id int64
  170. Uid int64 `xorm:"INDEX UNIQUE(s)"`
  171. OrgId int64 `xorm:"INDEX UNIQUE(s)"`
  172. IsPublic bool
  173. IsOwner bool
  174. NumTeams int
  175. }
  176. // IsOrganizationOwner returns true if given user is in the owner team.
  177. func IsOrganizationOwner(orgId, uid int64) bool {
  178. has, _ := x.Where("is_owner=?", true).And("uid=?", uid).And("org_id=?", orgId).Get(new(OrgUser))
  179. return has
  180. }
  181. // IsOrganizationMember returns true if given user is member of organization.
  182. func IsOrganizationMember(orgId, uid int64) bool {
  183. has, _ := x.Where("uid=?", uid).And("org_id=?", orgId).Get(new(OrgUser))
  184. return has
  185. }
  186. // IsPublicMembership returns ture if given user public his/her membership.
  187. func IsPublicMembership(orgId, uid int64) bool {
  188. has, _ := x.Where("uid=?", uid).And("org_id=?", orgId).And("is_public=?", true).Get(new(OrgUser))
  189. return has
  190. }
  191. // GetOrgUsersByUserId returns all organization-user relations by user ID.
  192. func GetOrgUsersByUserId(uid int64) ([]*OrgUser, error) {
  193. ous := make([]*OrgUser, 0, 10)
  194. err := x.Where("uid=?", uid).Find(&ous)
  195. return ous, err
  196. }
  197. // GetOrgUsersByOrgId returns all organization-user relations by organization ID.
  198. func GetOrgUsersByOrgId(orgId int64) ([]*OrgUser, error) {
  199. ous := make([]*OrgUser, 0, 10)
  200. err := x.Where("org_id=?", orgId).Find(&ous)
  201. return ous, err
  202. }
  203. // ChangeOrgUserStatus changes public or private membership status.
  204. func ChangeOrgUserStatus(orgId, uid int64, public bool) error {
  205. ou := new(OrgUser)
  206. has, err := x.Where("uid=?", uid).And("org_id=?", orgId).Get(ou)
  207. if err != nil {
  208. return err
  209. } else if !has {
  210. return nil
  211. }
  212. ou.IsPublic = public
  213. _, err = x.Id(ou.Id).AllCols().Update(ou)
  214. return err
  215. }
  216. // AddOrgUser adds new user to given organization.
  217. func AddOrgUser(orgId, uid int64) error {
  218. if IsOrganizationMember(orgId, uid) {
  219. return nil
  220. }
  221. sess := x.NewSession()
  222. defer sess.Close()
  223. if err := sess.Begin(); err != nil {
  224. return err
  225. }
  226. ou := &OrgUser{
  227. Uid: uid,
  228. OrgId: orgId,
  229. }
  230. if _, err := sess.Insert(ou); err != nil {
  231. sess.Rollback()
  232. return err
  233. } else if _, err = sess.Exec("UPDATE `user` SET num_members = num_members + 1 WHERE id = ?", orgId); err != nil {
  234. sess.Rollback()
  235. return err
  236. }
  237. return sess.Commit()
  238. }
  239. // RemoveOrgUser removes user from given organization.
  240. func RemoveOrgUser(orgId, uid int64) error {
  241. ou := new(OrgUser)
  242. has, err := x.Where("uid=?", uid).And("org_id=?", orgId).Get(ou)
  243. if err != nil {
  244. return err
  245. } else if !has {
  246. return nil
  247. }
  248. u, err := GetUserById(uid)
  249. if err != nil {
  250. return err
  251. }
  252. org, err := GetUserById(orgId)
  253. if err != nil {
  254. return err
  255. }
  256. // Check if the user to delete is the last member in owner team.
  257. if IsOrganizationOwner(orgId, uid) {
  258. t, err := org.GetOwnerTeam()
  259. if err != nil {
  260. return err
  261. }
  262. if t.NumMembers == 1 {
  263. return ErrLastOrgOwner
  264. }
  265. }
  266. sess := x.NewSession()
  267. defer sess.Close()
  268. if err := sess.Begin(); err != nil {
  269. return err
  270. }
  271. if _, err := sess.Id(ou.Id).Delete(ou); err != nil {
  272. sess.Rollback()
  273. return err
  274. } else if _, err = sess.Exec("UPDATE `user` SET num_members = num_members - 1 WHERE id = ?", orgId); err != nil {
  275. sess.Rollback()
  276. return err
  277. }
  278. // Delete all repository accesses.
  279. if err = org.GetRepositories(); err != nil {
  280. sess.Rollback()
  281. return err
  282. }
  283. access := &Access{
  284. UserName: u.LowerName,
  285. }
  286. for _, repo := range org.Repos {
  287. access.RepoName = path.Join(org.LowerName, repo.LowerName)
  288. if _, err = sess.Delete(access); err != nil {
  289. sess.Rollback()
  290. return err
  291. }
  292. }
  293. // Delete member in his/her teams.
  294. ts, err := GetUserTeams(org.Id, u.Id)
  295. if err != nil {
  296. return err
  297. }
  298. for _, t := range ts {
  299. if err = removeTeamMemberWithSess(org.Id, t.Id, u.Id, sess); err != nil {
  300. return err
  301. }
  302. }
  303. return sess.Commit()
  304. }
  305. // ___________
  306. // \__ ___/___ _____ _____
  307. // | |_/ __ \\__ \ / \
  308. // | |\ ___/ / __ \| Y Y \
  309. // |____| \___ >____ /__|_| /
  310. // \/ \/ \/
  311. type AuthorizeType int
  312. const (
  313. ORG_READABLE AuthorizeType = iota + 1
  314. ORG_WRITABLE
  315. ORG_ADMIN
  316. )
  317. const OWNER_TEAM = "Owners"
  318. // Team represents a organization team.
  319. type Team struct {
  320. Id int64
  321. OrgId int64 `xorm:"INDEX"`
  322. LowerName string
  323. Name string
  324. Description string
  325. Authorize AuthorizeType
  326. RepoIds string `xorm:"TEXT"`
  327. Repos []*Repository `xorm:"-"`
  328. Members []*User `xorm:"-"`
  329. NumRepos int
  330. NumMembers int
  331. }
  332. // IsOwnerTeam returns true if team is owner team.
  333. func (t *Team) IsOwnerTeam() bool {
  334. return t.Name == OWNER_TEAM
  335. }
  336. // IsTeamMember returns true if given user is a member of team.
  337. func (t *Team) IsMember(uid int64) bool {
  338. return IsTeamMember(t.OrgId, t.Id, uid)
  339. }
  340. // GetRepositories returns all repositories in team of organization.
  341. func (t *Team) GetRepositories() error {
  342. idStrs := strings.Split(t.RepoIds, "|")
  343. t.Repos = make([]*Repository, 0, len(idStrs))
  344. for _, str := range idStrs {
  345. if len(str) == 0 {
  346. continue
  347. }
  348. id := com.StrTo(str[1:]).MustInt64()
  349. if id == 0 {
  350. continue
  351. }
  352. repo, err := GetRepositoryById(id)
  353. if err != nil {
  354. return err
  355. }
  356. t.Repos = append(t.Repos, repo)
  357. }
  358. return nil
  359. }
  360. // GetMembers returns all members in team of organization.
  361. func (t *Team) GetMembers() (err error) {
  362. t.Members, err = GetTeamMembers(t.OrgId, t.Id)
  363. return err
  364. }
  365. // AddMember adds new member to team of organization.
  366. func (t *Team) AddMember(uid int64) error {
  367. return AddTeamMember(t.OrgId, t.Id, uid)
  368. }
  369. // RemoveMember removes member from team of organization.
  370. func (t *Team) RemoveMember(uid int64) error {
  371. return RemoveTeamMember(t.OrgId, t.Id, uid)
  372. }
  373. // NewTeam creates a record of new team.
  374. // It's caller's responsibility to assign organization ID.
  375. func NewTeam(t *Team) error {
  376. if !IsLegalName(t.Name) {
  377. return ErrTeamNameIllegal
  378. }
  379. has, err := x.Id(t.OrgId).Get(new(User))
  380. if err != nil {
  381. return err
  382. } else if !has {
  383. return ErrOrgNotExist
  384. }
  385. t.LowerName = strings.ToLower(t.Name)
  386. has, err = x.Where("org_id=?", t.OrgId).And("lower_name=?", t.LowerName).Get(new(Team))
  387. if err != nil {
  388. return err
  389. } else if has {
  390. return ErrTeamAlreadyExist
  391. }
  392. sess := x.NewSession()
  393. defer sess.Close()
  394. if err = sess.Begin(); err != nil {
  395. return err
  396. }
  397. if _, err = sess.Insert(t); err != nil {
  398. sess.Rollback()
  399. return err
  400. }
  401. // Update organization number of teams.
  402. if _, err = sess.Exec("UPDATE `user` SET num_teams = num_teams + 1 WHERE id = ?", t.OrgId); err != nil {
  403. sess.Rollback()
  404. return err
  405. }
  406. return sess.Commit()
  407. }
  408. // GetTeam returns team by given team name and organization.
  409. func GetTeam(orgId int64, name string) (*Team, error) {
  410. t := &Team{
  411. OrgId: orgId,
  412. LowerName: strings.ToLower(name),
  413. }
  414. has, err := x.Get(t)
  415. if err != nil {
  416. return nil, err
  417. } else if !has {
  418. return nil, ErrTeamNotExist
  419. }
  420. return t, nil
  421. }
  422. // GetTeamById returns team by given ID.
  423. func GetTeamById(teamId int64) (*Team, error) {
  424. t := new(Team)
  425. has, err := x.Id(teamId).Get(t)
  426. if err != nil {
  427. return nil, err
  428. } else if !has {
  429. return nil, ErrTeamNotExist
  430. }
  431. return t, nil
  432. }
  433. // GetHighestAuthorize returns highest repository authorize level for given user and team.
  434. func GetHighestAuthorize(orgId, uid, teamId, repoId int64) (AuthorizeType, error) {
  435. ts, err := GetUserTeams(orgId, uid)
  436. if err != nil {
  437. return 0, err
  438. }
  439. var auth AuthorizeType = 0
  440. for _, t := range ts {
  441. // Not current team and has given repository.
  442. if t.Id != teamId && strings.Contains(t.RepoIds, "$"+com.ToStr(repoId)+"|") {
  443. // Fast return.
  444. if t.Authorize == ORG_WRITABLE {
  445. return ORG_WRITABLE, nil
  446. }
  447. if t.Authorize > auth {
  448. auth = t.Authorize
  449. }
  450. }
  451. }
  452. return auth, nil
  453. }
  454. // UpdateTeam updates information of team.
  455. func UpdateTeam(t *Team, authChanged bool) (err error) {
  456. if !IsLegalName(t.Name) {
  457. return ErrTeamNameIllegal
  458. }
  459. if len(t.Description) > 255 {
  460. t.Description = t.Description[:255]
  461. }
  462. sess := x.NewSession()
  463. defer sess.Close()
  464. if err = sess.Begin(); err != nil {
  465. return err
  466. }
  467. // Update access for team members if needed.
  468. if authChanged && !t.IsOwnerTeam() {
  469. if err = t.GetRepositories(); err != nil {
  470. return err
  471. } else if err = t.GetMembers(); err != nil {
  472. return err
  473. }
  474. // Get organization.
  475. org, err := GetUserById(t.OrgId)
  476. if err != nil {
  477. return err
  478. }
  479. mode := READABLE
  480. if t.Authorize > ORG_READABLE {
  481. mode = WRITABLE
  482. }
  483. access := &Access{
  484. Mode: mode,
  485. }
  486. for _, repo := range t.Repos {
  487. access.RepoName = path.Join(org.LowerName, repo.LowerName)
  488. for _, u := range t.Members {
  489. // ORG_WRITABLE is the highest authorize level for now.
  490. // Skip checking others if current team has this level.
  491. if t.Authorize < ORG_WRITABLE {
  492. auth, err := GetHighestAuthorize(org.Id, u.Id, t.Id, repo.Id)
  493. if err != nil {
  494. sess.Rollback()
  495. return err
  496. }
  497. if auth >= t.Authorize {
  498. continue // Other team has higher or same authorize level.
  499. }
  500. }
  501. access.UserName = u.LowerName
  502. if _, err = sess.Update(access); err != nil {
  503. sess.Rollback()
  504. return err
  505. }
  506. }
  507. }
  508. }
  509. t.LowerName = strings.ToLower(t.Name)
  510. if _, err = sess.Id(t.Id).AllCols().Update(t); err != nil {
  511. sess.Rollback()
  512. return err
  513. }
  514. return sess.Commit()
  515. }
  516. // DeleteTeam deletes given team.
  517. // It's caller's responsibility to assign organization ID.
  518. func DeleteTeam(t *Team) error {
  519. if err := t.GetRepositories(); err != nil {
  520. return err
  521. } else if err = t.GetMembers(); err != nil {
  522. return err
  523. }
  524. // Get organization.
  525. org, err := GetUserById(t.OrgId)
  526. if err != nil {
  527. return err
  528. }
  529. sess := x.NewSession()
  530. defer sess.Close()
  531. if err = sess.Begin(); err != nil {
  532. return err
  533. }
  534. // Delete all accesses.
  535. mode := READABLE
  536. if t.Authorize > ORG_READABLE {
  537. mode = WRITABLE
  538. }
  539. access := new(Access)
  540. for _, repo := range t.Repos {
  541. access.RepoName = path.Join(org.LowerName, repo.LowerName)
  542. for _, u := range t.Members {
  543. access.UserName = u.LowerName
  544. access.Mode = mode
  545. auth, err := GetHighestAuthorize(org.Id, u.Id, t.Id, repo.Id)
  546. if err != nil {
  547. sess.Rollback()
  548. return err
  549. }
  550. if auth == 0 {
  551. if _, err = sess.Delete(access); err != nil {
  552. sess.Rollback()
  553. return err
  554. }
  555. } else if auth < t.Authorize {
  556. // Downgrade authorize level.
  557. mode := READABLE
  558. if auth > ORG_READABLE {
  559. mode = WRITABLE
  560. }
  561. access.Mode = mode
  562. if _, err = sess.Update(access); err != nil {
  563. sess.Rollback()
  564. return err
  565. }
  566. }
  567. }
  568. }
  569. // Delete team-user.
  570. if _, err = sess.Where("org_id=?", org.Id).Where("team_id=?", t.Id).Delete(new(TeamUser)); err != nil {
  571. sess.Rollback()
  572. return err
  573. }
  574. // Delete team.
  575. if _, err = sess.Id(t.Id).Delete(new(Team)); err != nil {
  576. sess.Rollback()
  577. return err
  578. }
  579. // Update organization number of teams.
  580. if _, err = sess.Exec("UPDATE `user` SET num_teams = num_teams - 1 WHERE id = ?", t.OrgId); err != nil {
  581. sess.Rollback()
  582. return err
  583. }
  584. return sess.Commit()
  585. }
  586. // ___________ ____ ___
  587. // \__ ___/___ _____ _____ | | \______ ___________
  588. // | |_/ __ \\__ \ / \| | / ___// __ \_ __ \
  589. // | |\ ___/ / __ \| Y Y \ | /\___ \\ ___/| | \/
  590. // |____| \___ >____ /__|_| /______//____ >\___ >__|
  591. // \/ \/ \/ \/ \/
  592. // TeamUser represents an team-user relation.
  593. type TeamUser struct {
  594. Id int64
  595. Uid int64
  596. OrgId int64 `xorm:"INDEX"`
  597. TeamId int64
  598. }
  599. // IsTeamMember returns true if given user is a member of team.
  600. func IsTeamMember(orgId, teamId, uid int64) bool {
  601. has, _ := x.Where("uid=?", uid).And("org_id=?", orgId).And("team_id=?", teamId).Get(new(TeamUser))
  602. return has
  603. }
  604. // GetTeamMembers returns all members in given team of organization.
  605. func GetTeamMembers(orgId, teamId int64) ([]*User, error) {
  606. tus := make([]*TeamUser, 0, 10)
  607. err := x.Where("org_id=?", orgId).And("team_id=?", teamId).Find(&tus)
  608. if err != nil {
  609. return nil, err
  610. }
  611. us := make([]*User, len(tus))
  612. for i, tu := range tus {
  613. us[i], err = GetUserById(tu.Uid)
  614. if err != nil {
  615. return nil, err
  616. }
  617. }
  618. return us, nil
  619. }
  620. // GetUserTeams returns all teams that user belongs to in given origanization.
  621. func GetUserTeams(orgId, uid int64) ([]*Team, error) {
  622. tus := make([]*TeamUser, 0, 5)
  623. if err := x.Where("uid=?", uid).And("org_id=?", orgId).Find(&tus); err != nil {
  624. return nil, err
  625. }
  626. ts := make([]*Team, len(tus))
  627. for i, tu := range tus {
  628. t := new(Team)
  629. has, err := x.Id(tu.TeamId).Get(t)
  630. if err != nil {
  631. return nil, err
  632. } else if !has {
  633. return nil, ErrTeamNotExist
  634. }
  635. ts[i] = t
  636. }
  637. return ts, nil
  638. }
  639. // AddTeamMember adds new member to given team of given organization.
  640. func AddTeamMember(orgId, teamId, uid int64) error {
  641. if IsTeamMember(orgId, teamId, uid) {
  642. return nil
  643. }
  644. if err := AddOrgUser(orgId, uid); err != nil {
  645. return err
  646. }
  647. // Get team and its repositories.
  648. t, err := GetTeamById(teamId)
  649. if err != nil {
  650. return err
  651. }
  652. t.NumMembers++
  653. if err = t.GetRepositories(); err != nil {
  654. return err
  655. }
  656. // Get organization.
  657. org, err := GetUserById(orgId)
  658. if err != nil {
  659. return err
  660. }
  661. // Get user.
  662. u, err := GetUserById(uid)
  663. if err != nil {
  664. return err
  665. }
  666. sess := x.NewSession()
  667. defer sess.Close()
  668. if err = sess.Begin(); err != nil {
  669. return err
  670. }
  671. tu := &TeamUser{
  672. Uid: uid,
  673. OrgId: orgId,
  674. TeamId: teamId,
  675. }
  676. mode := READABLE
  677. if t.Authorize > ORG_READABLE {
  678. mode = WRITABLE
  679. }
  680. access := &Access{
  681. UserName: u.LowerName,
  682. Mode: mode,
  683. }
  684. if _, err = sess.Insert(tu); err != nil {
  685. sess.Rollback()
  686. return err
  687. } else if _, err = sess.Id(t.Id).Update(t); err != nil {
  688. sess.Rollback()
  689. return err
  690. }
  691. // Give access to team repositories.
  692. for _, repo := range t.Repos {
  693. auth, err := GetHighestAuthorize(orgId, uid, teamId, repo.Id)
  694. if err != nil {
  695. sess.Rollback()
  696. return err
  697. }
  698. access.Id = 0
  699. access.RepoName = path.Join(org.LowerName, repo.LowerName)
  700. // Equal 0 means given access doesn't exist.
  701. if auth == 0 {
  702. if _, err = sess.Insert(access); err != nil {
  703. sess.Rollback()
  704. return err
  705. }
  706. } else if auth < t.Authorize {
  707. if _, err = sess.Update(access); err != nil {
  708. sess.Rollback()
  709. return err
  710. }
  711. }
  712. }
  713. fmt.Println("kao")
  714. // We make sure it exists before.
  715. ou := new(OrgUser)
  716. _, err = sess.Where("uid=?", uid).And("org_id=?", orgId).Get(ou)
  717. if err != nil {
  718. sess.Rollback()
  719. return err
  720. }
  721. ou.NumTeams++
  722. if t.IsOwnerTeam() {
  723. ou.IsOwner = true
  724. }
  725. if _, err = sess.Id(ou.Id).AllCols().Update(ou); err != nil {
  726. sess.Rollback()
  727. return err
  728. }
  729. return sess.Commit()
  730. }
  731. func removeTeamMemberWithSess(orgId, teamId, uid int64, sess *xorm.Session) error {
  732. if !IsTeamMember(orgId, teamId, uid) {
  733. return nil
  734. }
  735. // Get team and its repositories.
  736. t, err := GetTeamById(teamId)
  737. if err != nil {
  738. return err
  739. }
  740. // Check if the user to delete is the last member in owner team.
  741. if t.IsOwnerTeam() && t.NumMembers == 1 {
  742. return ErrLastOrgOwner
  743. }
  744. t.NumMembers--
  745. if err = t.GetRepositories(); err != nil {
  746. return err
  747. }
  748. // Get organization.
  749. org, err := GetUserById(orgId)
  750. if err != nil {
  751. return err
  752. }
  753. // Get user.
  754. u, err := GetUserById(uid)
  755. if err != nil {
  756. return err
  757. }
  758. tu := &TeamUser{
  759. Uid: uid,
  760. OrgId: orgId,
  761. TeamId: teamId,
  762. }
  763. if _, err := sess.Delete(tu); err != nil {
  764. sess.Rollback()
  765. return err
  766. } else if _, err = sess.Id(t.Id).AllCols().Update(t); err != nil {
  767. sess.Rollback()
  768. return err
  769. }
  770. // Delete access to team repositories.
  771. access := &Access{
  772. UserName: u.LowerName,
  773. }
  774. for _, repo := range t.Repos {
  775. auth, err := GetHighestAuthorize(orgId, uid, teamId, repo.Id)
  776. if err != nil {
  777. sess.Rollback()
  778. return err
  779. }
  780. // Delete access if this is the last team user belongs to.
  781. if auth == 0 {
  782. access.RepoName = path.Join(org.LowerName, repo.LowerName)
  783. _, err = sess.Delete(access)
  784. } else if auth < t.Authorize {
  785. // Downgrade authorize level.
  786. mode := READABLE
  787. if auth > ORG_READABLE {
  788. mode = WRITABLE
  789. }
  790. access.Mode = mode
  791. _, err = sess.Update(access)
  792. }
  793. if err != nil {
  794. sess.Rollback()
  795. return err
  796. }
  797. }
  798. // This must exist.
  799. ou := new(OrgUser)
  800. _, err = sess.Where("uid=?", uid).And("org_id=?", org.Id).Get(ou)
  801. if err != nil {
  802. sess.Rollback()
  803. return err
  804. }
  805. ou.NumTeams--
  806. if t.IsOwnerTeam() {
  807. ou.IsOwner = false
  808. }
  809. if _, err = sess.Id(ou.Id).AllCols().Update(ou); err != nil {
  810. sess.Rollback()
  811. return err
  812. }
  813. return nil
  814. }
  815. // RemoveTeamMember removes member from given team of given organization.
  816. func RemoveTeamMember(orgId, teamId, uid int64) error {
  817. sess := x.NewSession()
  818. defer sess.Close()
  819. if err := sess.Begin(); err != nil {
  820. return err
  821. }
  822. if err := removeTeamMemberWithSess(orgId, teamId, uid, sess); err != nil {
  823. return err
  824. }
  825. return sess.Commit()
  826. }