2014-02-18 00:38:50 +01:00
|
|
|
// Copyright 2014 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.
|
|
|
|
|
2014-02-17 16:57:23 +01:00
|
|
|
package models
|
|
|
|
|
|
|
|
import (
|
|
|
|
"strings"
|
|
|
|
"time"
|
2014-04-05 00:55:17 +02:00
|
|
|
|
2014-04-18 15:35:09 +02:00
|
|
|
"github.com/go-xorm/xorm"
|
2014-02-17 16:57:23 +01:00
|
|
|
)
|
|
|
|
|
2014-06-25 06:44:48 +02:00
|
|
|
type AccessType int
|
|
|
|
|
2014-02-17 16:57:23 +01:00
|
|
|
const (
|
2014-06-25 06:44:48 +02:00
|
|
|
READABLE AccessType = iota + 1
|
|
|
|
WRITABLE
|
2014-02-17 16:57:23 +01:00
|
|
|
)
|
|
|
|
|
2014-03-27 16:37:33 +01:00
|
|
|
// Access represents the accessibility of user to repository.
|
2014-02-17 16:57:23 +01:00
|
|
|
type Access struct {
|
|
|
|
Id int64
|
2014-08-26 12:11:15 +02:00
|
|
|
UserName string `xorm:"UNIQUE(s)"`
|
|
|
|
RepoName string `xorm:"UNIQUE(s)"` // <user name>/<repo name>
|
|
|
|
Mode AccessType `xorm:"UNIQUE(s)"`
|
|
|
|
Created time.Time `xorm:"CREATED"`
|
2014-02-17 16:57:23 +01:00
|
|
|
}
|
|
|
|
|
2014-03-17 19:03:58 +01:00
|
|
|
// AddAccess adds new access record.
|
2014-02-17 16:57:23 +01:00
|
|
|
func AddAccess(access *Access) error {
|
2014-03-30 22:01:50 +02:00
|
|
|
access.UserName = strings.ToLower(access.UserName)
|
|
|
|
access.RepoName = strings.ToLower(access.RepoName)
|
2014-06-21 06:51:41 +02:00
|
|
|
_, err := x.Insert(access)
|
2014-02-17 16:57:23 +01:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2014-04-03 21:50:55 +02:00
|
|
|
// UpdateAccess updates access information.
|
|
|
|
func UpdateAccess(access *Access) error {
|
|
|
|
access.UserName = strings.ToLower(access.UserName)
|
|
|
|
access.RepoName = strings.ToLower(access.RepoName)
|
2014-06-21 06:51:41 +02:00
|
|
|
_, err := x.Id(access.Id).Update(access)
|
2014-04-03 21:50:55 +02:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2014-05-01 17:32:12 +02:00
|
|
|
// DeleteAccess deletes access record.
|
|
|
|
func DeleteAccess(access *Access) error {
|
2014-06-21 06:51:41 +02:00
|
|
|
_, err := x.Delete(access)
|
2014-05-01 17:32:12 +02:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2014-04-05 00:55:17 +02:00
|
|
|
// UpdateAccess updates access information with session for rolling back.
|
|
|
|
func UpdateAccessWithSession(sess *xorm.Session, access *Access) error {
|
|
|
|
if _, err := sess.Id(access.Id).Update(access); err != nil {
|
|
|
|
sess.Rollback()
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2014-03-27 16:37:33 +01:00
|
|
|
// HasAccess returns true if someone can read or write to given repository.
|
2014-05-08 18:24:11 +02:00
|
|
|
// The repoName should be in format <username>/<reponame>.
|
2014-06-25 06:44:48 +02:00
|
|
|
func HasAccess(uname, repoName string, mode AccessType) (bool, error) {
|
2014-05-09 01:17:43 +02:00
|
|
|
if len(repoName) == 0 {
|
|
|
|
return false, nil
|
|
|
|
}
|
2014-04-12 03:47:39 +02:00
|
|
|
access := &Access{
|
2014-05-08 18:24:11 +02:00
|
|
|
UserName: strings.ToLower(uname),
|
2014-02-18 00:38:50 +01:00
|
|
|
RepoName: strings.ToLower(repoName),
|
2014-04-12 03:47:39 +02:00
|
|
|
}
|
2014-06-21 06:51:41 +02:00
|
|
|
has, err := x.Get(access)
|
2014-04-12 03:47:39 +02:00
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
} else if !has {
|
|
|
|
return false, nil
|
|
|
|
} else if mode > access.Mode {
|
|
|
|
return false, nil
|
|
|
|
}
|
|
|
|
return true, nil
|
2014-02-17 16:57:23 +01:00
|
|
|
}
|
2015-01-23 08:54:16 +01:00
|
|
|
|
|
|
|
// GetAccessibleRepositories finds all repositories where a user has access to,
|
|
|
|
// besides his own.
|
|
|
|
func (u *User) GetAccessibleRepositories() (map[*Repository]AccessType, error) {
|
|
|
|
accesses := make([]*Access, 0, 10)
|
|
|
|
if err := x.Find(&accesses, &Access{UserName: u.LowerName}); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
repos := make(map[*Repository]AccessType, len(accesses))
|
|
|
|
for _, access := range accesses {
|
|
|
|
repo, err := GetRepositoryByRef(access.RepoName)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2015-02-04 15:08:55 +01:00
|
|
|
err = repo.GetOwner()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2015-01-23 08:54:16 +01:00
|
|
|
repos[repo] = access.Mode
|
|
|
|
}
|
|
|
|
|
|
|
|
return repos, nil
|
|
|
|
}
|