add createrepository & deleterepository

2 files changed, 96 insertions, 0 deletions

diff --git a/models/models.go b/models/models.go
index 20fc0eb096..d5a8c8cf5c 100644
--- a/models/models.go
+++ b/models/models.go
@@ -3,3 +3,69 @@
 // license that can be found in the LICENSE file.
 
 package models
+
+import (
+	"os"
+	"path/filepath"
+	"github.com/lunny/xorm"
+
+	git "github.com/libgit2/git2go"
+)
+
+const (
+	UserRepo = iota
+	OrgRepo
+)
+
+type User struct {
+	Id   int64
+	Name string `xorm:"unique"`
+}
+
+type Org struct {
+	Id int64
+}
+
+type Repo struct {
+	Id      int64
+	OwnerId int64  `xorm:"unique(s)"`
+	Type    int    `xorm:"unique(s)"`
+	Name    string `xorm:"unique(s)"`
+}
+
+var (
+	orm *xorm.Engine
+)
+
+//
+// create a repository for a user or orgnaziation
+//
+func CreateUserRepository(root string, user *User, reposName string) 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 err
+	}
+
+	repo := Repo{OwnerId: user.Id, Type: UserRepo, Name: reposName}
+	_, err = orm.Insert(&repo)
+	if err != nil {
+		os.RemoveAll(f)
+	}
+	return err
+}
+
+//
+// delete a repository for a user or orgnaztion
+//
+func DeleteUserRepository(root string, user *User, reposName string) error {
+	err := os.RemoveAll(filepath.Join(root, user.Name, reposName))
+	if err != nil {
+		return err
+	}
+
+	_, err = orm.Delete(&Repo{OwnerId: user.Id, Type: UserRepo, Name: reposName})
+	return err
+}
diff --git a/models/models_test.go b/models/models_test.go
new file mode 100644
index 0000000000..80653d57fb
--- /dev/null
+++ b/models/models_test.go
@@ -0,0 +1,30 @@
+package models
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/lunny/xorm"
+	_ "github.com/mattn/go-sqlite3"
+)
+
+func init() {
+	var err error
+	orm, err = xorm.NewEngine("sqlite3", "./test.db")
+	if err != nil {
+		fmt.Println(err)
+	}
+
+	err = orm.Sync(&User{}, &Org{}, &Repo{})
+	if err != nil {
+		fmt.Println(err)
+	}
+}
+
+func TestCreateRepository(t *testing.T) {
+	user := User{Id: 1}
+	err := CreateUserRepository("test", &user, "test")
+	if err != nil {
+		t.Error(err)
+	}
+}