parent
a30e5bcaf8
commit
237193ef2a
@ -1,4 +1,5 @@ |
||||
gogs |
||||
*.exe |
||||
*.exe~ |
||||
.DS_Store |
||||
.DS_Store |
||||
*.db |
||||
|
@ -0,0 +1,119 @@ |
||||
package models |
||||
|
||||
import ( |
||||
"os" |
||||
"path/filepath" |
||||
"strings" |
||||
"time" |
||||
|
||||
git "github.com/libgit2/git2go" |
||||
) |
||||
|
||||
type Repo struct { |
||||
Id int64 |
||||
OwnerId int64 `xorm:"unique(s)"` |
||||
ForkId int64 |
||||
LowerName string `xorm:"unique(s) index not null"` |
||||
Name string `xorm:"index not null"` |
||||
NumWatchs int |
||||
NumStars int |
||||
NumForks int |
||||
Created time.Time `xorm:"created"` |
||||
Updated time.Time `xorm:"updated"` |
||||
} |
||||
|
||||
// check if repository is exist
|
||||
func IsRepositoryExist(user *User, reposName string) (bool, error) { |
||||
repo := Repo{OwnerId: user.Id} |
||||
// TODO: get repository by nocase name
|
||||
return orm.Where("lower_name = ?", strings.ToLower(reposName)).Get(&repo) |
||||
} |
||||
|
||||
//
|
||||
// create a repository for a user or orgnaziation
|
||||
//
|
||||
func CreateRepository(user *User, reposName string) (*Repo, error) { |
||||
p := filepath.Join(root, user.Name) |
||||
os.MkdirAll(p, os.ModePerm) |
||||
f := filepath.Join(p, reposName) |
||||
_, err := git.InitRepository(f, false) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
repo := Repo{OwnerId: user.Id, Name: reposName} |
||||
session := orm.NewSession() |
||||
defer session.Close() |
||||
session.Begin() |
||||
_, err = session.Insert(&repo) |
||||
if err != nil { |
||||
os.RemoveAll(f) |
||||
session.Rollback() |
||||
return nil, err |
||||
} |
||||
_, err = session.Exec("update user set num_repos = num_repos + 1 where id = ?", user.Id) |
||||
if err != nil { |
||||
os.RemoveAll(f) |
||||
session.Rollback() |
||||
return nil, err |
||||
} |
||||
err = session.Commit() |
||||
if err != nil { |
||||
os.RemoveAll(f) |
||||
session.Rollback() |
||||
return nil, err |
||||
} |
||||
return &repo, nil |
||||
} |
||||
|
||||
// list one user's repository
|
||||
func GetRepositories(user *User) ([]Repo, error) { |
||||
repos := make([]Repo, 0) |
||||
err := orm.Find(&repos, &Repo{OwnerId: user.Id}) |
||||
return repos, err |
||||
} |
||||
|
||||
func StarReposiory(user *User, repoName string) error { |
||||
return nil |
||||
} |
||||
|
||||
func UnStarRepository() { |
||||
|
||||
} |
||||
|
||||
func WatchRepository() { |
||||
|
||||
} |
||||
|
||||
func UnWatchRepository() { |
||||
|
||||
} |
||||
|
||||
//
|
||||
// delete a repository for a user or orgnaztion
|
||||
//
|
||||
func DeleteRepository(user *User, reposName string) error { |
||||
session := orm.NewSession() |
||||
_, err := session.Delete(&Repo{OwnerId: user.Id, Name: reposName}) |
||||
if err != nil { |
||||
session.Rollback() |
||||
return err |
||||
} |
||||
_, err = session.Exec("update user set num_repos = num_repos - 1 where id = ?", user.Id) |
||||
if err != nil { |
||||
session.Rollback() |
||||
return err |
||||
} |
||||
err = session.Commit() |
||||
if err != nil { |
||||
session.Rollback() |
||||
return err |
||||
} |
||||
|
||||
err = os.RemoveAll(filepath.Join(root, user.Name, reposName)) |
||||
if err != nil { |
||||
// TODO: log and delete manully
|
||||
return err |
||||
} |
||||
return nil |
||||
} |
@ -0,0 +1,148 @@ |
||||
package models |
||||
|
||||
import ( |
||||
"errors" |
||||
"fmt" |
||||
"strings" |
||||
"time" |
||||
|
||||
"github.com/dchest/scrypt" |
||||
) // user type
|
||||
const ( |
||||
Individual = iota + 1 |
||||
Organization |
||||
) |
||||
|
||||
// login type
|
||||
const ( |
||||
Plain = iota + 1 |
||||
LDAP |
||||
) |
||||
|
||||
type User struct { |
||||
Id int64 |
||||
LowerName string `xorm:"unique not null"` |
||||
Name string `xorm:"unique not null"` |
||||
Email string `xorm:"unique not null"` |
||||
Passwd string `xorm:"not null"` |
||||
LoginType int |
||||
Type int |
||||
NumFollowers int |
||||
NumFollowings int |
||||
NumStars int |
||||
NumRepos int |
||||
Avatar string `xorm:"varchar(2048) not null"` |
||||
Created time.Time `xorm:"created"` |
||||
Updated time.Time `xorm:"updated"` |
||||
} |
||||
|
||||
type Follow struct { |
||||
Id int64 |
||||
UserId int64 `xorm:"unique(s)"` |
||||
FollowId int64 `xorm:"unique(s)"` |
||||
Created time.Time `xorm:"created"` |
||||
} |
||||
|
||||
const ( |
||||
OpCreateRepo = iota + 1 |
||||
OpDeleteRepo |
||||
OpStarRepo |
||||
OpFollowRepo |
||||
OpCommitRepo |
||||
OpPullRequest |
||||
) |
||||
|
||||
type Action struct { |
||||
Id int64 |
||||
UserId int64 |
||||
OpType int |
||||
RepoId int64 |
||||
Content string |
||||
Created time.Time `xorm:"created"` |
||||
} |
||||
|
||||
var ( |
||||
ErrUserNotExist = errors.New("User not exist") |
||||
) |
||||
|
||||
// user's name should be noncased unique
|
||||
func IsUserExist(name string) (bool, error) { |
||||
return orm.Get(&User{LowerName: strings.ToLower(name)}) |
||||
} |
||||
|
||||
func RegisterUser(user *User) error { |
||||
_, err := orm.Insert(user) |
||||
return err |
||||
} |
||||
|
||||
func UpdateUser(user *User) error { |
||||
_, err := orm.Id(user.Id).Update(user) |
||||
return err |
||||
} |
||||
|
||||
func (user *User) EncodePasswd(pass string) error { |
||||
newPasswd, err := scrypt.Key([]byte(user.Passwd), []byte("!#@FDEWREWR&*("), 16384, 8, 1, 64) |
||||
user.Passwd = fmt.Sprintf("%x", newPasswd) |
||||
return err |
||||
} |
||||
|
||||
func LoginUserPlain(name, passwd string) (*User, error) { |
||||
user := User{Name: name} |
||||
err := user.EncodePasswd(passwd) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
has, err := orm.Get(&user) |
||||
if !has { |
||||
err = ErrUserNotExist |
||||
} |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return &user, nil |
||||
} |
||||
|
||||
func FollowUser(userId int64, followId int64) error { |
||||
session := orm.NewSession() |
||||
defer session.Close() |
||||
session.Begin() |
||||
_, err := session.Insert(&Follow{UserId: userId, FollowId: followId}) |
||||
if err != nil { |
||||
session.Rollback() |
||||
return err |
||||
} |
||||
_, err = session.Exec("update user set num_followers = num_followers + 1 where id = ?", followId) |
||||
if err != nil { |
||||
session.Rollback() |
||||
return err |
||||
} |
||||
_, err = session.Exec("update user set num_followings = num_followings + 1 where id = ?", userId) |
||||
if err != nil { |
||||
session.Rollback() |
||||
return err |
||||
} |
||||
return session.Commit() |
||||
} |
||||
|
||||
func UnFollowUser(userId int64, unFollowId int64) error { |
||||
session := orm.NewSession() |
||||
defer session.Close() |
||||
session.Begin() |
||||
_, err := session.Delete(&Follow{UserId: userId, FollowId: unFollowId}) |
||||
if err != nil { |
||||
session.Rollback() |
||||
return err |
||||
} |
||||
_, err = session.Exec("update user set num_followers = num_followers - 1 where id = ?", unFollowId) |
||||
if err != nil { |
||||
session.Rollback() |
||||
return err |
||||
} |
||||
_, err = session.Exec("update user set num_followings = num_followings - 1 where id = ?", userId) |
||||
if err != nil { |
||||
session.Rollback() |
||||
return err |
||||
} |
||||
return session.Commit() |
||||
} |
Loading…
Reference in new issue