From a922c3ff6a65d6d0550f36d866a301a6737ca8a2 Mon Sep 17 00:00:00 2001
From: Unknown <joe2010xtmf@163.com>
Date: Wed, 19 Mar 2014 23:20:55 -0400
Subject: Watch backend

---
 models/models.go |  3 ++-
 models/repo.go   | 19 ++++++++++++++-----
 2 files changed, 16 insertions(+), 6 deletions(-)

(limited to 'models')

diff --git a/models/models.go b/models/models.go
index d6a6200e..2e0bb759 100644
--- a/models/models.go
+++ b/models/models.go
@@ -88,7 +88,8 @@ func setEngine() {
 
 func init() {
 	setEngine()
-	if err := orm.Sync(new(User), new(PublicKey), new(Repository), new(Access), new(Action)); err != nil {
+	if err := orm.Sync(new(User), new(PublicKey), new(Repository), new(Access),
+		new(Action), new(Watch)); err != nil {
 		fmt.Printf("sync database struct error: %v\n", err)
 		os.Exit(2)
 	}
diff --git a/models/repo.go b/models/repo.go
index fce7d7f5..187862fe 100644
--- a/models/repo.go
+++ b/models/repo.go
@@ -43,11 +43,20 @@ type Repository struct {
 	Updated     time.Time `xorm:"updated"`
 }
 
-type Star struct {
-	Id      int64
-	RepoId  int64
-	UserId  int64
-	Created time.Time `xorm:"created"`
+// Watch is connection request for receiving repository notifycation.
+type Watch struct {
+	Id     int64
+	RepoId int64 `xorm:"UNIQUE(watch)"`
+	UserId int64 `xorm:"UNIQUE(watch)"`
+}
+
+func WatchRepo(userId, repoId int64, watch bool) (err error) {
+	if watch {
+		_, err = orm.Insert(&Watch{RepoId: repoId, UserId: userId})
+	} else {
+		_, err = orm.Delete(&Watch{0, repoId, userId})
+	}
+	return err
 }
 
 var (
-- 
cgit v1.2.3