gitea-1120-rc1/models/access.go

81 lines
1.9 KiB
Go
Raw Normal View History

2014-02-18 07:38:50 +08: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 23:57:23 +08:00
package models
import (
"strings"
"time"
2014-04-05 06:55:17 +08:00
2014-04-18 21:35:09 +08:00
"github.com/go-xorm/xorm"
2014-02-17 23:57:23 +08:00
)
2014-06-25 12:44:48 +08:00
type AccessType int
2014-02-17 23:57:23 +08:00
const (
2014-06-25 12:44:48 +08:00
READABLE AccessType = iota + 1
WRITABLE
2014-02-17 23:57:23 +08:00
)
2014-03-27 23:37:33 +08:00
// Access represents the accessibility of user to repository.
2014-02-17 23:57:23 +08:00
type Access struct {
Id int64
2014-06-25 12:44:48 +08: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 23:57:23 +08:00
}
// AddAccess adds new access record.
2014-02-17 23:57:23 +08:00
func AddAccess(access *Access) error {
2014-03-31 04:01:50 +08:00
access.UserName = strings.ToLower(access.UserName)
access.RepoName = strings.ToLower(access.RepoName)
2014-06-21 12:51:41 +08:00
_, err := x.Insert(access)
2014-02-17 23:57:23 +08:00
return err
}
2014-04-04 03:50:55 +08: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 12:51:41 +08:00
_, err := x.Id(access.Id).Update(access)
2014-04-04 03:50:55 +08:00
return err
}
2014-05-01 23:32:12 +08:00
// DeleteAccess deletes access record.
func DeleteAccess(access *Access) error {
2014-06-21 12:51:41 +08:00
_, err := x.Delete(access)
2014-05-01 23:32:12 +08:00
return err
}
2014-04-05 06:55:17 +08: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 23:37:33 +08:00
// HasAccess returns true if someone can read or write to given repository.
2014-05-09 00:24:11 +08:00
// The repoName should be in format <username>/<reponame>.
2014-06-25 12:44:48 +08:00
func HasAccess(uname, repoName string, mode AccessType) (bool, error) {
if len(repoName) == 0 {
return false, nil
}
2014-04-12 09:47:39 +08:00
access := &Access{
2014-05-09 00:24:11 +08:00
UserName: strings.ToLower(uname),
2014-02-18 07:38:50 +08:00
RepoName: strings.ToLower(repoName),
2014-04-12 09:47:39 +08:00
}
2014-06-21 12:51:41 +08:00
has, err := x.Get(access)
2014-04-12 09:47:39 +08: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 23:57:23 +08:00
}