|
@@ -0,0 +1,161 @@
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+package models
|
|
|
+
|
|
|
+import (
|
|
|
+ "fmt"
|
|
|
+ "time"
|
|
|
+)
|
|
|
+
|
|
|
+
|
|
|
+type Collaboration struct {
|
|
|
+ ID int64 `xorm:"pk autoincr"`
|
|
|
+ RepoID int64 `xorm:"UNIQUE(s) INDEX NOT NULL"`
|
|
|
+ UserID int64 `xorm:"UNIQUE(s) INDEX NOT NULL"`
|
|
|
+ Mode AccessMode `xorm:"DEFAULT 2 NOT NULL"`
|
|
|
+ Created time.Time `xorm:"CREATED"`
|
|
|
+}
|
|
|
+
|
|
|
+func (c *Collaboration) ModeName() string {
|
|
|
+ switch c.Mode {
|
|
|
+ case ACCESS_MODE_READ:
|
|
|
+ return "Read"
|
|
|
+ case ACCESS_MODE_WRITE:
|
|
|
+ return "Write"
|
|
|
+ case ACCESS_MODE_ADMIN:
|
|
|
+ return "Admin"
|
|
|
+ }
|
|
|
+ return "Undefined"
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+func (repo *Repository) AddCollaborator(u *User) error {
|
|
|
+ collaboration := &Collaboration{
|
|
|
+ RepoID: repo.ID,
|
|
|
+ UserID: u.Id,
|
|
|
+ }
|
|
|
+
|
|
|
+ has, err := x.Get(collaboration)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ } else if has {
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+ collaboration.Mode = ACCESS_MODE_WRITE
|
|
|
+
|
|
|
+ sess := x.NewSession()
|
|
|
+ defer sessionRelease(sess)
|
|
|
+ if err = sess.Begin(); err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ if _, err = sess.InsertOne(collaboration); err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ if repo.Owner.IsOrganization() {
|
|
|
+ err = repo.recalculateTeamAccesses(sess, 0)
|
|
|
+ } else {
|
|
|
+ err = repo.recalculateAccesses(sess)
|
|
|
+ }
|
|
|
+ if err != nil {
|
|
|
+ return fmt.Errorf("recalculateAccesses 'team=%v': %v", repo.Owner.IsOrganization(), err)
|
|
|
+ }
|
|
|
+
|
|
|
+ return sess.Commit()
|
|
|
+}
|
|
|
+
|
|
|
+func (repo *Repository) getCollaborations(e Engine) ([]*Collaboration, error) {
|
|
|
+ collaborations := make([]*Collaboration, 0)
|
|
|
+ return collaborations, e.Find(&collaborations, &Collaboration{RepoID: repo.ID})
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+type Collaborator struct {
|
|
|
+ *User
|
|
|
+ Collaboration *Collaboration
|
|
|
+}
|
|
|
+
|
|
|
+func (repo *Repository) getCollaborators(e Engine) ([]*Collaborator, error) {
|
|
|
+ collaborations, err := repo.getCollaborations(e)
|
|
|
+ if err != nil {
|
|
|
+ return nil, fmt.Errorf("getCollaborations: %v", err)
|
|
|
+ }
|
|
|
+
|
|
|
+ collaborators := make([]*Collaborator, len(collaborations))
|
|
|
+ for i, c := range collaborations {
|
|
|
+ user, err := getUserByID(e, c.UserID)
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ collaborators[i] = &Collaborator{
|
|
|
+ User: user,
|
|
|
+ Collaboration: c,
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return collaborators, nil
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+func (repo *Repository) GetCollaborators() ([]*Collaborator, error) {
|
|
|
+ return repo.getCollaborators(x)
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+func (repo *Repository) ChangeCollaborationAccessMode(uid int64, mode AccessMode) error {
|
|
|
+
|
|
|
+ if mode <= ACCESS_MODE_NONE || mode > ACCESS_MODE_OWNER {
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+
|
|
|
+ collaboration := &Collaboration{
|
|
|
+ RepoID: repo.ID,
|
|
|
+ UserID: uid,
|
|
|
+ }
|
|
|
+ has, err := x.Get(collaboration)
|
|
|
+ if err != nil {
|
|
|
+ return fmt.Errorf("get collaboration: %v", err)
|
|
|
+ } else if !has {
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+
|
|
|
+ collaboration.Mode = mode
|
|
|
+
|
|
|
+ sess := x.NewSession()
|
|
|
+ defer sessionRelease(sess)
|
|
|
+ if err = sess.Begin(); err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ if _, err = sess.Id(collaboration.ID).AllCols().Update(collaboration); err != nil {
|
|
|
+ return fmt.Errorf("update collaboration: %v", err)
|
|
|
+ } else if _, err = sess.Exec("UPDATE access SET mode = ? WHERE user_id = ? AND repo_id = ?", mode, uid, repo.ID); err != nil {
|
|
|
+ return fmt.Errorf("update access table: %v", err)
|
|
|
+ }
|
|
|
+
|
|
|
+ return sess.Commit()
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+func (repo *Repository) DeleteCollaboration(uid int64) (err error) {
|
|
|
+ collaboration := &Collaboration{
|
|
|
+ RepoID: repo.ID,
|
|
|
+ UserID: uid,
|
|
|
+ }
|
|
|
+
|
|
|
+ sess := x.NewSession()
|
|
|
+ defer sessionRelease(sess)
|
|
|
+ if err = sess.Begin(); err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ if has, err := sess.Delete(collaboration); err != nil || has == 0 {
|
|
|
+ return err
|
|
|
+ } else if err = repo.recalculateAccesses(sess); err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ return sess.Commit()
|
|
|
+}
|