mirror of https://github.com/gogits/gogs.git
Browse Source
Collaborators have write access as default, and can be changed via repository collaboration settings page to change between read, write and admin.pull/2766/head
Unknwon
9 years ago
16 changed files with 292 additions and 164 deletions
@ -0,0 +1,161 @@
|
||||
// Copyright 2016 The Gogs Authors. All rights reserved.
|
||||
// Use of this source code is governed by a MIT-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package models |
||||
|
||||
import ( |
||||
"fmt" |
||||
"time" |
||||
) |
||||
|
||||
// Collaboration represent the relation between an individual and a repository.
|
||||
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" |
||||
} |
||||
|
||||
// AddCollaborator adds new collaboration relation between an individual and a repository.
|
||||
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}) |
||||
} |
||||
|
||||
// Collaborator represents a user with collaboration details.
|
||||
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 |
||||
} |
||||
|
||||
// GetCollaborators returns the collaborators for a repository
|
||||
func (repo *Repository) GetCollaborators() ([]*Collaborator, error) { |
||||
return repo.getCollaborators(x) |
||||
} |
||||
|
||||
// ChangeCollaborationAccessMode sets new access mode for the collaboration.
|
||||
func (repo *Repository) ChangeCollaborationAccessMode(uid int64, mode AccessMode) error { |
||||
// Discard invalid input
|
||||
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() |
||||
} |
||||
|
||||
// DeleteCollaboration removes collaboration relation between the user and repository.
|
||||
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() |
||||
} |
||||
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
Loading…
Reference in new issue