diff options
Diffstat (limited to 'models')
| -rw-r--r-- | models/access.go | 41 | ||||
| -rw-r--r-- | models/action.go | 139 | ||||
| -rw-r--r-- | models/issue.go | 171 | ||||
| -rw-r--r-- | models/models.go | 95 | ||||
| -rw-r--r-- | models/models_test.go | 50 | ||||
| -rw-r--r-- | models/publickey.go | 228 | ||||
| -rw-r--r-- | models/repo.go | 744 | ||||
| -rw-r--r-- | models/user.go | 373 |
8 files changed, 1841 insertions, 0 deletions
diff --git a/models/access.go b/models/access.go new file mode 100644 index 00000000..36d9405f --- /dev/null +++ b/models/access.go @@ -0,0 +1,41 @@ +// 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. + +package models + +import ( + "strings" + "time" +) + +// Access types. +const ( + AU_READABLE = iota + 1 + AU_WRITABLE +) + +// Access represents the accessibility of user and repository. +type Access struct { + Id int64 + UserName string `xorm:"unique(s)"` + RepoName string `xorm:"unique(s)"` + Mode int `xorm:"unique(s)"` + Created time.Time `xorm:"created"` +} + +// AddAccess adds new access record. +func AddAccess(access *Access) error { + _, err := orm.Insert(access) + return err +} + +// HasAccess returns true if someone can read or write given repository. +func HasAccess(userName, repoName string, mode int) (bool, error) { + return orm.Get(&Access{ + Id: 0, + UserName: strings.ToLower(userName), + RepoName: strings.ToLower(repoName), + Mode: mode, + }) +} diff --git a/models/action.go b/models/action.go new file mode 100644 index 00000000..44d7aea8 --- /dev/null +++ b/models/action.go @@ -0,0 +1,139 @@ +// 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. + +package models + +import ( + "encoding/json" + "time" + + "github.com/gogits/gogs/modules/base" + "github.com/gogits/gogs/modules/log" +) + +// Operation types of user action. +const ( + OP_CREATE_REPO = iota + 1 + OP_DELETE_REPO + OP_STAR_REPO + OP_FOLLOW_REPO + OP_COMMIT_REPO + OP_PULL_REQUEST +) + +// Action represents user operation type and information to the repository. +type Action struct { + Id int64 + UserId int64 // Receiver user id. + OpType int // Operations: CREATE DELETE STAR ... + ActUserId int64 // Action user id. + ActUserName string // Action user name. + RepoId int64 + RepoName string + RefName string + Content string `xorm:"TEXT"` + Created time.Time `xorm:"created"` +} + +func (a Action) GetOpType() int { + return a.OpType +} + +func (a Action) GetActUserName() string { + return a.ActUserName +} + +func (a Action) GetRepoName() string { + return a.RepoName +} + +func (a Action) GetBranch() string { + return a.RefName +} + +func (a Action) GetContent() string { + return a.Content +} + +// CommitRepoAction records action for commit repository. +func CommitRepoAction(userId int64, userName string, + repoId int64, repoName string, refName string, commits *base.PushCommits) error { + log.Trace("action.CommitRepoAction: %d/%s", userId, repoName) + + bs, err := json.Marshal(commits) + if err != nil { + log.Error("action.CommitRepoAction(json): %d/%s", userId, repoName) + return err + } + + // Add feeds for user self and all watchers. + watches, err := GetWatches(repoId) + if err != nil { + log.Error("action.CommitRepoAction(get watches): %d/%s", userId, repoName) + return err + } + watches = append(watches, Watch{UserId: userId}) + + for i := range watches { + if userId == watches[i].UserId && i > 0 { + continue // Do not add twice in case author watches his/her repository. + } + + _, err = orm.InsertOne(&Action{ + UserId: watches[i].UserId, + ActUserId: userId, + ActUserName: userName, + OpType: OP_COMMIT_REPO, + Content: string(bs), + RepoId: repoId, + RepoName: repoName, + RefName: refName, + }) + if err != nil { + log.Error("action.CommitRepoAction(notify watches): %d/%s", userId, repoName) + } + return err + } + + // Update repository last update time. + repo, err := GetRepositoryByName(userId, repoName) + if err != nil { + log.Error("action.CommitRepoAction(GetRepositoryByName): %d/%s", userId, repoName) + return err + } + repo.IsBare = false + if err = UpdateRepository(repo); err != nil { + log.Error("action.CommitRepoAction(UpdateRepository): %d/%s", userId, repoName) + return err + } + return nil +} + +// NewRepoAction records action for create repository. +func NewRepoAction(user *User, repo *Repository) error { + _, err := orm.InsertOne(&Action{ + UserId: user.Id, + ActUserId: user.Id, + ActUserName: user.Name, + OpType: OP_CREATE_REPO, + RepoId: repo.Id, + RepoName: repo.Name, + }) + + log.Trace("action.NewRepoAction: %s/%s", user.LowerName, repo.LowerName) + return err +} + +// GetFeeds returns action list of given user in given context. +func GetFeeds(userid, offset int64, isProfile bool) ([]Action, error) { + actions := make([]Action, 0, 20) + sess := orm.Limit(20, int(offset)).Desc("id").Where("user_id=?", userid) + if isProfile { + sess.And("act_user_id=?", userid) + } else { + sess.And("act_user_id!=?", userid) + } + err := sess.Find(&actions) + return actions, err +} diff --git a/models/issue.go b/models/issue.go new file mode 100644 index 00000000..fe43a94b --- /dev/null +++ b/models/issue.go @@ -0,0 +1,171 @@ +// 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. + +package models + +import ( + "errors" + "strings" + "time" + + "github.com/gogits/gogs/modules/base" +) + +var ( + ErrIssueNotExist = errors.New("Issue does not exist") +) + +// Issue represents an issue or pull request of repository. +type Issue struct { + Id int64 + Index int64 // Index in one repository. + Name string + RepoId int64 `xorm:"index"` + PosterId int64 + MilestoneId int64 + AssigneeId int64 + IsPull bool // Indicates whether is a pull request or not. + IsClosed bool + Labels string `xorm:"TEXT"` + Mentions string `xorm:"TEXT"` + Content string `xorm:"TEXT"` + NumComments int + Created time.Time `xorm:"created"` + Updated time.Time `xorm:"updated"` +} + +// CreateIssue creates new issue for repository. +func CreateIssue(userId, repoId, milestoneId, assigneeId int64, name, labels, content string, isPull bool) (*Issue, error) { + count, err := GetIssueCount(repoId) + if err != nil { + return nil, err + } + + // TODO: find out mentions + mentions := "" + + issue := &Issue{ + Index: count + 1, + Name: name, + RepoId: repoId, + PosterId: userId, + MilestoneId: milestoneId, + AssigneeId: assigneeId, + IsPull: isPull, + Labels: labels, + Mentions: mentions, + Content: content, + } + _, err = orm.Insert(issue) + // TODO: newIssueAction + return issue, err +} + +// GetIssueCount returns count of issues in the repository. +func GetIssueCount(repoId int64) (int64, error) { + return orm.Count(&Issue{RepoId: repoId}) +} + +// GetIssueById returns issue object by given id. +func GetIssueByIndex(repoId, index int64) (*Issue, error) { + issue := &Issue{RepoId: repoId, Index: index} + has, err := orm.Get(issue) + if err != nil { + return nil, err + } else if !has { + return nil, ErrIssueNotExist + } + return issue, nil +} + +// GetIssues returns a list of issues by given conditions. +func GetIssues(userId, repoId, posterId, milestoneId int64, page int, isClosed, isMention bool, labels, sortType string) ([]Issue, error) { + sess := orm.Limit(20, (page-1)*20) + + if repoId > 0 { + sess.Where("repo_id=?", repoId).And("is_closed=?", isClosed) + } else { + sess.Where("is_closed=?", isClosed) + } + + if userId > 0 { + sess.And("assignee_id=?", userId) + } else if posterId > 0 { + sess.And("poster_id=?", posterId) + } else if isMention { + sess.And("mentions like '%$" + base.ToStr(userId) + "|%'") + } + + if milestoneId > 0 { + sess.And("milestone_id=?", milestoneId) + } + + if len(labels) > 0 { + for _, label := range strings.Split(labels, ",") { + sess.And("mentions like '%$" + label + "|%'") + } + } + + switch sortType { + case "oldest": + sess.Asc("created") + case "recentupdate": + sess.Desc("updated") + case "leastupdate": + sess.Asc("updated") + case "mostcomment": + sess.Desc("num_comments") + case "leastcomment": + sess.Asc("num_comments") + default: + sess.Desc("created") + } + + var issues []Issue + err := sess.Find(&issues) + return issues, err +} + +// UpdateIssue updates information of issue. +func UpdateIssue(issue *Issue) error { + _, err := orm.Update(issue, &Issue{RepoId: issue.RepoId, Index: issue.Index}) + return err +} + +func CloseIssue() { +} + +func ReopenIssue() { +} + +// Label represents a list of labels of repository for issues. +type Label struct { + Id int64 + RepoId int64 `xorm:"index"` + Names string + Colors string +} + +// Milestone represents a milestone of repository. +type Milestone struct { + Id int64 + Name string + RepoId int64 `xorm:"index"` + IsClosed bool + Content string + NumIssues int + DueDate time.Time + Created time.Time `xorm:"created"` +} + +// Comment represents a comment in commit and issue page. +type Comment struct { + Id int64 + PosterId int64 + IssueId int64 + CommitId int64 + Line int + Content string + Created time.Time `xorm:"created"` +} diff --git a/models/models.go b/models/models.go new file mode 100644 index 00000000..ad19a929 --- /dev/null +++ b/models/models.go @@ -0,0 +1,95 @@ +// 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. + +package models + +import ( + "fmt" + "os" + "path" + + _ "github.com/go-sql-driver/mysql" + _ "github.com/lib/pq" + "github.com/lunny/xorm" + + "github.com/gogits/gogs/modules/base" +) + +var ( + orm *xorm.Engine + + DbCfg struct { + Type, Host, Name, User, Pwd, Path, SslMode string + } +) + +func LoadModelsConfig() { + DbCfg.Type = base.Cfg.MustValue("database", "DB_TYPE") + DbCfg.Host = base.Cfg.MustValue("database", "HOST") + DbCfg.Name = base.Cfg.MustValue("database", "NAME") + DbCfg.User = base.Cfg.MustValue("database", "USER") + DbCfg.Pwd = base.Cfg.MustValue("database", "PASSWD") + DbCfg.SslMode = base.Cfg.MustValue("database", "SSL_MODE") + DbCfg.Path = base.Cfg.MustValue("database", "PATH", "data/gogs.db") +} + +func setEngine() { + var err error + switch DbCfg.Type { + case "mysql": + orm, err = xorm.NewEngine("mysql", fmt.Sprintf("%s:%s@%s/%s?charset=utf8", + DbCfg.User, DbCfg.Pwd, DbCfg.Host, DbCfg.Name)) + case "postgres": + orm, err = xorm.NewEngine("postgres", fmt.Sprintf("user=%s password=%s dbname=%s sslmode=%s", + DbCfg.User, DbCfg.Pwd, DbCfg.Name, DbCfg.SslMode)) + case "sqlite3": + os.MkdirAll(path.Dir(DbCfg.Path), os.ModePerm) + orm, err = xorm.NewEngine("sqlite3", DbCfg.Path) + default: + fmt.Printf("Unknown database type: %s\n", DbCfg.Type) + os.Exit(2) + } + if err != nil { + fmt.Printf("models.init(fail to conntect database): %v\n", err) + os.Exit(2) + } + + // WARNNING: for serv command, MUST remove the output to os.stdout, + // so use log file to instead print to stdout. + + //x.ShowDebug = true + //orm.ShowErr = true + f, err := os.Create("xorm.log") + if err != nil { + fmt.Printf("models.init(fail to create xorm.log): %v\n", err) + os.Exit(2) + } + orm.Logger = f + orm.ShowSQL = true +} + +func NewEngine() { + setEngine() + if err := orm.Sync(new(User), new(PublicKey), new(Repository), new(Watch), + new(Action), new(Access), new(Issue)); err != nil { + fmt.Printf("sync database struct error: %v\n", err) + os.Exit(2) + } +} + +type Statistic struct { + Counter struct { + User, PublicKey, Repo, Watch, Action, Access int64 + } +} + +func GetStatistic() (stats Statistic) { + stats.Counter.User, _ = orm.Count(new(User)) + stats.Counter.PublicKey, _ = orm.Count(new(PublicKey)) + stats.Counter.Repo, _ = orm.Count(new(Repository)) + stats.Counter.Watch, _ = orm.Count(new(Watch)) + stats.Counter.Action, _ = orm.Count(new(Action)) + stats.Counter.Access, _ = orm.Count(new(Access)) + return +} diff --git a/models/models_test.go b/models/models_test.go new file mode 100644 index 00000000..d0f734d6 --- /dev/null +++ b/models/models_test.go @@ -0,0 +1,50 @@ +// 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. + +package models + +import ( + "fmt" + "testing" + + "github.com/lunny/xorm" + _ "github.com/mattn/go-sqlite3" +) + +func init() { + LoadModelsConfig() + NewEngine() + + var err error + orm, err = xorm.NewEngine("sqlite3", "./test.db") + if err != nil { + fmt.Println(err) + } + + orm.ShowSQL = true + orm.ShowDebug = true + + err = orm.Sync(&User{}, &Repo{}) + if err != nil { + fmt.Println(err) + } + + root = "test" +} + +func TestCreateRepository(t *testing.T) { + user := User{Id: 1, Type: Individual} + _, err := CreateRepository(&user, "test") + if err != nil { + t.Error(err) + } +} + +func TestDeleteRepository(t *testing.T) { + user := User{Id: 1, Type: Individual} + err := DeleteRepository(&user, "test") + if err != nil { + t.Error(err) + } +} diff --git a/models/publickey.go b/models/publickey.go new file mode 100644 index 00000000..3f2fcabd --- /dev/null +++ b/models/publickey.go @@ -0,0 +1,228 @@ +// 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. + +package models + +import ( + "bufio" + "errors" + "fmt" + "io" + "io/ioutil" + "os" + "os/exec" + "path" + "path/filepath" + "strings" + "sync" + "time" + + "github.com/Unknwon/com" + + "github.com/gogits/gogs/modules/log" +) + +const ( + // "### autogenerated by gitgos, DO NOT EDIT\n" + TPL_PUBLICK_KEY = `command="%s serv key-%d",no-port-forwarding,no-X11-forwarding,no-agent-forwarding,no-pty %s` +) + +var ( + ErrKeyAlreadyExist = errors.New("Public key already exist") +) + +var sshOpLocker = sync.Mutex{} + +var ( + sshPath string + appPath string +) + +// exePath returns the executable path. +func exePath() (string, error) { + file, err := exec.LookPath(os.Args[0]) + if err != nil { + return "", err + } + return filepath.Abs(file) +} + +// homeDir returns the home directory of current user. +func homeDir() string { + home, err := com.HomeDir() + if err != nil { + return "/" + } + return home +} + +func init() { + var err error + + appPath, err = exePath() + if err != nil { + fmt.Printf("publickey.init(fail to get app path): %v\n", err) + os.Exit(2) + } + + // Determine and create .ssh path. + sshPath = filepath.Join(homeDir(), ".ssh") + if err = os.MkdirAll(sshPath, os.ModePerm); err != nil { + fmt.Printf("publickey.init(fail to create sshPath(%s)): %v\n", sshPath, err) + os.Exit(2) + } +} + +// PublicKey represents a SSH key of user. +type PublicKey struct { + Id int64 + OwnerId int64 `xorm:"index"` + Name string `xorm:"unique not null"` + Fingerprint string + Content string `xorm:"TEXT not null"` + Created time.Time `xorm:"created"` + Updated time.Time `xorm:"updated"` +} + +// GenAuthorizedKey returns formatted public key string. +func GenAuthorizedKey(keyId int64, key string) string { + return fmt.Sprintf(TPL_PUBLICK_KEY+"\n", appPath, keyId, key) +} + +// AddPublicKey adds new public key to database and SSH key file. +func AddPublicKey(key *PublicKey) (err error) { + // Check if public key name has been used. + has, err := orm.Get(key) + if err != nil { + return err + } else if has { + return ErrKeyAlreadyExist + } + + // Calculate fingerprint. + tmpPath := strings.Replace(filepath.Join(os.TempDir(), fmt.Sprintf("%d", time.Now().Nanosecond()), + "id_rsa.pub"), "\\", "/", -1) + os.MkdirAll(path.Dir(tmpPath), os.ModePerm) + if err = ioutil.WriteFile(tmpPath, []byte(key.Content), os.ModePerm); err != nil { + return err + } + stdout, _, err := com.ExecCmd("ssh-keygen", "-l", "-f", tmpPath) + if err != nil { + return err + } else if len(stdout) < 2 { + return errors.New("Not enough output for calculating fingerprint") + } + key.Fingerprint = strings.Split(stdout, " ")[1] + + // Save SSH key. + if _, err = orm.Insert(key); err != nil { + return err + } + if err = SaveAuthorizedKeyFile(key); err != nil { + if _, err2 := orm.Delete(key); err2 != nil { + return err2 + } + return err + } + + return nil +} + +func rewriteAuthorizedKeys(key *PublicKey, p, tmpP string) error { + // Delete SSH key in SSH key file. + sshOpLocker.Lock() + defer sshOpLocker.Unlock() + + fr, err := os.Open(p) + if err != nil { + return err + } + defer fr.Close() + + fw, err := os.Create(tmpP) + if err != nil { + return err + } + defer fw.Close() + + buf := bufio.NewReader(fr) + for { + line, errRead := buf.ReadString('\n') + line = strings.TrimSpace(line) + + if errRead != nil { + if errRead != io.EOF { + return errRead + } + + // Reached end of file, if nothing to read then break, + // otherwise handle the last line. + if len(line) == 0 { + break + } + } + + // Found the line and copy rest of file. + if strings.Contains(line, fmt.Sprintf("key-%d", key.Id)) && strings.Contains(line, key.Content) { + continue + } + // Still finding the line, copy the line that currently read. + if _, err = fw.WriteString(line + "\n"); err != nil { + return err + } + + if errRead == io.EOF { + break + } + } + return nil +} + +// DeletePublicKey deletes SSH key information both in database and authorized_keys file. +func DeletePublicKey(key *PublicKey) (err error) { + // Delete SSH key in database. + has, err := orm.Id(key.Id).Get(key) + if err != nil { + return err + } else if !has { + return errors.New("Public key does not exist") + } + if _, err = orm.Delete(key); err != nil { + return err + } + + p := filepath.Join(sshPath, "authorized_keys") + tmpP := filepath.Join(sshPath, "authorized_keys.tmp") + log.Trace("ssh.DeletePublicKey(authorized_keys): %s", p) + + if err = rewriteAuthorizedKeys(key, p, tmpP); err != nil { + return err + } else if err = os.Remove(p); err != nil { + return err + } + return os.Rename(tmpP, p) +} + +// ListPublicKey returns a list of public keys that user has. +func ListPublicKey(userId int64) ([]PublicKey, error) { + keys := make([]PublicKey, 0) + err := orm.Find(&keys, &PublicKey{OwnerId: userId}) + return keys, err +} + +// SaveAuthorizedKeyFile writes SSH key content to SSH key file. +func SaveAuthorizedKeyFile(key *PublicKey) error { + sshOpLocker.Lock() + defer sshOpLocker.Unlock() + + p := filepath.Join(sshPath, "authorized_keys") + f, err := os.OpenFile(p, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0600) + if err != nil { + return err + } + defer f.Close() + + _, err = f.WriteString(GenAuthorizedKey(key.Id, key.Content)) + return err +} diff --git a/models/repo.go b/models/repo.go new file mode 100644 index 00000000..d5f9be72 --- /dev/null +++ b/models/repo.go @@ -0,0 +1,744 @@ +// 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. + +package models + +import ( + "container/list" + "errors" + "fmt" + "io/ioutil" + "os" + "os/exec" + "path" + "path/filepath" + "regexp" + "strings" + "sync" + "time" + "unicode/utf8" + + "github.com/Unknwon/cae/zip" + "github.com/Unknwon/com" + + "github.com/gogits/git" + + "github.com/gogits/gogs/modules/base" + "github.com/gogits/gogs/modules/log" +) + +var ( + ErrRepoAlreadyExist = errors.New("Repository already exist") + ErrRepoNotExist = errors.New("Repository does not exist") + ErrRepoFileNotExist = errors.New("Target Repo file does not exist") + ErrRepoNameIllegal = errors.New("Repository name contains illegal characters") + ErrRepoFileNotLoaded = fmt.Errorf("repo file not loaded") +) + +var gitInitLocker = sync.Mutex{} + +var ( + LanguageIgns, Licenses []string +) + +func LoadRepoConfig() { + LanguageIgns = strings.Split(base.Cfg.MustValue("repository", "LANG_IGNS"), "|") + Licenses = strings.Split(base.Cfg.MustValue("repository", "LICENSES"), "|") +} + +func NewRepoContext() { + zip.Verbose = false + + // Check if server has basic git setting. + stdout, _, err := com.ExecCmd("git", "config", "--get", "user.name") + if err != nil { + fmt.Printf("repo.init(fail to get git user.name): %v", err) + os.Exit(2) + } else if len(stdout) == 0 { + if _, _, err = com.ExecCmd("git", "config", "--global", "user.email", "gogitservice@gmail.com"); err != nil { + fmt.Printf("repo.init(fail to set git user.email): %v", err) + os.Exit(2) + } else if _, _, err = com.ExecCmd("git", "config", "--global", "user.name", "Gogs"); err != nil { + fmt.Printf("repo.init(fail to set git user.name): %v", err) + os.Exit(2) + } + } + + // Initialize illegal patterns. + for i := range illegalPatterns[1:] { + pattern := "" + for j := range illegalPatterns[i+1] { + pattern += "[" + string(illegalPatterns[i+1][j]-32) + string(illegalPatterns[i+1][j]) + "]" + } + illegalPatterns[i+1] = pattern + } +} + +// Repository represents a git repository. +type Repository struct { + Id int64 + OwnerId int64 `xorm:"unique(s)"` + ForkId int64 + LowerName string `xorm:"unique(s) index not null"` + Name string `xorm:"index not null"` + Description string + Website string + NumWatches int + NumStars int + NumForks int + IsPrivate bool + IsBare bool + Created time.Time `xorm:"created"` + Updated time.Time `xorm:"updated"` +} + +// IsRepositoryExist returns true if the repository with given name under user has already existed. +func IsRepositoryExist(user *User, repoName string) (bool, error) { + repo := Repository{OwnerId: user.Id} + has, err := orm.Where("lower_name = ?", strings.ToLower(repoName)).Get(&repo) + if err != nil { + return has, err + } + s, err := os.Stat(RepoPath(user.Name, repoName)) + if err != nil { + return false, nil // Error simply means does not exist, but we don't want to show up. + } + return s.IsDir(), nil +} + +var ( + // Define as all lower case!! + illegalPatterns = []string{"[.][Gg][Ii][Tt]", "raw", "user", "help", "stars", "issues", "pulls", "commits", "admin", "repo", "template", "admin"} +) + +// IsLegalName returns false if name contains illegal characters. +func IsLegalName(repoName string) bool { + for _, pattern := range illegalPatterns { + has, _ := regexp.MatchString(pattern, repoName) + if has { + return false + } + } + return true +} + +// CreateRepository creates a repository for given user or orgnaziation. +func CreateRepository(user *User, repoName, desc, repoLang, license string, private bool, initReadme bool) (*Repository, error) { + if !IsLegalName(repoName) { + return nil, ErrRepoNameIllegal + } + + isExist, err := IsRepositoryExist(user, repoName) + if err != nil { + return nil, err + } else if isExist { + return nil, ErrRepoAlreadyExist + } + + repo := &Repository{ + OwnerId: user.Id, + Name: repoName, + LowerName: strings.ToLower(repoName), + Description: desc, + IsPrivate: private, + IsBare: repoLang == "" && license == "" && !initReadme, + } + + repoPath := RepoPath(user.Name, repoName) + if err = initRepository(repoPath, user, repo, initReadme, repoLang, license); err != nil { + return nil, err + } + session := orm.NewSession() + defer session.Close() + session.Begin() + + if _, err = session.Insert(repo); err != nil { + if err2 := os.RemoveAll(repoPath); err2 != nil { + log.Error("repo.CreateRepository(repo): %v", err) + return nil, errors.New(fmt.Sprintf( + "delete repo directory %s/%s failed(1): %v", user.Name, repoName, err2)) + } + session.Rollback() + return nil, err + } + + access := Access{ + UserName: user.Name, + RepoName: repo.Name, + Mode: AU_WRITABLE, + } + if _, err = session.Insert(&access); err != nil { + session.Rollback() + if err2 := os.RemoveAll(repoPath); err2 != nil { + log.Error("repo.CreateRepository(access): %v", err) + return nil, errors.New(fmt.Sprintf( + "delete repo directory %s/%s failed(2): %v", user.Name, repoName, err2)) + } + return nil, err + } + + rawSql := "UPDATE `user` SET num_repos = num_repos + 1 WHERE id = ?" + if _, err = session.Exec(rawSql, user.Id); err != nil { + session.Rollback() + if err2 := os.RemoveAll(repoPath); err2 != nil { + log.Error("repo.CreateRepository(repo count): %v", err) + return nil, errors.New(fmt.Sprintf( + "delete repo directory %s/%s failed(3): %v", user.Name, repoName, err2)) + } + return nil, err + } + + if err = session.Commit(); err != nil { + session.Rollback() + if err2 := os.RemoveAll(repoPath); err2 != nil { + log.Error("repo.CreateRepository(commit): %v", err) + return nil, errors.New(fmt.Sprintf( + "delete repo directory %s/%s failed(3): %v", user.Name, repoName, err2)) + } + return nil, err + } + + c := exec.Command("git", "update-server-info") + c.Dir = repoPath + err = c.Run() + if err != nil { + log.Error("repo.CreateRepository(exec update-server-info): %v", err) + } + + return repo, NewRepoAction(user, repo) +} + +// extractGitBareZip extracts git-bare.zip to repository path. +func extractGitBareZip(repoPath string) error { + z, err := zip.Open("conf/content/git-bare.zip") + if err != nil { + fmt.Println("shi?") + return err + } + defer z.Close() + + return z.ExtractTo(repoPath) +} + +// initRepoCommit temporarily changes with work directory. +func initRepoCommit(tmpPath string, sig *git.Signature) error { + gitInitLocker.Lock() + defer gitInitLocker.Unlock() + + // Change work directory. + curPath, err := os.Getwd() + if err != nil { + return err + } else if err = os.Chdir(tmpPath); err != nil { + return err + } + defer os.Chdir(curPath) + + var stderr string + if _, stderr, err = com.ExecCmd("git", "add", "--all"); err != nil { + return err + } + log.Info("stderr(1): %s", stderr) + if _, stderr, err = com.ExecCmd("git", "commit", fmt.Sprintf("--author='%s <%s>'", sig.Name, sig.Email), + "-m", "Init commit"); err != nil { + return err + } + log.Info("stderr(2): %s", stderr) + if _, stderr, err = com.ExecCmd("git", "push", "origin", "master"); err != nil { + return err + } + log.Info("stderr(3): %s", stderr) + return nil +} + +// InitRepository initializes README and .gitignore if needed. +func initRepository(f string, user *User, repo *Repository, initReadme bool, repoLang, license string) error { + repoPath := RepoPath(user.Name, repo.Name) + + // Create bare new repository. + if err := extractGitBareZip(repoPath); err != nil { + return err + } + + /* + // hook/post-update + pu, err := os.OpenFile(filepath.Join(repoPath, "hooks", "post-update"), os.O_CREATE|os.O_WRONLY, 0777) + if err != nil { + return err + } + defer pu.Close() + // TODO: Windows .bat + if _, err = pu.WriteString(fmt.Sprintf("#!/usr/bin/env bash\n%s update\n", appPath)); err != nil { + return err + } + + // hook/post-update + pu2, err := os.OpenFile(filepath.Join(repoPath, "hooks", "post-receive"), os.O_CREATE|os.O_WRONLY, 0777) + if err != nil { + return err + } + defer pu2.Close() + // TODO: Windows .bat + if _, err = pu2.WriteString("#!/usr/bin/env bash\ngit update-server-info\n"); err != nil { + return err + } + */ + + // Initialize repository according to user's choice. + fileName := map[string]string{} + if initReadme { + fileName["readme"] = "README.md" + } + if repoLang != "" { + fileName["gitign"] = ".gitignore" + } + if license != "" { + fileName["license"] = "LICENSE" + } + + // Clone to temprory path and do the init commit. + tmpDir := filepath.Join(os.TempDir(), fmt.Sprintf("%d", time.Now().Nanosecond())) + os.MkdirAll(tmpDir, os.ModePerm) + + if _, _, err := com.ExecCmd("git", "clone", repoPath, tmpDir); err != nil { + return err + } + + // README + if initReadme { + defaultReadme := repo.Name + "\n" + strings.Repeat("=", + utf8.RuneCountInString(repo.Name)) + "\n\n" + repo.Description + if err := ioutil.WriteFile(filepath.Join(tmpDir, fileName["readme"]), + []byte(defaultReadme), 0644); err != nil { + return err + } + } + + // .gitignore + if repoLang != "" { + filePath := "conf/gitignore/" + repoLang + if com.IsFile(filePath) { + if _, err := com.Copy(filePath, + filepath.Join(tmpDir, fileName["gitign"])); err != nil { + return err + } + } + } + + // LICENSE + if license != "" { + filePath := "conf/license/" + license + if com.IsFile(filePath) { + if _, err := com.Copy(filePath, + filepath.Join(tmpDir, fileName["license"])); err != nil { + return err + } + } + } + + if len(fileName) == 0 { + return nil + } + + // Apply changes and commit. + if err := initRepoCommit(tmpDir, user.NewGitSig()); err != nil { + return err + } + return nil +} + +// UserRepo reporesents a repository with user name. +type UserRepo struct { + *Repository + UserName string +} + +// GetRepos returns given number of repository objects with offset. +func GetRepos(num, offset int) ([]UserRepo, error) { + repos := make([]Repository, 0, num) + if err := orm.Limit(num, offset).Asc("id").Find(&repos); err != nil { + return nil, err + } + + urepos := make([]UserRepo, len(repos)) + for i := range repos { + urepos[i].Repository = &repos[i] + u := new(User) + has, err := orm.Id(urepos[i].Repository.OwnerId).Get(u) + if err != nil { + return nil, err + } else if !has { + return nil, ErrUserNotExist + } + urepos[i].UserName = u.Name + } + + return urepos, nil +} + +func RepoPath(userName, repoName string) string { + return filepath.Join(UserPath(userName), repoName+".git") +} + +func UpdateRepository(repo *Repository) error { + if len(repo.Description) > 255 { + repo.Description = repo.Description[:255] + } + if len(repo.Website) > 255 { + repo.Website = repo.Website[:255] + } + + _, err := orm.Id(repo.Id).AllCols().Update(repo) + return err +} + +// DeleteRepository deletes a repository for a user or orgnaztion. +func DeleteRepository(userId, repoId int64, userName string) (err error) { + repo := &Repository{Id: repoId, OwnerId: userId} + has, err := orm.Get(repo) + if err != nil { + return err + } else if !has { + return ErrRepoNotExist + } + + session := orm.NewSession() + if err = session.Begin(); err != nil { + return err + } + if _, err = session.Delete(&Repository{Id: repoId}); err != nil { + session.Rollback() + return err + } + if _, err := session.Delete(&Access{UserName: userName, RepoName: repo.Name}); err != nil { + session.Rollback() + return err + } + rawSql := "UPDATE `user` SET num_repos = num_repos - 1 WHERE id = ?" + if _, err = session.Exec(rawSql, userId); err != nil { + session.Rollback() + return err + } + if _, err = session.Delete(&Watch{RepoId: repoId}); err != nil { + session.Rollback() + return err + } + if err = session.Commit(); err != nil { + session.Rollback() + return err + } + if err = os.RemoveAll(RepoPath(userName, repo.Name)); err != nil { + // TODO: log and delete manully + log.Error("delete repo %s/%s failed: %v", userName, repo.Name, err) + return err + } + return nil +} + +// GetRepositoryByName returns the repository by given name under user if exists. +func GetRepositoryByName(userId int64, repoName string) (*Repository, error) { + repo := &Repository{ + OwnerId: userId, + LowerName: strings.ToLower(repoName), + } + has, err := orm.Get(repo) + if err != nil { + return nil, err + } else if !has { + return nil, ErrRepoNotExist + } + return repo, err +} + +// GetRepositoryById returns the repository by given id if exists. +func GetRepositoryById(id int64) (repo *Repository, err error) { + has, err := orm.Id(id).Get(repo) + if err != nil { + return nil, err + } else if !has { + return nil, ErrRepoNotExist + } + return repo, err +} + +// GetRepositories returns the list of repositories of given user. +func GetRepositories(user *User) ([]Repository, error) { + repos := make([]Repository, 0, 10) + err := orm.Desc("updated").Find(&repos, &Repository{OwnerId: user.Id}) + return repos, err +} + +func GetRepositoryCount(user *User) (int64, error) { + return orm.Count(&Repository{OwnerId: user.Id}) +} + +// Watch is connection request for receiving repository notifycation. +type Watch struct { + Id int64 + RepoId int64 `xorm:"UNIQUE(watch)"` + UserId int64 `xorm:"UNIQUE(watch)"` +} + +// Watch or unwatch repository. +func WatchRepo(userId, repoId int64, watch bool) (err error) { + if watch { + if _, err = orm.Insert(&Watch{RepoId: repoId, UserId: userId}); err != nil { + return err + } + + rawSql := "UPDATE `repository` SET num_watches = num_watches + 1 WHERE id = ?" + _, err = orm.Exec(rawSql, repoId) + } else { + if _, err = orm.Delete(&Watch{0, repoId, userId}); err != nil { + return err + } + rawSql := "UPDATE `repository` SET num_watches = num_watches - 1 WHERE id = ?" + _, err = orm.Exec(rawSql, repoId) + } + return err +} + +// GetWatches returns all watches of given repository. +func GetWatches(repoId int64) ([]Watch, error) { + watches := make([]Watch, 0, 10) + err := orm.Find(&watches, &Watch{RepoId: repoId}) + return watches, err +} + +// IsWatching checks if user has watched given repository. +func IsWatching(userId, repoId int64) bool { + has, _ := orm.Get(&Watch{0, repoId, userId}) + return has +} + +func StarReposiory(user *User, repoName string) error { + return nil +} + +func UnStarRepository() { + +} + +func WatchRepository() { + +} + +func UnWatchRepository() { + +} + +func ForkRepository(reposName string, userId int64) { + +} + +// RepoFile represents a file object in git repository. +type RepoFile struct { + *git.TreeEntry + Path string + Size int64 + Repo *git.Repository + Commit *git.Commit +} + +// LookupBlob returns the content of an object. +func (file *RepoFile) LookupBlob() (*git.Blob, error) { + if file.Repo == nil { + return nil, ErrRepoFileNotLoaded + } + + return file.Repo.LookupBlob(file.Id) +} + +// GetBranches returns all branches of given repository. +func GetBranches(userName, reposName string) ([]string, error) { + repo, err := git.OpenRepository(RepoPath(userName, reposName)) + if err != nil { + return nil, err + } + + refs, err := repo.AllReferences() + if err != nil { + return nil, err + } + + brs := make([]string, len(refs)) + for i, ref := range refs { + brs[i] = ref.Name + } + return brs, nil +} + +func GetTargetFile(userName, reposName, branchName, commitId, rpath string) (*RepoFile, error) { + repo, err := git.OpenRepository(RepoPath(userName, reposName)) + if err != nil { + return nil, err + } + + commit, err := repo.GetCommit(branchName, commitId) + if err != nil { + return nil, err + } + + parts := strings.Split(path.Clean(rpath), "/") + + var entry *git.TreeEntry + tree := commit.Tree + for i, part := range parts { + if i == len(parts)-1 { + entry = tree.EntryByName(part) + if entry == nil { + return nil, ErrRepoFileNotExist + } + } else { + tree, err = repo.SubTree(tree, part) + if err != nil { + return nil, err + } + } + } + + size, err := repo.ObjectSize(entry.Id) + if err != nil { + return nil, err + } + + repoFile := &RepoFile{ + entry, + rpath, + size, + repo, + commit, + } + + return repoFile, nil +} + +// GetReposFiles returns a list of file object in given directory of repository. +func GetReposFiles(userName, reposName, branchName, commitId, rpath string) ([]*RepoFile, error) { + repo, err := git.OpenRepository(RepoPath(userName, reposName)) + if err != nil { + return nil, err + } + + commit, err := repo.GetCommit(branchName, commitId) + if err != nil { + return nil, err + } + + var repodirs []*RepoFile + var repofiles []*RepoFile + commit.Tree.Walk(func(dirname string, entry *git.TreeEntry) int { + if dirname == rpath { + // TODO: size get method shoule be improved + size, err := repo.ObjectSize(entry.Id) + if err != nil { + return 0 + } + + var cm = commit + var i int + for { + i = i + 1 + //fmt.Println(".....", i, cm.Id(), cm.ParentCount()) + if cm.ParentCount() == 0 { + break + } else if cm.ParentCount() == 1 { + pt, _ := repo.SubTree(cm.Parent(0).Tree, dirname) + if pt == nil { + break + } + pEntry := pt.EntryByName(entry.Name) + if pEntry == nil || !pEntry.Id.Equal(entry.Id) { + break + } else { + cm = cm.Parent(0) + } + } else { + var emptyCnt = 0 + var sameIdcnt = 0 + var lastSameCm *git.Commit + //fmt.Println(".....", cm.ParentCount()) + for i := 0; i < cm.ParentCount(); i++ { + //fmt.Println("parent", i, cm.Parent(i).Id()) + p := cm.Parent(i) + pt, _ := repo.SubTree(p.Tree, dirname) + var pEntry *git.TreeEntry + if pt != nil { + pEntry = pt.EntryByName(entry.Name) + } + + //fmt.Println("pEntry", pEntry) + + if pEntry == nil { + emptyCnt = emptyCnt + 1 + if emptyCnt+sameIdcnt == cm.ParentCount() { + if lastSameCm == nil { + goto loop + } else { + cm = lastSameCm + break + } + } + } else { + //fmt.Println(i, "pEntry", pEntry.Id, "entry", entry.Id) + if !pEntry.Id.Equal(entry.Id) { + goto loop + } else { + lastSameCm = cm.Parent(i) + sameIdcnt = sameIdcnt + 1 + if emptyCnt+sameIdcnt == cm.ParentCount() { + // TODO: now follow the first parent commit? + cm = lastSameCm + //fmt.Println("sameId...") + break + } + } + } + } + } + } + + loop: + + rp := &RepoFile{ + entry, + path.Join(dirname, entry.Name), + size, + repo, + cm, + } + + if entry.IsFile() { + repofiles = append(repofiles, rp) + } else if entry.IsDir() { + repodirs = append(repodirs, rp) + } + } + return 0 + }) + + return append(repodirs, repofiles...), nil +} + +func GetCommit(userName, repoName, branchname, commitid string) (*git.Commit, error) { + repo, err := git.OpenRepository(RepoPath(userName, repoName)) + if err != nil { + return nil, err + } + + return repo.GetCommit(branchname, commitid) +} + +// GetCommits returns all commits of given branch of repository. +func GetCommits(userName, reposName, branchname string) (*list.List, error) { + repo, err := git.OpenRepository(RepoPath(userName, reposName)) + if err != nil { + return nil, err + } + r, err := repo.LookupReference(fmt.Sprintf("refs/heads/%s", branchname)) + if err != nil { + return nil, err + } + return r.AllCommits() +} diff --git a/models/user.go b/models/user.go new file mode 100644 index 00000000..6ca16ec3 --- /dev/null +++ b/models/user.go @@ -0,0 +1,373 @@ +// 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. + +package models + +import ( + "encoding/hex" + "errors" + "fmt" + "os" + "path/filepath" + "strings" + "time" + + "github.com/dchest/scrypt" + + "github.com/gogits/git" + + "github.com/gogits/gogs/modules/base" + "github.com/gogits/gogs/modules/log" +) + +// User types. +const ( + UT_INDIVIDUAL = iota + 1 + UT_ORGANIZATION +) + +// Login types. +const ( + LT_PLAIN = iota + 1 + LT_LDAP +) + +var ( + ErrUserOwnRepos = errors.New("User still have ownership of repositories") + ErrUserAlreadyExist = errors.New("User already exist") + ErrUserNotExist = errors.New("User does not exist") + ErrEmailAlreadyUsed = errors.New("E-mail already used") + ErrUserNameIllegal = errors.New("User name contains illegal characters") +) + +// User represents the object of individual and member of organization. +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"` + AvatarEmail string `xorm:"not null"` + Location string + Website string + IsActive bool + IsAdmin bool + Rands string `xorm:"VARCHAR(10)"` + Created time.Time `xorm:"created"` + Updated time.Time `xorm:"updated"` +} + +// HomeLink returns the user home page link. +func (user *User) HomeLink() string { + return "/user/" + user.LowerName +} + +// AvatarLink returns the user gravatar link. +func (user *User) AvatarLink() string { + if base.Service.EnableCacheAvatar { + return "/avatar/" + user.Avatar + } + return "http://1.gravatar.com/avatar/" + user.Avatar +} + +// NewGitSig generates and returns the signature of given user. +func (user *User) NewGitSig() *git.Signature { + return &git.Signature{ + Name: user.Name, + Email: user.Email, + When: time.Now(), + } +} + +// EncodePasswd encodes password to safe format. +func (user *User) EncodePasswd() error { + newPasswd, err := scrypt.Key([]byte(user.Passwd), []byte(base.SecretKey), 16384, 8, 1, 64) + user.Passwd = fmt.Sprintf("%x", newPasswd) + return err +} + +// Member represents user is member of organization. +type Member struct { + Id int64 + OrgId int64 `xorm:"unique(member) index"` + UserId int64 `xorm:"unique(member)"` +} + +// IsUserExist checks if given user name exist, +// the user name should be noncased unique. +func IsUserExist(name string) (bool, error) { + return orm.Get(&User{LowerName: strings.ToLower(name)}) +} + +// IsEmailUsed returns true if the e-mail has been used. +func IsEmailUsed(email string) (bool, error) { + return orm.Get(&User{Email: email}) +} + +// return a user salt token +func GetUserSalt() string { + return base.GetRandomString(10) +} + +// RegisterUser creates record of a new user. +func RegisterUser(user *User) (*User, error) { + if !IsLegalName(user.Name) { + return nil, ErrUserNameIllegal + } + + isExist, err := IsUserExist(user.Name) + if err != nil { + return nil, err + } else if isExist { + return nil, ErrUserAlreadyExist + } + + isExist, err = IsEmailUsed(user.Email) + if err != nil { + return nil, err + } else if isExist { + return nil, ErrEmailAlreadyUsed + } + + user.LowerName = strings.ToLower(user.Name) + user.Avatar = base.EncodeMd5(user.Email) + user.AvatarEmail = user.Email + user.Rands = GetUserSalt() + if err = user.EncodePasswd(); err != nil { + return nil, err + } else if _, err = orm.Insert(user); err != nil { + return nil, err + } else if err = os.MkdirAll(UserPath(user.Name), os.ModePerm); err != nil { + if _, err := orm.Id(user.Id).Delete(&User{}); err != nil { + return nil, errors.New(fmt.Sprintf( + "both create userpath %s and delete table record faild: %v", user.Name, err)) + } + return nil, err + } + + if user.Id == 1 { + user.IsAdmin = true + user.IsActive = true + _, err = orm.Id(user.Id).UseBool().Update(user) + } + return user, err +} + +// GetUsers returns given number of user objects with offset. +func GetUsers(num, offset int) ([]User, error) { + users := make([]User, 0, num) + err := orm.Limit(num, offset).Asc("id").Find(&users) + return users, err +} + +// get user by erify code +func getVerifyUser(code string) (user *User) { + if len(code) <= base.TimeLimitCodeLength { + return nil + } + + // use tail hex username query user + hexStr := code[base.TimeLimitCodeLength:] + if b, err := hex.DecodeString(hexStr); err == nil { + if user, err = GetUserByName(string(b)); user != nil { + return user + } + log.Error("user.getVerifyUser: %v", err) + } + + return nil +} + +// verify active code when active account +func VerifyUserActiveCode(code string) (user *User) { + minutes := base.Service.ActiveCodeLives + + if user = getVerifyUser(code); user != nil { + // time limit code + prefix := code[:base.TimeLimitCodeLength] + data := base.ToStr(user.Id) + user.Email + user.LowerName + user.Passwd + user.Rands + + if base.VerifyTimeLimitCode(data, minutes, prefix) { + return user + } + } + return nil +} + +// UpdateUser updates user's information. +func UpdateUser(user *User) (err error) { + if len(user.Location) > 255 { + user.Location = user.Location[:255] + } + if len(user.Website) > 255 { + user.Website = user.Website[:255] + } + + _, err = orm.Id(user.Id).AllCols().Update(user) + return err +} + +// DeleteUser completely deletes everything of the user. +func DeleteUser(user *User) error { + // Check ownership of repository. + count, err := GetRepositoryCount(user) + if err != nil { + return errors.New("modesl.GetRepositories: " + err.Error()) + } else if count > 0 { + return ErrUserOwnRepos + } + + // TODO: check issues, other repos' commits + + // Delete all feeds. + if _, err = orm.Delete(&Action{UserId: user.Id}); err != nil { + return err + } + + // Delete all SSH keys. + keys := make([]PublicKey, 0, 10) + if err = orm.Find(&keys, &PublicKey{OwnerId: user.Id}); err != nil { + return err + } + for _, key := range keys { + if err = DeletePublicKey(&key); err != nil { + return err + } + } + + // Delete user directory. + if err = os.RemoveAll(UserPath(user.Name)); err != nil { + return err + } + + _, err = orm.Delete(user) + // TODO: delete and update follower information. + return err +} + +// UserPath returns the path absolute path of user repositories. +func UserPath(userName string) string { + return filepath.Join(base.RepoRootPath, strings.ToLower(userName)) +} + +func GetUserByKeyId(keyId int64) (*User, error) { + user := new(User) + rawSql := "SELECT a.* FROM `user` AS a, public_key AS b WHERE a.id = b.owner_id AND b.id=?" + has, err := orm.Sql(rawSql, keyId).Get(user) + if err != nil { + return nil, err + } else if !has { + err = errors.New("not exist key owner") + return nil, err + } + return user, nil +} + +// GetUserById returns the user object by given id if exists. +func GetUserById(id int64) (*User, error) { + user := new(User) + has, err := orm.Id(id).Get(user) + if err != nil { + return nil, err + } + if !has { + return nil, ErrUserNotExist + } + return user, nil +} + +// GetUserByName returns the user object by given name if exists. +func GetUserByName(name string) (*User, error) { + if len(name) == 0 { + return nil, ErrUserNotExist + } + user := &User{LowerName: strings.ToLower(name)} + has, err := orm.Get(user) + if err != nil { + return nil, err + } else if !has { + return nil, ErrUserNotExist + } + return user, nil +} + +// LoginUserPlain validates user by raw user name and password. +func LoginUserPlain(name, passwd string) (*User, error) { + user := User{LowerName: strings.ToLower(name), Passwd: passwd} + if err := user.EncodePasswd(); err != nil { + return nil, err + } + + has, err := orm.Get(&user) + if err != nil { + return nil, err + } else if !has { + err = ErrUserNotExist + } + return &user, err +} + +// Follow is connection request for receiving user notifycation. +type Follow struct { + Id int64 + UserId int64 `xorm:"unique(follow)"` + FollowId int64 `xorm:"unique(follow)"` +} + +// FollowUser marks someone be another's follower. +func FollowUser(userId int64, followId int64) (err error) { + session := orm.NewSession() + defer session.Close() + session.Begin() + + if _, err = session.Insert(&Follow{UserId: userId, FollowId: followId}); err != nil { + session.Rollback() + return err + } + + rawSql := "UPDATE `user` SET num_followers = num_followers + 1 WHERE id = ?" + if _, err = session.Exec(rawSql, followId); err != nil { + session.Rollback() + return err + } + + rawSql = "UPDATE `user` SET num_followings = num_followings + 1 WHERE id = ?" + if _, err = session.Exec(rawSql, userId); err != nil { + session.Rollback() + return err + } + return session.Commit() +} + +// UnFollowUser unmarks someone be another's follower. +func UnFollowUser(userId int64, unFollowId int64) (err error) { + session := orm.NewSession() + defer session.Close() + session.Begin() + + if _, err = session.Delete(&Follow{UserId: userId, FollowId: unFollowId}); err != nil { + session.Rollback() + return err + } + + rawSql := "UPDATE `user` SET num_followers = num_followers - 1 WHERE id = ?" + if _, err = session.Exec(rawSql, unFollowId); err != nil { + session.Rollback() + return err + } + + rawSql = "UPDATE `user` SET num_followings = num_followings - 1 WHERE id = ?" + if _, err = session.Exec(rawSql, userId); err != nil { + session.Rollback() + return err + } + return session.Commit() +} |