REST cleanup. Idiomatic naming of datastore funcs
This commit is contained in:
parent
1ad5e3f597
commit
0c3f9e5bde
28 changed files with 458 additions and 471 deletions
|
@ -2,7 +2,10 @@ package bolt
|
|||
|
||||
import (
|
||||
"errors"
|
||||
|
||||
"github.com/boltdb/bolt"
|
||||
|
||||
"github.com/drone/drone/datastore"
|
||||
)
|
||||
|
||||
var (
|
||||
|
@ -54,6 +57,13 @@ func New(path string) (*DB, error) {
|
|||
return nil
|
||||
})
|
||||
|
||||
// REMOVE BELOW
|
||||
var ds datastore.Datastore
|
||||
if ds == nil {
|
||||
ds = &DB{db}
|
||||
}
|
||||
// REMOVE ABOVE
|
||||
|
||||
return &DB{db}, nil
|
||||
}
|
||||
|
||||
|
|
|
@ -10,9 +10,9 @@ import (
|
|||
"github.com/drone/drone/common"
|
||||
)
|
||||
|
||||
// GetBuild gets the specified build number for the
|
||||
// named repository and build number
|
||||
func (db *DB) GetBuild(repo string, build int) (*common.Build, error) {
|
||||
// Build gets the specified build number for the
|
||||
// named repository and build number.
|
||||
func (db *DB) Build(repo string, build int) (*common.Build, error) {
|
||||
build_ := &common.Build{}
|
||||
key := []byte(repo + "/" + strconv.Itoa(build))
|
||||
|
||||
|
@ -23,12 +23,12 @@ func (db *DB) GetBuild(repo string, build int) (*common.Build, error) {
|
|||
return build_, err
|
||||
}
|
||||
|
||||
// GetBuildList gets a list of recent builds for the
|
||||
// BuildList gets a list of recent builds for the
|
||||
// named repository.
|
||||
func (db *DB) GetBuildList(repo string) ([]*common.Build, error) {
|
||||
func (db *DB) BuildList(repo string) ([]*common.Build, error) {
|
||||
// TODO (bradrydzewski) we can do this more efficiently
|
||||
var builds []*common.Build
|
||||
build, err := db.GetBuildLast(repo)
|
||||
build, err := db.BuildLast(repo)
|
||||
if err == ErrKeyNotFound {
|
||||
return builds, nil
|
||||
} else if err != nil {
|
||||
|
@ -54,9 +54,9 @@ func (db *DB) GetBuildList(repo string) ([]*common.Build, error) {
|
|||
return builds, err
|
||||
}
|
||||
|
||||
// GetBuildLast gets the last executed build for the
|
||||
// BuildLast gets the last executed build for the
|
||||
// named repository.
|
||||
func (db *DB) GetBuildLast(repo string) (*common.Build, error) {
|
||||
func (db *DB) BuildLast(repo string) (*common.Build, error) {
|
||||
key := []byte(repo)
|
||||
build := &common.Build{}
|
||||
err := db.View(func(t *bolt.Tx) error {
|
||||
|
@ -71,9 +71,49 @@ func (db *DB) GetBuildLast(repo string) (*common.Build, error) {
|
|||
return build, err
|
||||
}
|
||||
|
||||
// GetBuildStatus gets the named build status for the
|
||||
// named repository and build number.
|
||||
func (db *DB) GetBuildStatus(repo string, build int, status string) (*common.Status, error) {
|
||||
// SetBuild inserts or updates a build for the named
|
||||
// repository. The build number is incremented and
|
||||
// assigned to the provided build.
|
||||
func (db *DB) SetBuild(repo string, build *common.Build) error {
|
||||
repokey := []byte(repo)
|
||||
build.Updated = time.Now().UTC().Unix()
|
||||
|
||||
return db.Update(func(t *bolt.Tx) error {
|
||||
|
||||
if build.Number == 0 {
|
||||
raw, err := raw(t, bucketBuildSeq, repokey)
|
||||
|
||||
var next_seq uint32
|
||||
switch err {
|
||||
case ErrKeyNotFound:
|
||||
next_seq = 1
|
||||
case nil:
|
||||
next_seq = 1 + binary.LittleEndian.Uint32(raw)
|
||||
default:
|
||||
return err
|
||||
}
|
||||
|
||||
// covert our seqno to raw value
|
||||
raw = make([]byte, 4) // TODO(benschumacher) replace magic number 4 (uint32)
|
||||
binary.LittleEndian.PutUint32(raw, next_seq)
|
||||
err = t.Bucket(bucketBuildSeq).Put(repokey, raw)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// fill out the build structure
|
||||
build.Number = int(next_seq)
|
||||
build.Created = time.Now().UTC().Unix()
|
||||
}
|
||||
|
||||
key := []byte(repo + "/" + strconv.Itoa(build.Number))
|
||||
return update(t, bucketBuild, key, build)
|
||||
})
|
||||
}
|
||||
|
||||
// Status returns the status for the given repository
|
||||
// and build number.
|
||||
func (db *DB) Status(repo string, build int, status string) (*common.Status, error) {
|
||||
status_ := &common.Status{}
|
||||
key := []byte(repo + "/" + strconv.Itoa(build) + "/" + status)
|
||||
|
||||
|
@ -84,9 +124,9 @@ func (db *DB) GetBuildStatus(repo string, build int, status string) (*common.Sta
|
|||
return status_, err
|
||||
}
|
||||
|
||||
// GetBuildStatusList gets a list of all build statues for
|
||||
// the named repository and build number.
|
||||
func (db *DB) GetBuildStatusList(repo string, build int) ([]*common.Status, error) {
|
||||
// StatusList returned a list of all build statues for
|
||||
// the given repository and build number.
|
||||
func (db *DB) StatusList(repo string, build int) ([]*common.Status, error) {
|
||||
// TODO (bradrydzewski) explore efficiency of cursor vs index
|
||||
|
||||
statuses := []*common.Status{}
|
||||
|
@ -105,59 +145,13 @@ func (db *DB) GetBuildStatusList(repo string, build int) ([]*common.Status, erro
|
|||
return statuses, err
|
||||
}
|
||||
|
||||
// InsertBuild inserts a new build for the named repository
|
||||
func (db *DB) InsertBuild(repo string, build *common.Build) error {
|
||||
key := []byte(repo)
|
||||
|
||||
return db.Update(func(t *bolt.Tx) error {
|
||||
raw, err := raw(t, bucketBuildSeq, key)
|
||||
|
||||
var next_seq uint32
|
||||
switch err {
|
||||
case ErrKeyNotFound:
|
||||
next_seq = 1
|
||||
case nil:
|
||||
next_seq = 1 + binary.LittleEndian.Uint32(raw)
|
||||
default:
|
||||
return err
|
||||
}
|
||||
|
||||
// covert our seqno to raw value
|
||||
raw = make([]byte, 4) // TODO(benschumacher) replace magic number 4 (uint32)
|
||||
binary.LittleEndian.PutUint32(raw, next_seq)
|
||||
err = t.Bucket(bucketBuildSeq).Put(key, raw)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// fill out the build structure
|
||||
build.Number = int(next_seq)
|
||||
build.Created = time.Now().UTC().Unix()
|
||||
|
||||
key = []byte(repo + "/" + strconv.Itoa(build.Number))
|
||||
return insert(t, bucketBuild, key, build)
|
||||
})
|
||||
}
|
||||
|
||||
// InsertBuildStatus inserts a new build status for the
|
||||
// SetStatus inserts a new build status for the
|
||||
// named repository and build number. If the status already
|
||||
// exists an error is returned.
|
||||
func (db *DB) InsertBuildStatus(repo string, build int, status *common.Status) error {
|
||||
func (db *DB) SetStatus(repo string, build int, status *common.Status) error {
|
||||
key := []byte(repo + "/" + strconv.Itoa(build) + "/" + status.Context)
|
||||
|
||||
return db.Update(func(t *bolt.Tx) error {
|
||||
return update(t, bucketBuildStatus, key, status)
|
||||
})
|
||||
}
|
||||
|
||||
// UpdateBuild updates an existing build for the named
|
||||
// repository. If the build already exists and error is
|
||||
// returned.
|
||||
func (db *DB) UpdateBuild(repo string, build *common.Build) error {
|
||||
key := []byte(repo + "/" + strconv.Itoa(build.Number))
|
||||
build.Updated = time.Now().UTC().Unix()
|
||||
|
||||
return db.Update(func(t *bolt.Tx) error {
|
||||
return update(t, bucketBuild, key, build)
|
||||
})
|
||||
}
|
||||
|
|
|
@ -24,41 +24,41 @@ func TestBuild(t *testing.T) {
|
|||
})
|
||||
|
||||
g.It("Should sequence builds", func() {
|
||||
err := db.InsertBuild(repo, &common.Build{State: "pending"})
|
||||
err := db.SetBuild(repo, &common.Build{State: "pending"})
|
||||
g.Assert(err).Equal(nil)
|
||||
|
||||
// the first build should always be numero 1
|
||||
build, err := db.GetBuild(repo, 1)
|
||||
build, err := db.Build(repo, 1)
|
||||
g.Assert(err).Equal(nil)
|
||||
g.Assert(build.State).Equal("pending")
|
||||
|
||||
// add another build, just for fun
|
||||
err = db.InsertBuild(repo, &common.Build{State: "success"})
|
||||
err = db.SetBuild(repo, &common.Build{State: "success"})
|
||||
g.Assert(err).Equal(nil)
|
||||
|
||||
// get the next build
|
||||
build, err = db.GetBuild(repo, 2)
|
||||
build, err = db.Build(repo, 2)
|
||||
g.Assert(err).Equal(nil)
|
||||
g.Assert(build.State).Equal("success")
|
||||
})
|
||||
|
||||
g.It("Should get the latest builds", func() {
|
||||
db.InsertBuild(repo, &common.Build{State: "success"})
|
||||
db.InsertBuild(repo, &common.Build{State: "success"})
|
||||
db.InsertBuild(repo, &common.Build{State: "pending"})
|
||||
db.SetBuild(repo, &common.Build{State: "success"})
|
||||
db.SetBuild(repo, &common.Build{State: "success"})
|
||||
db.SetBuild(repo, &common.Build{State: "pending"})
|
||||
|
||||
build, err := db.GetBuildLast(repo)
|
||||
build, err := db.BuildLast(repo)
|
||||
g.Assert(err).Equal(nil)
|
||||
g.Assert(build.State).Equal("pending")
|
||||
g.Assert(build.Number).Equal(3)
|
||||
})
|
||||
|
||||
g.It("Should get the recent list of builds", func() {
|
||||
db.InsertBuild(repo, &common.Build{State: "success"})
|
||||
db.InsertBuild(repo, &common.Build{State: "success"})
|
||||
db.InsertBuild(repo, &common.Build{State: "pending"})
|
||||
db.SetBuild(repo, &common.Build{State: "success"})
|
||||
db.SetBuild(repo, &common.Build{State: "success"})
|
||||
db.SetBuild(repo, &common.Build{State: "pending"})
|
||||
|
||||
builds, err := db.GetBuildList(repo)
|
||||
builds, err := db.BuildList(repo)
|
||||
g.Assert(err).Equal(nil)
|
||||
g.Assert(len(builds)).Equal(3)
|
||||
})
|
||||
|
|
|
@ -8,8 +8,8 @@ import (
|
|||
"github.com/drone/drone/common"
|
||||
)
|
||||
|
||||
// GetRepo gets the repository by name.
|
||||
func (db *DB) GetRepo(repo string) (*common.Repo, error) {
|
||||
// Repo returns the repository with the given name.
|
||||
func (db *DB) Repo(repo string) (*common.Repo, error) {
|
||||
repo_ := &common.Repo{}
|
||||
key := []byte(repo)
|
||||
|
||||
|
@ -20,9 +20,42 @@ func (db *DB) GetRepo(repo string) (*common.Repo, error) {
|
|||
return repo_, err
|
||||
}
|
||||
|
||||
// GetRepoParams gets the private environment parameters
|
||||
// RepoList returns a list of repositories for the
|
||||
// given user account.
|
||||
func (db *DB) RepoList(login string) ([]*common.Repo, error) {
|
||||
repos := []*common.Repo{}
|
||||
err := db.View(func(t *bolt.Tx) error {
|
||||
// get the index of user tokens and unmarshal
|
||||
// to a string array.
|
||||
var keys [][]byte
|
||||
err := get(t, bucketUserRepos, []byte(login), &keys)
|
||||
if err != nil && err != ErrKeyNotFound {
|
||||
return err
|
||||
}
|
||||
|
||||
// for each item in the index, get the repository
|
||||
// and append to the array
|
||||
for _, key := range keys {
|
||||
repo := &common.Repo{}
|
||||
err := get(t, bucketRepo, key, repo)
|
||||
if err == ErrKeyNotFound {
|
||||
// TODO if we come across ErrKeyNotFound it means
|
||||
// we need to re-build the index
|
||||
continue
|
||||
} else if err != nil {
|
||||
return err
|
||||
}
|
||||
repos = append(repos, repo)
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
return repos, err
|
||||
}
|
||||
|
||||
// RepoParams returns the private environment parameters
|
||||
// for the given repository.
|
||||
func (db *DB) GetRepoParams(repo string) (map[string]string, error) {
|
||||
func (db *DB) RepoParams(repo string) (map[string]string, error) {
|
||||
params := map[string]string{}
|
||||
key := []byte(repo)
|
||||
|
||||
|
@ -33,9 +66,9 @@ func (db *DB) GetRepoParams(repo string) (map[string]string, error) {
|
|||
return params, err
|
||||
}
|
||||
|
||||
// GetRepoParams gets the private and public rsa keys
|
||||
// RepoKeypair returns the private and public rsa keys
|
||||
// for the given repository.
|
||||
func (db *DB) GetRepoKeys(repo string) (*common.Keypair, error) {
|
||||
func (db *DB) RepoKeypair(repo string) (*common.Keypair, error) {
|
||||
keypair := &common.Keypair{}
|
||||
key := []byte(repo)
|
||||
|
||||
|
@ -46,9 +79,8 @@ func (db *DB) GetRepoKeys(repo string) (*common.Keypair, error) {
|
|||
return keypair, err
|
||||
}
|
||||
|
||||
// UpdateRepos updates a repository. If the repository
|
||||
// does not exist an error is returned.
|
||||
func (db *DB) UpdateRepo(repo *common.Repo) error {
|
||||
// SetRepo inserts or updates a repository.
|
||||
func (db *DB) SetRepo(repo *common.Repo) error {
|
||||
key := []byte(repo.FullName)
|
||||
repo.Updated = time.Now().UTC().Unix()
|
||||
|
||||
|
@ -57,9 +89,9 @@ func (db *DB) UpdateRepo(repo *common.Repo) error {
|
|||
})
|
||||
}
|
||||
|
||||
// InsertRepo inserts a repository in the datastore and
|
||||
// subscribes the user to that repository.
|
||||
func (db *DB) InsertRepo(user *common.User, repo *common.Repo) error {
|
||||
// SetRepoNotExists updates a repository. If the repository
|
||||
// already exists ErrConflict is returned.
|
||||
func (db *DB) SetRepoNotExists(user *common.User, repo *common.Repo) error {
|
||||
repokey := []byte(repo.FullName)
|
||||
repo.Created = time.Now().UTC().Unix()
|
||||
repo.Updated = time.Now().UTC().Unix()
|
||||
|
@ -78,9 +110,9 @@ func (db *DB) InsertRepo(user *common.User, repo *common.Repo) error {
|
|||
})
|
||||
}
|
||||
|
||||
// UpsertRepoParams inserts or updates the private
|
||||
// SetRepoParams inserts or updates the private
|
||||
// environment parameters for the named repository.
|
||||
func (db *DB) UpsertRepoParams(repo string, params map[string]string) error {
|
||||
func (db *DB) SetRepoParams(repo string, params map[string]string) error {
|
||||
key := []byte(repo)
|
||||
|
||||
return db.Update(func(t *bolt.Tx) error {
|
||||
|
@ -88,9 +120,9 @@ func (db *DB) UpsertRepoParams(repo string, params map[string]string) error {
|
|||
})
|
||||
}
|
||||
|
||||
// UpsertRepoKeys inserts or updates the private and
|
||||
// SetRepoKeypair inserts or updates the private and
|
||||
// public keypair for the named repository.
|
||||
func (db *DB) UpsertRepoKeys(repo string, keypair *common.Keypair) error {
|
||||
func (db *DB) SetRepoKeypair(repo string, keypair *common.Keypair) error {
|
||||
key := []byte(repo)
|
||||
|
||||
return db.Update(func(t *bolt.Tx) error {
|
||||
|
@ -98,8 +130,8 @@ func (db *DB) UpsertRepoKeys(repo string, keypair *common.Keypair) error {
|
|||
})
|
||||
}
|
||||
|
||||
// DeleteRepo deletes the repository.
|
||||
func (db *DB) DeleteRepo(repo *common.Repo) error {
|
||||
// DelRepo deletes the repository.
|
||||
func (db *DB) DelRepo(repo *common.Repo) error {
|
||||
//TODO(benschumacher) rework this to use BoltDB's txn wrapper
|
||||
|
||||
t, err := db.Begin(true)
|
||||
|
@ -119,9 +151,13 @@ func (db *DB) DeleteRepo(repo *common.Repo) error {
|
|||
return t.Commit()
|
||||
}
|
||||
|
||||
// GetSubscriber gets the subscriber by login for the
|
||||
// named repository.
|
||||
func (db *DB) GetSubscriber(login, repo string) (*common.Subscriber, error) {
|
||||
// Subscribed returns true if the user is subscribed
|
||||
// to the named repository.
|
||||
//
|
||||
// TODO (bradrydzewski) we are currently storing the subscription
|
||||
// data in a wrapper element called common.Subscriber. This is
|
||||
// no longer necessary.
|
||||
func (db *DB) Subscribed(login, repo string) (bool, error) {
|
||||
sub := &common.Subscriber{}
|
||||
err := db.View(func(t *bolt.Tx) error {
|
||||
repokey := []byte(repo)
|
||||
|
@ -142,12 +178,12 @@ func (db *DB) GetSubscriber(login, repo string) (*common.Subscriber, error) {
|
|||
}
|
||||
return nil
|
||||
})
|
||||
return sub, err
|
||||
return sub.Subscribed, err
|
||||
}
|
||||
|
||||
// InsertSubscriber inserts a subscriber for the named
|
||||
// SetSubscriber inserts a subscriber for the named
|
||||
// repository.
|
||||
func (db *DB) InsertSubscriber(login, repo string) error {
|
||||
func (db *DB) SetSubscriber(login, repo string) error {
|
||||
return db.Update(func(t *bolt.Tx) error {
|
||||
userkey := []byte(login)
|
||||
repokey := []byte(repo)
|
||||
|
@ -155,9 +191,9 @@ func (db *DB) InsertSubscriber(login, repo string) error {
|
|||
})
|
||||
}
|
||||
|
||||
// DeleteSubscriber removes the subscriber by login for the
|
||||
// DelSubscriber removes the subscriber by login for the
|
||||
// named repository.
|
||||
func (db *DB) DeleteSubscriber(login, repo string) error {
|
||||
func (db *DB) DelSubscriber(login, repo string) error {
|
||||
return db.Update(func(t *bolt.Tx) error {
|
||||
userkey := []byte(login)
|
||||
repokey := []byte(repo)
|
||||
|
|
|
@ -11,7 +11,7 @@ import (
|
|||
|
||||
// GetTask gets the task at index N for the named
|
||||
// repository and build number.
|
||||
func (db *DB) GetTask(repo string, build int, task int) (*common.Task, error) {
|
||||
func (db *DB) Task(repo string, build int, task int) (*common.Task, error) {
|
||||
key := []byte(repo + "/" + strconv.Itoa(build) + "/" + strconv.Itoa(task))
|
||||
task_ := &common.Task{}
|
||||
err := db.View(func(t *bolt.Tx) error {
|
||||
|
@ -20,27 +20,12 @@ func (db *DB) GetTask(repo string, build int, task int) (*common.Task, error) {
|
|||
return task_, err
|
||||
}
|
||||
|
||||
// GetTaskLogs gets the task logs at index N for
|
||||
// the named repository and build number.
|
||||
func (db *DB) GetTaskLogs(repo string, build int, task int) (io.Reader, error) {
|
||||
key := []byte(repo + "/" + strconv.Itoa(build) + "/" + strconv.Itoa(task))
|
||||
|
||||
var log []byte
|
||||
err := db.View(func(t *bolt.Tx) error {
|
||||
var err error
|
||||
log, err = raw(t, bucketBuildLogs, key)
|
||||
return err
|
||||
})
|
||||
buf := bytes.NewBuffer(log)
|
||||
return buf, err
|
||||
}
|
||||
|
||||
// GetTaskList gets all tasks for the named repository
|
||||
// TaskList gets all tasks for the named repository
|
||||
// and build number.
|
||||
func (db *DB) GetTaskList(repo string, build int) ([]*common.Task, error) {
|
||||
func (db *DB) TaskList(repo string, build int) ([]*common.Task, error) {
|
||||
// fetch the build so that we can get the
|
||||
// number of child tasks.
|
||||
build_, err := db.GetBuild(repo, build)
|
||||
build_, err := db.Build(repo, build)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
@ -70,18 +55,18 @@ func (db *DB) GetTaskList(repo string, build int) ([]*common.Task, error) {
|
|||
return tasks, nil
|
||||
}
|
||||
|
||||
// UpsertTask inserts or updates a task for the named
|
||||
// SetTask inserts or updates a task for the named
|
||||
// repository and build number.
|
||||
func (db *DB) UpsertTask(repo string, build int, task *common.Task) error {
|
||||
func (db *DB) SetTask(repo string, build int, task *common.Task) error {
|
||||
key := []byte(repo + "/" + strconv.Itoa(build) + "/" + strconv.Itoa(task.Number))
|
||||
return db.Update(func(t *bolt.Tx) error {
|
||||
return update(t, bucketBuildTasks, key, task)
|
||||
})
|
||||
}
|
||||
|
||||
// UpsertTaskLogs inserts or updates a task logs for the
|
||||
// SetLogs inserts or updates a task logs for the
|
||||
// named repository and build number.
|
||||
func (db *DB) UpsertTaskLogs(repo string, build int, task int, log []byte) error {
|
||||
func (db *DB) SetLogs(repo string, build int, task int, log []byte) error {
|
||||
key := []byte(repo + "/" + strconv.Itoa(build) + "/" + strconv.Itoa(task))
|
||||
t, err := db.Begin(true)
|
||||
if err != nil {
|
||||
|
@ -94,3 +79,18 @@ func (db *DB) UpsertTaskLogs(repo string, build int, task int, log []byte) error
|
|||
}
|
||||
return t.Commit()
|
||||
}
|
||||
|
||||
// LogReader gets the task logs at index N for
|
||||
// the named repository and build number.
|
||||
func (db *DB) LogReader(repo string, build int, task int) (io.Reader, error) {
|
||||
key := []byte(repo + "/" + strconv.Itoa(build) + "/" + strconv.Itoa(task))
|
||||
|
||||
var log []byte
|
||||
err := db.View(func(t *bolt.Tx) error {
|
||||
var err error
|
||||
log, err = raw(t, bucketBuildLogs, key)
|
||||
return err
|
||||
})
|
||||
buf := bytes.NewBuffer(log)
|
||||
return buf, err
|
||||
}
|
||||
|
|
|
@ -5,19 +5,48 @@ import (
|
|||
"github.com/drone/drone/common"
|
||||
)
|
||||
|
||||
// GetToken gets a token by sha value.
|
||||
func (db *DB) GetToken(user, label string) (*common.Token, error) {
|
||||
// Token returns the token for the given user and label.
|
||||
func (db *DB) Token(user, label string) (*common.Token, error) {
|
||||
token := &common.Token{}
|
||||
key := []byte(user + "/" + label)
|
||||
|
||||
err := db.View(func(t *bolt.Tx) error {
|
||||
return get(t, bucketTokens, key, token)
|
||||
})
|
||||
return token, err
|
||||
}
|
||||
|
||||
// InsertToken inserts a new user token in the datastore.
|
||||
// If the token already exists and error is returned.
|
||||
func (db *DB) InsertToken(token *common.Token) error {
|
||||
// TokenList returns a list of all tokens for the given
|
||||
// user login.
|
||||
func (db *DB) TokenList(login string) ([]*common.Token, error) {
|
||||
tokens := []*common.Token{}
|
||||
userkey := []byte(login)
|
||||
err := db.Update(func(t *bolt.Tx) error {
|
||||
// get the index of user tokens and unmarshal
|
||||
// to a string array.
|
||||
var keys [][]byte
|
||||
err := get(t, bucketUserTokens, userkey, &keys)
|
||||
if err != nil && err != ErrKeyNotFound {
|
||||
return err
|
||||
}
|
||||
// for each item in the index, get the repository
|
||||
// and append to the array
|
||||
for _, key := range keys {
|
||||
token := &common.Token{}
|
||||
raw := t.Bucket(bucketTokens).Get(key)
|
||||
err = decode(raw, token)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
tokens = append(tokens, token)
|
||||
}
|
||||
return nil
|
||||
})
|
||||
return tokens, err
|
||||
}
|
||||
|
||||
// SetToken inserts a new user token in the datastore.
|
||||
func (db *DB) SetToken(token *common.Token) error {
|
||||
key := []byte(token.Login + "/" + token.Label)
|
||||
return db.Update(func(t *bolt.Tx) error {
|
||||
err := push(t, bucketUserTokens, []byte(token.Login), key)
|
||||
|
@ -28,14 +57,14 @@ func (db *DB) InsertToken(token *common.Token) error {
|
|||
})
|
||||
}
|
||||
|
||||
// DeleteUser deletes the token.
|
||||
func (db *DB) DeleteToken(token *common.Token) error {
|
||||
// DelToken deletes the token.
|
||||
func (db *DB) DelToken(token *common.Token) error {
|
||||
key := []byte(token.Login + "/" + token.Label)
|
||||
return db.Update(func(t *bolt.Tx) error {
|
||||
err := splice(t, bucketUserTokens, []byte(token.Login), key)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return delete(t, bucketUser, key)
|
||||
return delete(t, bucketTokens, key)
|
||||
})
|
||||
}
|
||||
|
|
|
@ -22,20 +22,44 @@ func TestToken(t *testing.T) {
|
|||
os.Remove(db.Path())
|
||||
})
|
||||
|
||||
g.It("Should find by label")
|
||||
g.It("Should list for user", func() {
|
||||
db.InsertUser(&common.User{Login: "octocat"})
|
||||
err1 := db.InsertToken(&common.Token{Login: "octocat", Label: "gist"})
|
||||
err2 := db.InsertToken(&common.Token{Login: "octocat", Label: "github"})
|
||||
db.SetUserNotExists(&common.User{Login: "octocat"})
|
||||
err1 := db.SetToken(&common.Token{Login: "octocat", Label: "gist"})
|
||||
err2 := db.SetToken(&common.Token{Login: "octocat", Label: "github"})
|
||||
g.Assert(err1).Equal(nil)
|
||||
g.Assert(err2).Equal(nil)
|
||||
|
||||
list, err := db.GetUserTokens("octocat")
|
||||
list, err := db.TokenList("octocat")
|
||||
g.Assert(err).Equal(nil)
|
||||
g.Assert(len(list)).Equal(2)
|
||||
})
|
||||
g.It("Should delete")
|
||||
g.It("Should insert")
|
||||
g.It("Should not insert if exists")
|
||||
|
||||
g.It("Should insert", func() {
|
||||
db.SetUserNotExists(&common.User{Login: "octocat"})
|
||||
err := db.SetToken(&common.Token{Login: "octocat", Label: "gist"})
|
||||
g.Assert(err).Equal(nil)
|
||||
|
||||
token, err := db.Token("octocat", "gist")
|
||||
g.Assert(err).Equal(nil)
|
||||
g.Assert(token.Label).Equal("gist")
|
||||
g.Assert(token.Login).Equal("octocat")
|
||||
})
|
||||
|
||||
g.It("Should delete", func() {
|
||||
db.SetUserNotExists(&common.User{Login: "octocat"})
|
||||
err := db.SetToken(&common.Token{Login: "octocat", Label: "gist"})
|
||||
g.Assert(err).Equal(nil)
|
||||
|
||||
token, err := db.Token("octocat", "gist")
|
||||
g.Assert(err).Equal(nil)
|
||||
g.Assert(token.Label).Equal("gist")
|
||||
g.Assert(token.Login).Equal("octocat")
|
||||
|
||||
err = db.DelToken(token)
|
||||
g.Assert(err).Equal(nil)
|
||||
|
||||
token, err = db.Token("octocat", "gist")
|
||||
g.Assert(err != nil).IsTrue()
|
||||
})
|
||||
})
|
||||
}
|
||||
|
|
|
@ -7,8 +7,8 @@ import (
|
|||
"github.com/drone/drone/common"
|
||||
)
|
||||
|
||||
// GetUser gets a user by user login.
|
||||
func (db *DB) GetUser(login string) (*common.User, error) {
|
||||
// User returns a user by user login.
|
||||
func (db *DB) User(login string) (*common.User, error) {
|
||||
user := &common.User{}
|
||||
key := []byte(login)
|
||||
|
||||
|
@ -19,74 +19,8 @@ func (db *DB) GetUser(login string) (*common.User, error) {
|
|||
return user, err
|
||||
}
|
||||
|
||||
// GetUserTokens gets a list of all tokens for
|
||||
// the given user login.
|
||||
func (db *DB) GetUserTokens(login string) ([]*common.Token, error) {
|
||||
t, err := db.Begin(false)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer t.Rollback()
|
||||
tokens := []*common.Token{}
|
||||
|
||||
// get the index of user tokens and unmarshal
|
||||
// to a string array.
|
||||
var keys [][]byte
|
||||
err = get(t, bucketUserTokens, []byte(login), &keys)
|
||||
if err != nil && err != ErrKeyNotFound {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// for each item in the index, get the repository
|
||||
// and append to the array
|
||||
for _, key := range keys {
|
||||
token := &common.Token{}
|
||||
raw := t.Bucket(bucketTokens).Get(key)
|
||||
err = decode(raw, token)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
tokens = append(tokens, token)
|
||||
}
|
||||
return tokens, nil
|
||||
}
|
||||
|
||||
// GetUserRepos gets a list of repositories for the
|
||||
// given user account.
|
||||
func (db *DB) GetUserRepos(login string) ([]*common.Repo, error) {
|
||||
repos := []*common.Repo{}
|
||||
err := db.View(func(t *bolt.Tx) error {
|
||||
// get the index of user tokens and unmarshal
|
||||
// to a string array.
|
||||
var keys [][]byte
|
||||
err := get(t, bucketUserRepos, []byte(login), &keys)
|
||||
if err != nil && err != ErrKeyNotFound {
|
||||
return err
|
||||
}
|
||||
|
||||
// for each item in the index, get the repository
|
||||
// and append to the array
|
||||
for _, key := range keys {
|
||||
repo := &common.Repo{}
|
||||
err := get(t, bucketRepo, key, repo)
|
||||
if err == ErrKeyNotFound {
|
||||
// TODO if we come across ErrKeyNotFound it means
|
||||
// we need to re-build the index
|
||||
continue
|
||||
} else if err != nil {
|
||||
return err
|
||||
}
|
||||
repos = append(repos, repo)
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
return repos, err
|
||||
}
|
||||
|
||||
// GetUserCount gets a count of all registered users
|
||||
// in the system.
|
||||
func (db *DB) GetUserCount() (int, error) {
|
||||
// UserCount returns a count of all registered users.
|
||||
func (db *DB) UserCount() (int, error) {
|
||||
var out int
|
||||
var err = db.View(func(t *bolt.Tx) error {
|
||||
out = t.Bucket(bucketUser).Stats().KeyN
|
||||
|
@ -95,8 +29,8 @@ func (db *DB) GetUserCount() (int, error) {
|
|||
return out, err
|
||||
}
|
||||
|
||||
// GetUserList gets a list of all registered users.
|
||||
func (db *DB) GetUserList() ([]*common.User, error) {
|
||||
// UserList returns a list of all registered users.
|
||||
func (db *DB) UserList() ([]*common.User, error) {
|
||||
users := []*common.User{}
|
||||
err := db.View(func(t *bolt.Tx) error {
|
||||
return t.Bucket(bucketUser).ForEach(func(key, raw []byte) error {
|
||||
|
@ -112,9 +46,8 @@ func (db *DB) GetUserList() ([]*common.User, error) {
|
|||
return users, err
|
||||
}
|
||||
|
||||
// UpdateUser updates an existing user. If the user
|
||||
// does not exists an error is returned.
|
||||
func (db *DB) UpdateUser(user *common.User) error {
|
||||
// SetUser inserts or updates a user.
|
||||
func (db *DB) SetUser(user *common.User) error {
|
||||
key := []byte(user.Login)
|
||||
user.Updated = time.Now().UTC().Unix()
|
||||
|
||||
|
@ -123,9 +56,9 @@ func (db *DB) UpdateUser(user *common.User) error {
|
|||
})
|
||||
}
|
||||
|
||||
// InsertUser inserts a new user into the datastore. If
|
||||
// the user login already exists an error is returned.
|
||||
func (db *DB) InsertUser(user *common.User) error {
|
||||
// SetUserNotExists inserts a new user into the datastore.
|
||||
// If the user login already exists ErrConflict is returned.
|
||||
func (db *DB) SetUserNotExists(user *common.User) error {
|
||||
key := []byte(user.Login)
|
||||
user.Created = time.Now().UTC().Unix()
|
||||
user.Updated = time.Now().UTC().Unix()
|
||||
|
@ -135,8 +68,8 @@ func (db *DB) InsertUser(user *common.User) error {
|
|||
})
|
||||
}
|
||||
|
||||
// DeleteUser deletes the user.
|
||||
func (db *DB) DeleteUser(user *common.User) error {
|
||||
// DelUser deletes the user.
|
||||
func (db *DB) DelUser(user *common.User) error {
|
||||
key := []byte(user.Login)
|
||||
// TODO(bradrydzewski) delete user subscriptions
|
||||
// TODO(bradrydzewski) delete user tokens
|
||||
|
|
|
@ -23,17 +23,17 @@ func TestUser(t *testing.T) {
|
|||
})
|
||||
|
||||
g.It("Should find", func() {
|
||||
db.InsertUser(&common.User{Login: "octocat"})
|
||||
user, err := db.GetUser("octocat")
|
||||
db.SetUserNotExists(&common.User{Login: "octocat"})
|
||||
user, err := db.User("octocat")
|
||||
g.Assert(err).Equal(nil)
|
||||
g.Assert(user.Login).Equal("octocat")
|
||||
})
|
||||
|
||||
g.It("Should insert", func() {
|
||||
err := db.InsertUser(&common.User{Login: "octocat"})
|
||||
err := db.SetUserNotExists(&common.User{Login: "octocat"})
|
||||
g.Assert(err).Equal(nil)
|
||||
|
||||
user, err := db.GetUser("octocat")
|
||||
user, err := db.User("octocat")
|
||||
g.Assert(err).Equal(nil)
|
||||
g.Assert(user.Login).Equal("octocat")
|
||||
g.Assert(user.Created != 0).IsTrue()
|
||||
|
@ -41,50 +41,50 @@ func TestUser(t *testing.T) {
|
|||
})
|
||||
|
||||
g.It("Should not insert if exists", func() {
|
||||
db.InsertUser(&common.User{Login: "octocat"})
|
||||
err := db.InsertUser(&common.User{Login: "octocat"})
|
||||
db.SetUser(&common.User{Login: "octocat"})
|
||||
err := db.SetUserNotExists(&common.User{Login: "octocat"})
|
||||
g.Assert(err).Equal(ErrKeyExists)
|
||||
})
|
||||
|
||||
g.It("Should update", func() {
|
||||
db.InsertUser(&common.User{Login: "octocat"})
|
||||
user, err := db.GetUser("octocat")
|
||||
db.SetUserNotExists(&common.User{Login: "octocat"})
|
||||
user, err := db.User("octocat")
|
||||
g.Assert(err).Equal(nil)
|
||||
|
||||
user.Email = "octocat@github.com"
|
||||
err = db.UpdateUser(user)
|
||||
err = db.SetUser(user)
|
||||
g.Assert(err).Equal(nil)
|
||||
|
||||
user_, err := db.GetUser("octocat")
|
||||
user_, err := db.User("octocat")
|
||||
g.Assert(err).Equal(nil)
|
||||
g.Assert(user_.Login).Equal(user.Login)
|
||||
g.Assert(user_.Email).Equal(user.Email)
|
||||
})
|
||||
|
||||
g.It("Should delete", func() {
|
||||
db.InsertUser(&common.User{Login: "octocat"})
|
||||
user, err := db.GetUser("octocat")
|
||||
db.SetUserNotExists(&common.User{Login: "octocat"})
|
||||
user, err := db.User("octocat")
|
||||
g.Assert(err).Equal(nil)
|
||||
|
||||
err = db.DeleteUser(user)
|
||||
err = db.DelUser(user)
|
||||
g.Assert(err).Equal(nil)
|
||||
|
||||
_, err = db.GetUser("octocat")
|
||||
_, err = db.User("octocat")
|
||||
g.Assert(err).Equal(ErrKeyNotFound)
|
||||
})
|
||||
|
||||
g.It("Should list", func() {
|
||||
db.InsertUser(&common.User{Login: "bert"})
|
||||
db.InsertUser(&common.User{Login: "ernie"})
|
||||
users, err := db.GetUserList()
|
||||
db.SetUserNotExists(&common.User{Login: "bert"})
|
||||
db.SetUserNotExists(&common.User{Login: "ernie"})
|
||||
users, err := db.UserList()
|
||||
g.Assert(err).Equal(nil)
|
||||
g.Assert(len(users)).Equal(2)
|
||||
})
|
||||
|
||||
g.It("Should count", func() {
|
||||
db.InsertUser(&common.User{Login: "bert"})
|
||||
db.InsertUser(&common.User{Login: "ernie"})
|
||||
count, err := db.GetUserCount()
|
||||
db.SetUserNotExists(&common.User{Login: "bert"})
|
||||
db.SetUserNotExists(&common.User{Login: "ernie"})
|
||||
count, err := db.UserCount()
|
||||
g.Assert(err).Equal(nil)
|
||||
g.Assert(count).Equal(2)
|
||||
})
|
||||
|
|
|
@ -6,138 +6,136 @@ import (
|
|||
"github.com/drone/drone/common"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrConflict = "Key not unique"
|
||||
ErrNotFound = "Key not found"
|
||||
)
|
||||
|
||||
type Datastore interface {
|
||||
// GetUser gets a user by user login.
|
||||
GetUser(string) (*common.User, error)
|
||||
// User returns a user by user login.
|
||||
User(string) (*common.User, error)
|
||||
|
||||
// GetUserTokens gets a list of all tokens for
|
||||
// the given user login.
|
||||
GetUserTokens(string) ([]*common.Token, error)
|
||||
// UserCount returns a count of all registered users.
|
||||
UserCount() (int, error)
|
||||
|
||||
// GetUserRepos gets a list of repositories for the
|
||||
// given user account.
|
||||
GetUserRepos(string) ([]*common.Repo, error)
|
||||
// UserList returns a list of all registered users.
|
||||
UserList() ([]*common.User, error)
|
||||
|
||||
// GetUserCount gets a count of all registered users
|
||||
// in the system.
|
||||
GetUserCount() (int, error)
|
||||
// SetUser inserts or updates a user.
|
||||
SetUser(*common.User) error
|
||||
|
||||
// GetUserList gets a list of all registered users.
|
||||
GetUserList() ([]*common.User, error)
|
||||
// SetUserNotExists inserts a new user into the datastore.
|
||||
// If the user login already exists ErrConflict is returned.
|
||||
SetUserNotExists(*common.User) error
|
||||
|
||||
// UpdateUser updates an existing user. If the user
|
||||
// does not exists an error is returned.
|
||||
UpdateUser(*common.User) error
|
||||
// Del deletes the user.
|
||||
DelUser(*common.User) error
|
||||
|
||||
// InsertUser inserts a new user into the datastore. If
|
||||
// the user login already exists an error is returned.
|
||||
InsertUser(*common.User) error
|
||||
// Token returns the token for the given user and label.
|
||||
Token(string, string) (*common.Token, error)
|
||||
|
||||
// DeleteUser deletes the user.
|
||||
DeleteUser(*common.User) error
|
||||
// TokenList returns a list of all tokens for the given
|
||||
// user login.
|
||||
TokenList(string) ([]*common.Token, error)
|
||||
|
||||
// GetToken gets a token by sha value.
|
||||
GetToken(string, string) (*common.Token, error)
|
||||
// SetToken inserts a new user token in the datastore.
|
||||
SetToken(*common.Token) error
|
||||
|
||||
// InsertToken inserts a new user token in the datastore.
|
||||
// If the token already exists and error is returned.
|
||||
InsertToken(*common.Token) error
|
||||
// DelToken deletes the token.
|
||||
DelToken(*common.Token) error
|
||||
|
||||
// DeleteUser deletes the token.
|
||||
DeleteToken(*common.Token) error
|
||||
// Subscribed returns true if the user is subscribed
|
||||
// to the named repository.
|
||||
Subscribed(string, string) (bool, error)
|
||||
|
||||
// GetSubscriber gets the subscriber by login for the
|
||||
// named repository.
|
||||
GetSubscriber(string, string) (*common.Subscriber, error)
|
||||
|
||||
// InsertSubscriber inserts a subscriber for the named
|
||||
// SetSubscriber inserts a subscriber for the named
|
||||
// repository.
|
||||
InsertSubscriber(string, string) error
|
||||
SetSubscriber(string, string) error
|
||||
|
||||
// DeleteSubscriber removes the subscriber by login for the
|
||||
// DelSubscriber removes the subscriber by login for the
|
||||
// named repository.
|
||||
DeleteSubscriber(string, string) error
|
||||
DelSubscriber(string, string) error
|
||||
|
||||
// GetRepo gets the repository by name.
|
||||
GetRepo(string) (*common.Repo, error)
|
||||
// Repo returns the repository with the given name.
|
||||
Repo(string) (*common.Repo, error)
|
||||
|
||||
// GetRepoParams gets the private environment parameters
|
||||
// RepoList returns a list of repositories for the
|
||||
// given user account.
|
||||
RepoList(string) ([]*common.Repo, error)
|
||||
|
||||
// RepoParams returns the private environment parameters
|
||||
// for the given repository.
|
||||
GetRepoParams(string) (map[string]string, error)
|
||||
RepoParams(string) (map[string]string, error)
|
||||
|
||||
// GetRepoParams gets the private and public rsa keys
|
||||
// RepoKeypair returns the private and public rsa keys
|
||||
// for the given repository.
|
||||
GetRepoKeys(string) (*common.Keypair, error)
|
||||
RepoKeypair(string) (*common.Keypair, error)
|
||||
|
||||
// UpdateRepos updates a repository. If the repository
|
||||
// does not exist an error is returned.
|
||||
UpdateRepo(*common.Repo) error
|
||||
// SetRepo inserts or updates a repository.
|
||||
SetRepo(*common.Repo) error
|
||||
|
||||
// InsertRepo inserts a repository in the datastore and
|
||||
// subscribes the user to that repository.
|
||||
InsertRepo(*common.User, *common.Repo) error
|
||||
// SetRepo updates a repository. If the repository
|
||||
// already exists ErrConflict is returned.
|
||||
SetRepoNotExists(*common.User, *common.Repo) error
|
||||
|
||||
// UpsertRepoParams inserts or updates the private
|
||||
// SetRepoParams inserts or updates the private
|
||||
// environment parameters for the named repository.
|
||||
UpsertRepoParams(string, map[string]string) error
|
||||
SetRepoParams(string, map[string]string) error
|
||||
|
||||
// UpsertRepoKeys inserts or updates the private and
|
||||
// SetRepoKeypair inserts or updates the private and
|
||||
// public keypair for the named repository.
|
||||
UpsertRepoKeys(string, *common.Keypair) error
|
||||
SetRepoKeypair(string, *common.Keypair) error
|
||||
|
||||
// DeleteRepo deletes the repository.
|
||||
DeleteRepo(*common.Repo) error
|
||||
// DelRepo deletes the repository.
|
||||
DelRepo(*common.Repo) error
|
||||
|
||||
// GetBuild gets the specified build number for the
|
||||
// Build gets the specified build number for the
|
||||
// named repository and build number
|
||||
GetBuild(string, int) (*common.Build, error)
|
||||
Build(string, int) (*common.Build, error)
|
||||
|
||||
// GetBuildList gets a list of recent builds for the
|
||||
// BuildList gets a list of recent builds for the
|
||||
// named repository.
|
||||
GetBuildList(string) ([]*common.Build, error)
|
||||
BuildList(string) ([]*common.Build, error)
|
||||
|
||||
// GetBuildLast gets the last executed build for the
|
||||
// BuildLast gets the last executed build for the
|
||||
// named repository.
|
||||
GetBuildLast(string) (*common.Build, error)
|
||||
BuildLast(string) (*common.Build, error)
|
||||
|
||||
// GetBuildStatus gets the named build status for the
|
||||
// named repository and build number.
|
||||
GetBuildStatus(string, int, string) (*common.Status, error)
|
||||
// SetBuild inserts or updates a build for the named
|
||||
// repository. The build number is incremented and
|
||||
// assigned to the provided build.
|
||||
SetBuild(string, *common.Build) error
|
||||
|
||||
// GetBuildStatusList gets a list of all build statues for
|
||||
// the named repository and build number.
|
||||
GetBuildStatusList(string, int) ([]*common.Status, error)
|
||||
// Status returns the status for the given repository
|
||||
// and build number.
|
||||
Status(string, int, string) (*common.Status, error)
|
||||
|
||||
// InsertBuild inserts a new build for the named repository
|
||||
InsertBuild(string, *common.Build) error
|
||||
// StatusList returned a list of all build statues for
|
||||
// the given repository and build number.
|
||||
StatusList(string, int) ([]*common.Status, error)
|
||||
|
||||
// InsertBuildStatus inserts a new build status for the
|
||||
// SetStatus inserts a new build status for the
|
||||
// named repository and build number. If the status already
|
||||
// exists an error is returned.
|
||||
InsertBuildStatus(string, int, *common.Status) error
|
||||
|
||||
// UpdateBuild updates an existing build for the named
|
||||
// repository. If the build already exists and error is
|
||||
// returned.
|
||||
UpdateBuild(string, *common.Build) error
|
||||
SetStatus(string, int, *common.Status) error
|
||||
|
||||
// GetTask gets the task at index N for the named
|
||||
// repository and build number.
|
||||
GetTask(string, int, int) (*common.Task, error)
|
||||
Task(string, int, int) (*common.Task, error)
|
||||
|
||||
// GetTaskLogs gets the task logs at index N for
|
||||
// the named repository and build number.
|
||||
GetTaskLogs(string, int, int) (io.Reader, error)
|
||||
|
||||
// GetTaskList gets all tasks for the named repository
|
||||
// TaskList gets all tasks for the named repository
|
||||
// and build number.
|
||||
GetTaskList(string, int) ([]*common.Task, error)
|
||||
TaskList(string, int) ([]*common.Task, error)
|
||||
|
||||
// UpsertTask inserts or updates a task for the named
|
||||
// SetTask inserts or updates a task for the named
|
||||
// repository and build number.
|
||||
UpsertTask(string, int, *common.Task) error
|
||||
SetTask(string, int, *common.Task) error
|
||||
|
||||
// UpsertTaskLogs inserts or updates a task logs for the
|
||||
// LogReader gets the task logs at index N for
|
||||
// the named repository and build number.
|
||||
LogReader(string, int, int) (io.Reader, error)
|
||||
|
||||
// SetLogs inserts or updates a task logs for the
|
||||
// named repository and build number.
|
||||
UpsertTaskLogs(string, int, int, []byte) error
|
||||
SetLogs(string, int, int, []byte) error
|
||||
}
|
||||
|
|
76
drone.go
76
drone.go
|
@ -42,7 +42,7 @@ func main() {
|
|||
user.Use(server.SetSession(session))
|
||||
|
||||
user.GET("", server.GetUserCurr)
|
||||
user.PUT("", server.PutUserCurr)
|
||||
user.PATCH("", server.PutUserCurr)
|
||||
user.GET("/repos", server.GetUserRepos)
|
||||
user.GET("/tokens", server.GetUserTokens)
|
||||
user.POST("/tokens", server.PostToken)
|
||||
|
@ -55,8 +55,8 @@ func main() {
|
|||
|
||||
users.GET("", server.GetUsers)
|
||||
users.GET("/:name", server.GetUser)
|
||||
users.PUT("/:name", server.PutUser)
|
||||
users.POST("/:name", server.PostUser)
|
||||
users.PATCH("/:name", server.PutUser)
|
||||
users.DELETE("/:name", server.DeleteUser)
|
||||
}
|
||||
|
||||
|
@ -72,67 +72,25 @@ func main() {
|
|||
repo.Use(server.CheckPush())
|
||||
|
||||
repo.GET("", server.GetRepo)
|
||||
repo.PUT("", server.PutRepo)
|
||||
repo.PATCH("", server.PutRepo)
|
||||
repo.DELETE("", server.DeleteRepo)
|
||||
repo.POST("/watch", server.Subscribe)
|
||||
repo.DELETE("/unwatch", server.Unsubscribe)
|
||||
|
||||
repo.GET("/builds", server.GetBuilds)
|
||||
repo.GET("/builds/:number", server.GetBuild)
|
||||
//TODO repo.POST("/builds/:number", server.RestartBuild)
|
||||
//TODO repo.DELETE("/builds/:number", server.CancelBuild)
|
||||
repo.GET("/builds/:number/tasks", server.GetTasks)
|
||||
repo.GET("/builds/:number/tasks/:task", server.GetTask)
|
||||
repo.GET("/builds/:number/tasks/:task/log", server.GetTaskLogs)
|
||||
|
||||
// repo.GET("/status/:number/status/:context", server.GetStatus)
|
||||
repo.GET("/status/:number", server.GetStatusList)
|
||||
repo.POST("/status/:number", server.PostStatus)
|
||||
}
|
||||
}
|
||||
|
||||
subscribers := api.Group("/subscribers/:owner/:name")
|
||||
{
|
||||
subscribers.Use(server.SetRepo())
|
||||
subscribers.Use(server.SetPerm())
|
||||
subscribers.Use(server.CheckPull())
|
||||
|
||||
subscribers.POST("", server.Subscribe)
|
||||
subscribers.DELETE("", server.Unsubscribe)
|
||||
}
|
||||
|
||||
builds := api.Group("/builds/:owner/:name")
|
||||
{
|
||||
builds.Use(server.SetRepo())
|
||||
builds.Use(server.SetPerm())
|
||||
builds.Use(server.CheckPull())
|
||||
builds.Use(server.CheckPush())
|
||||
|
||||
builds.GET("", server.GetBuilds)
|
||||
builds.GET("/:number", server.GetBuild)
|
||||
//TODO builds.POST("/:number", server.RestartBuild)
|
||||
//TODO builds.DELETE("/:number", server.CancelBuild)
|
||||
}
|
||||
|
||||
tasks := api.Group("/tasks/:owner/:name/:number")
|
||||
{
|
||||
tasks.Use(server.SetRepo())
|
||||
tasks.Use(server.SetPerm())
|
||||
tasks.Use(server.CheckPull())
|
||||
tasks.Use(server.CheckPush())
|
||||
|
||||
tasks.GET("", server.GetTasks)
|
||||
tasks.GET("/:task", server.GetTask)
|
||||
}
|
||||
|
||||
logs := api.Group("/logs/:owner/:name/:number/:task")
|
||||
{
|
||||
logs.Use(server.SetRepo())
|
||||
logs.Use(server.SetPerm())
|
||||
logs.Use(server.CheckPull())
|
||||
logs.Use(server.CheckPush())
|
||||
|
||||
logs.GET("", server.GetTaskLogs)
|
||||
}
|
||||
|
||||
status := api.Group("/status/:owner/:name/:number")
|
||||
{
|
||||
status.Use(server.SetRepo())
|
||||
status.Use(server.SetPerm())
|
||||
status.Use(server.CheckPull())
|
||||
status.Use(server.CheckPush())
|
||||
|
||||
status.GET("/:context", server.GetStatus)
|
||||
status.GET("", server.GetStatusList)
|
||||
status.POST("", server.PostStatus)
|
||||
}
|
||||
|
||||
badges := api.Group("/badges/:owner/:name")
|
||||
{
|
||||
badges.Use(server.SetRepo())
|
||||
|
|
|
@ -59,9 +59,9 @@ func GetBadge(c *gin.Context) {
|
|||
//
|
||||
// TODO(bradrydzewski) this will not return in-progress builds, which it should
|
||||
func GetCC(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
repo := ToRepo(c)
|
||||
last, err := ds.GetBuildLast(repo.FullName)
|
||||
last, err := store.BuildLast(repo.FullName)
|
||||
if err != nil {
|
||||
c.Fail(404, err)
|
||||
return
|
||||
|
|
|
@ -13,14 +13,14 @@ import (
|
|||
// GET /api/builds/:owner/:name/:number
|
||||
//
|
||||
func GetBuild(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
repo := ToRepo(c)
|
||||
num, err := strconv.Atoi(c.Params.ByName("number"))
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
return
|
||||
}
|
||||
build, err := ds.GetBuild(repo.FullName, num)
|
||||
build, err := store.Build(repo.FullName, num)
|
||||
if err != nil {
|
||||
c.Fail(404, err)
|
||||
} else {
|
||||
|
@ -34,9 +34,9 @@ func GetBuild(c *gin.Context) {
|
|||
// GET /api/builds/:owner/:name
|
||||
//
|
||||
func GetBuilds(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
repo := ToRepo(c)
|
||||
builds, err := ds.GetBuildList(repo.FullName)
|
||||
builds, err := store.BuildList(repo.FullName)
|
||||
if err != nil {
|
||||
c.Fail(404, err)
|
||||
} else {
|
||||
|
|
|
@ -42,7 +42,7 @@ func PostHook(c *gin.Context) {
|
|||
return
|
||||
}
|
||||
|
||||
repo, err := store.GetRepo(hook.Repo.FullName)
|
||||
repo, err := store.Repo(hook.Repo.FullName)
|
||||
if err != nil {
|
||||
log.Errorf("failure to find repo %s from hook. %s", hook.Repo.FullName, err)
|
||||
c.Fail(404, err)
|
||||
|
@ -64,7 +64,7 @@ func PostHook(c *gin.Context) {
|
|||
return
|
||||
}
|
||||
|
||||
user, err := store.GetUser(repo.User.Login)
|
||||
user, err := store.User(repo.User.Login)
|
||||
if err != nil {
|
||||
log.Errorf("failure to find repo owner %s. %s", repo.User.Login, err)
|
||||
c.Fail(500, err)
|
||||
|
@ -84,7 +84,7 @@ func PostHook(c *gin.Context) {
|
|||
return
|
||||
}
|
||||
|
||||
err = store.InsertBuild(repo.FullName, build)
|
||||
err = store.SetBuild(repo.FullName, build)
|
||||
if err != nil {
|
||||
c.Fail(500, err)
|
||||
return
|
||||
|
|
|
@ -51,9 +51,9 @@ func GetLogin(c *gin.Context) {
|
|||
|
||||
// get the user from the database
|
||||
login := ToUser(c)
|
||||
u, err := store.GetUser(login.Login)
|
||||
u, err := store.User(login.Login)
|
||||
if err != nil {
|
||||
count, err := store.GetUserCount()
|
||||
count, err := store.UserCount()
|
||||
if err != nil {
|
||||
log.Errorf("cannot register %s. %s", login.Login, err)
|
||||
c.Redirect(303, "/login#error=internal_error")
|
||||
|
@ -79,7 +79,7 @@ func GetLogin(c *gin.Context) {
|
|||
u.Gravatar = gravatar.Generate(u.Email)
|
||||
|
||||
// insert the user into the database
|
||||
if err := store.InsertUser(u); err != nil {
|
||||
if err := store.SetUserNotExists(u); err != nil {
|
||||
log.Errorf("cannot insert %s. %s", login.Login, err)
|
||||
c.Redirect(303, "/login#error=internal_error")
|
||||
return
|
||||
|
@ -100,7 +100,7 @@ func GetLogin(c *gin.Context) {
|
|||
u.Email = login.Email
|
||||
u.Gravatar = gravatar.Generate(u.Email)
|
||||
|
||||
if err := store.UpdateUser(u); err != nil {
|
||||
if err := store.SetUser(u); err != nil {
|
||||
log.Errorf("cannot update %s. %s", u.Login, err)
|
||||
c.Redirect(303, "/login#error=internal_error")
|
||||
return
|
||||
|
|
|
@ -17,9 +17,10 @@ import (
|
|||
// additional repository meta-data.
|
||||
type repoResp struct {
|
||||
*common.Repo
|
||||
Perms *common.Perm `json:"permissions,omitempty"`
|
||||
Watch *common.Subscriber `json:"subscription,omitempty"`
|
||||
Params map[string]string `json:"params,omitempty"`
|
||||
Perms *common.Perm `json:"permissions,omitempty"`
|
||||
Watch *common.Subscriber `json:"subscription,omitempty"`
|
||||
Keypair *common.Keypair `json:"keypair,omitempty"`
|
||||
Params map[string]string `json:"params,omitempty"`
|
||||
}
|
||||
|
||||
// repoReq is a data structure used for receiving
|
||||
|
@ -52,11 +53,11 @@ func GetRepo(c *gin.Context) {
|
|||
repo := ToRepo(c)
|
||||
user := ToUser(c)
|
||||
perm := ToPerm(c)
|
||||
data := repoResp{repo, perm, nil, nil}
|
||||
data := repoResp{repo, perm, nil, nil, nil}
|
||||
// if the user is an administrator of the project
|
||||
// we should display the private parameter data.
|
||||
if perm.Admin {
|
||||
data.Params, _ = store.GetRepoParams(repo.FullName)
|
||||
data.Params, _ = store.RepoParams(repo.FullName)
|
||||
}
|
||||
// if the user is authenticated, we should display
|
||||
// if she is watching the current repository.
|
||||
|
@ -66,7 +67,9 @@ func GetRepo(c *gin.Context) {
|
|||
}
|
||||
|
||||
// check to see if the user is subscribing to the repo
|
||||
data.Watch, _ = store.GetSubscriber(user.Login, repo.FullName)
|
||||
data.Watch = &common.Subscriber{}
|
||||
data.Watch.Subscribed, _ = store.Subscribed(user.Login, repo.FullName)
|
||||
data.Keypair, _ = store.RepoKeypair(repo.FullName)
|
||||
|
||||
c.JSON(200, data)
|
||||
}
|
||||
|
@ -89,7 +92,7 @@ func PutRepo(c *gin.Context) {
|
|||
}
|
||||
|
||||
if in.Params != nil {
|
||||
err := store.UpsertRepoParams(repo.FullName, *in.Params)
|
||||
err := store.SetRepoParams(repo.FullName, *in.Params)
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
return
|
||||
|
@ -111,17 +114,19 @@ func PutRepo(c *gin.Context) {
|
|||
repo.Timeout = *in.Timeout
|
||||
}
|
||||
|
||||
err := store.UpdateRepo(repo)
|
||||
err := store.SetRepo(repo)
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
return
|
||||
}
|
||||
|
||||
data := repoResp{repo, perm, nil, nil}
|
||||
data.Params, _ = store.GetRepoParams(repo.FullName)
|
||||
data := repoResp{repo, perm, nil, nil, nil}
|
||||
data.Params, _ = store.RepoParams(repo.FullName)
|
||||
|
||||
// check to see if the user is subscribing to the repo
|
||||
data.Watch, _ = store.GetSubscriber(user.Login, repo.FullName)
|
||||
data.Watch = &common.Subscriber{}
|
||||
data.Watch.Subscribed, _ = store.Subscribed(user.Login, repo.FullName)
|
||||
data.Keypair, _ = store.RepoKeypair(repo.FullName)
|
||||
|
||||
c.JSON(200, data)
|
||||
}
|
||||
|
@ -147,7 +152,7 @@ func DeleteRepo(c *gin.Context) {
|
|||
c.Fail(400, err)
|
||||
}
|
||||
|
||||
err = ds.DeleteRepo(r)
|
||||
err = ds.DelRepo(r)
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
}
|
||||
|
@ -213,14 +218,14 @@ func PostRepo(c *gin.Context) {
|
|||
println(link)
|
||||
|
||||
// persist the repository
|
||||
err = store.InsertRepo(user, r)
|
||||
err = store.SetRepoNotExists(user, r)
|
||||
if err != nil {
|
||||
c.Fail(500, err)
|
||||
return
|
||||
}
|
||||
|
||||
// persisty the repository key pair
|
||||
err = store.UpsertRepoKeys(r.FullName, keypair)
|
||||
err = store.SetRepoKeypair(r.FullName, keypair)
|
||||
if err != nil {
|
||||
c.Fail(500, err)
|
||||
return
|
||||
|
|
|
@ -114,7 +114,7 @@ func SetUser(s session.Session) gin.HandlerFunc {
|
|||
return
|
||||
}
|
||||
|
||||
u, err := ds.GetUser(token.Login)
|
||||
u, err := ds.User(token.Login)
|
||||
if err == nil {
|
||||
c.Set("user", u)
|
||||
}
|
||||
|
@ -129,7 +129,7 @@ func SetUser(s session.Session) gin.HandlerFunc {
|
|||
// to verify the token we fetch from the datastore
|
||||
// and check to see if the token issued date matches
|
||||
// what we found in the jwt (in case the label is re-used)
|
||||
t, err := ds.GetToken(token.Login, token.Label)
|
||||
t, err := ds.Token(token.Login, token.Label)
|
||||
if err != nil || t.Issued != token.Issued {
|
||||
c.AbortWithStatus(403)
|
||||
return
|
||||
|
@ -145,7 +145,7 @@ func SetRepo() gin.HandlerFunc {
|
|||
u := ToUser(c)
|
||||
owner := c.Params.ByName("owner")
|
||||
name := c.Params.ByName("name")
|
||||
r, err := ds.GetRepo(owner + "/" + name)
|
||||
r, err := ds.Repo(owner + "/" + name)
|
||||
switch {
|
||||
case err != nil && u != nil:
|
||||
c.Fail(401, err)
|
||||
|
|
|
@ -14,7 +14,7 @@
|
|||
* @param {string} Name of the repository.
|
||||
*/
|
||||
this.list = function(repoName) {
|
||||
return $http.get('/api/builds/'+repoName);
|
||||
return $http.get('/api/repos/'+repoName+'/builds');
|
||||
};
|
||||
|
||||
/**
|
||||
|
@ -24,11 +24,11 @@
|
|||
* @param {number} Number of the build.
|
||||
*/
|
||||
this.get = function(repoName, buildNumber) {
|
||||
return $http.get('/api/builds/'+repoName+'/'+buildNumber);
|
||||
return $http.get('/api/repos/'+repoName+'/builds/'+buildNumber);
|
||||
};
|
||||
}
|
||||
|
||||
angular
|
||||
.module('drone')
|
||||
.service('builds', BuildService);
|
||||
})();
|
||||
})();
|
||||
|
|
|
@ -16,11 +16,11 @@
|
|||
* @param {number} Number of the task.
|
||||
*/
|
||||
this.get = function(repoName, number, step) {
|
||||
return $http.get('/api/logs/'+repoName+'/'+number+'/'+step);
|
||||
return $http.get('/api/repos/'+repoName+'/builds/'+number+'/tasks/'+step+'/log');
|
||||
};
|
||||
}
|
||||
|
||||
angular
|
||||
.module('drone')
|
||||
.service('logs', LogService);
|
||||
})();
|
||||
})();
|
||||
|
|
|
@ -43,7 +43,7 @@
|
|||
* @param {object} JSON representation of a repository.
|
||||
*/
|
||||
this.update = function(repo) {
|
||||
return $http.put('/api/repos/'+repo.full_name, repo);
|
||||
return $http.patch('/api/repos/'+repo.full_name, repo);
|
||||
};
|
||||
|
||||
/**
|
||||
|
@ -61,7 +61,7 @@
|
|||
* @param {string} Name of the repository.
|
||||
*/
|
||||
this.watch = function(repoName) {
|
||||
return $http.post('/api/subscribers/'+repoName);
|
||||
return $http.post('/api/repos/'+repoName+'/watch');
|
||||
};
|
||||
|
||||
/**
|
||||
|
@ -70,11 +70,11 @@
|
|||
* @param {string} Name of the repository.
|
||||
*/
|
||||
this.unwatch = function(repoName) {
|
||||
return $http.delete('/api/subscribers/'+repoName);
|
||||
return $http.delete('/api/repos/'+repoName+'/unwatch');
|
||||
};
|
||||
}
|
||||
|
||||
angular
|
||||
.module('drone')
|
||||
.service('repos', RepoService);
|
||||
})();
|
||||
})();
|
||||
|
|
|
@ -15,7 +15,7 @@
|
|||
* @param {number} Number of the build.
|
||||
*/
|
||||
this.list = function(repoName, number) {
|
||||
return $http.get('/api/tasks/'+repoName+'/'+number);
|
||||
return $http.get('/api/repos/'+repoName+'/builds/'+number+'/tasks');
|
||||
};
|
||||
|
||||
/**
|
||||
|
@ -26,7 +26,7 @@
|
|||
* @param {number} Number of the task.
|
||||
*/
|
||||
this.get = function(repoName, number, step) {
|
||||
return $http.get('/api/tasks/'+repoName+'/'+name+'/'+step);
|
||||
return $http.get('/api/repos/'+repoName+'/builds/'+number+'/tasks/'+step);
|
||||
};
|
||||
|
||||
/**
|
||||
|
@ -37,11 +37,11 @@
|
|||
* @param {number} Number of the task.
|
||||
*/
|
||||
this.get = function(repoName, number, step) {
|
||||
return $http.get('/api/tasks/'+repoName+'/'+name+'/'+step);
|
||||
return $http.get('/api/repos/'+repoName+'/builds/'+number+'/tasks/'+step);
|
||||
};
|
||||
}
|
||||
|
||||
angular
|
||||
.module('drone')
|
||||
.service('tasks', TaskService);
|
||||
})();
|
||||
})();
|
||||
|
|
|
@ -45,7 +45,7 @@
|
|||
* Updates an existing user
|
||||
*/
|
||||
this.put = function(user) {
|
||||
return $http.put('/api/users/'+user.login, user);
|
||||
return $http.patch('/api/users/'+user.login, user);
|
||||
};
|
||||
|
||||
/**
|
||||
|
|
|
@ -16,12 +16,12 @@ import (
|
|||
// GET /api/status/:owner/:name/:number/:context
|
||||
//
|
||||
func GetStatus(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
repo := ToRepo(c)
|
||||
num, _ := strconv.Atoi(c.Params.ByName("number"))
|
||||
ctx := c.Params.ByName("context")
|
||||
|
||||
status, err := ds.GetBuildStatus(repo.FullName, num, ctx)
|
||||
status, err := store.Status(repo.FullName, num, ctx)
|
||||
if err != nil {
|
||||
c.Fail(404, err)
|
||||
} else {
|
||||
|
@ -36,7 +36,7 @@ func GetStatus(c *gin.Context) {
|
|||
// POST /api/status/:owner/:name/:number
|
||||
//
|
||||
func PostStatus(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
repo := ToRepo(c)
|
||||
num, err := strconv.Atoi(c.Params.ByName("number"))
|
||||
if err != nil {
|
||||
|
@ -48,7 +48,7 @@ func PostStatus(c *gin.Context) {
|
|||
c.AbortWithStatus(400)
|
||||
return
|
||||
}
|
||||
if err := ds.InsertBuildStatus(repo.Name, num, in); err != nil {
|
||||
if err := store.SetStatus(repo.Name, num, in); err != nil {
|
||||
c.Fail(400, err)
|
||||
} else {
|
||||
c.JSON(201, in)
|
||||
|
@ -62,11 +62,11 @@ func PostStatus(c *gin.Context) {
|
|||
// GET /api/status/:owner/:name/:number/:context
|
||||
//
|
||||
func GetStatusList(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
repo := ToRepo(c)
|
||||
num, _ := strconv.Atoi(c.Params.ByName("number"))
|
||||
|
||||
list, err := ds.GetBuildStatusList(repo.FullName, num)
|
||||
list, err := store.StatusList(repo.FullName, num)
|
||||
if err != nil {
|
||||
c.Fail(404, err)
|
||||
} else {
|
||||
|
|
|
@ -15,7 +15,7 @@ func Unsubscribe(c *gin.Context) {
|
|||
repo := ToRepo(c)
|
||||
user := ToUser(c)
|
||||
|
||||
err := store.DeleteSubscriber(user.Login, repo.FullName)
|
||||
err := store.DelSubscriber(user.Login, repo.FullName)
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
} else {
|
||||
|
@ -33,7 +33,7 @@ func Subscribe(c *gin.Context) {
|
|||
repo := ToRepo(c)
|
||||
user := ToUser(c)
|
||||
|
||||
err := store.InsertSubscriber(user.Login, repo.FullName)
|
||||
err := store.SetSubscriber(user.Login, repo.FullName)
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
} else {
|
||||
|
|
|
@ -14,12 +14,12 @@ import (
|
|||
// GET /api/tasks/:owner/:name/:number/:task
|
||||
//
|
||||
func GetTask(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
repo := ToRepo(c)
|
||||
b, _ := strconv.Atoi(c.Params.ByName("number"))
|
||||
t, _ := strconv.Atoi(c.Params.ByName("task"))
|
||||
|
||||
task, err := ds.GetTask(repo.FullName, b, t)
|
||||
task, err := store.Task(repo.FullName, b, t)
|
||||
if err != nil {
|
||||
c.Fail(404, err)
|
||||
} else {
|
||||
|
@ -34,11 +34,11 @@ func GetTask(c *gin.Context) {
|
|||
// GET /api/tasks/:owner/:name/:number
|
||||
//
|
||||
func GetTasks(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
repo := ToRepo(c)
|
||||
num, _ := strconv.Atoi(c.Params.ByName("number"))
|
||||
|
||||
tasks, err := ds.GetTaskList(repo.FullName, num)
|
||||
tasks, err := store.TaskList(repo.FullName, num)
|
||||
if err != nil {
|
||||
c.Fail(404, err)
|
||||
} else {
|
||||
|
@ -53,18 +53,18 @@ func GetTasks(c *gin.Context) {
|
|||
// GET /api/logs/:owner/:name/:number/:task
|
||||
//
|
||||
func GetTaskLogs(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
repo := ToRepo(c)
|
||||
full, _ := strconv.ParseBool(c.Params.ByName("full"))
|
||||
build, _ := strconv.Atoi(c.Params.ByName("number"))
|
||||
task, _ := strconv.Atoi(c.Params.ByName("task"))
|
||||
|
||||
logs, err := ds.GetTaskLogs(repo.FullName, build, task)
|
||||
r, err := store.LogReader(repo.FullName, build, task)
|
||||
if err != nil {
|
||||
c.Fail(404, err)
|
||||
} else if full {
|
||||
io.Copy(c.Writer, logs)
|
||||
io.Copy(c.Writer, r)
|
||||
} else {
|
||||
io.Copy(c.Writer, io.LimitReader(logs, 2000000))
|
||||
io.Copy(c.Writer, io.LimitReader(r, 2000000))
|
||||
}
|
||||
}
|
||||
|
|
|
@ -36,7 +36,7 @@ func PostToken(c *gin.Context) {
|
|||
token.Kind = common.TokenUser
|
||||
token.Issued = time.Now().UTC().Unix()
|
||||
|
||||
err := store.InsertToken(token)
|
||||
err := store.SetToken(token)
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
}
|
||||
|
@ -57,11 +57,11 @@ func DelToken(c *gin.Context) {
|
|||
user := ToUser(c)
|
||||
label := c.Params.ByName("label")
|
||||
|
||||
token, err := store.GetToken(user.Login, label)
|
||||
token, err := store.Token(user.Login, label)
|
||||
if err != nil {
|
||||
c.Fail(404, err)
|
||||
}
|
||||
err = store.DeleteToken(token)
|
||||
err = store.DelToken(token)
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
}
|
||||
|
|
|
@ -24,20 +24,20 @@ func GetUserCurr(c *gin.Context) {
|
|||
// PUT /api/user
|
||||
//
|
||||
func PutUserCurr(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
me := ToUser(c)
|
||||
store := ToDatastore(c)
|
||||
user := ToUser(c)
|
||||
|
||||
in := &common.User{}
|
||||
if !c.BindWith(in, binding.JSON) {
|
||||
return
|
||||
}
|
||||
me.Email = in.Email
|
||||
me.Gravatar = gravatar.Generate(in.Email)
|
||||
err := ds.UpdateUser(me)
|
||||
user.Email = in.Email
|
||||
user.Gravatar = gravatar.Generate(in.Email)
|
||||
err := store.SetUser(user)
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
} else {
|
||||
c.JSON(200, me)
|
||||
c.JSON(200, user)
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -48,9 +48,9 @@ func PutUserCurr(c *gin.Context) {
|
|||
// GET /api/user/repos
|
||||
//
|
||||
func GetUserRepos(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
me := ToUser(c)
|
||||
repos, err := ds.GetUserRepos(me.Login)
|
||||
store := ToDatastore(c)
|
||||
user := ToUser(c)
|
||||
repos, err := store.RepoList(user.Login)
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
} else {
|
||||
|
@ -65,9 +65,9 @@ func GetUserRepos(c *gin.Context) {
|
|||
// GET /api/user/tokens
|
||||
//
|
||||
func GetUserTokens(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
me := ToUser(c)
|
||||
tokens, err := ds.GetUserTokens(me.Login)
|
||||
store := ToDatastore(c)
|
||||
user := ToUser(c)
|
||||
tokens, err := store.TokenList(user.Login)
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
} else {
|
||||
|
|
|
@ -14,8 +14,8 @@ import (
|
|||
// GET /api/users
|
||||
//
|
||||
func GetUsers(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
users, err := ds.GetUserList()
|
||||
store := ToDatastore(c)
|
||||
users, err := store.UserList()
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
} else {
|
||||
|
@ -30,12 +30,12 @@ func GetUsers(c *gin.Context) {
|
|||
// POST /api/users
|
||||
//
|
||||
func PostUser(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
name := c.Params.ByName("name")
|
||||
user := &common.User{Login: name, Name: name}
|
||||
user.Token = c.Request.FormValue("token")
|
||||
user.Secret = c.Request.FormValue("secret")
|
||||
if err := ds.InsertUser(user); err != nil {
|
||||
if err := store.SetUserNotExists(user); err != nil {
|
||||
c.Fail(400, err)
|
||||
} else {
|
||||
c.JSON(201, user)
|
||||
|
@ -49,9 +49,9 @@ func PostUser(c *gin.Context) {
|
|||
// GET /api/users/:name
|
||||
//
|
||||
func GetUser(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
name := c.Params.ByName("name")
|
||||
user, err := ds.GetUser(name)
|
||||
user, err := store.User(name)
|
||||
if err != nil {
|
||||
c.Fail(404, err)
|
||||
} else {
|
||||
|
@ -66,10 +66,10 @@ func GetUser(c *gin.Context) {
|
|||
// PUT /api/users/:name
|
||||
//
|
||||
func PutUser(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
me := ToUser(c)
|
||||
name := c.Params.ByName("name")
|
||||
user, err := ds.GetUser(name)
|
||||
user, err := store.User(name)
|
||||
if err != nil {
|
||||
c.Fail(404, err)
|
||||
return
|
||||
|
@ -88,7 +88,7 @@ func PutUser(c *gin.Context) {
|
|||
user.Admin = in.Admin
|
||||
}
|
||||
|
||||
err = ds.UpdateUser(user)
|
||||
err = store.SetUser(user)
|
||||
if err != nil {
|
||||
c.Fail(400, err)
|
||||
} else {
|
||||
|
@ -103,10 +103,10 @@ func PutUser(c *gin.Context) {
|
|||
// DELETE /api/users/:name
|
||||
//
|
||||
func DeleteUser(c *gin.Context) {
|
||||
ds := ToDatastore(c)
|
||||
store := ToDatastore(c)
|
||||
me := ToUser(c)
|
||||
name := c.Params.ByName("name")
|
||||
user, err := ds.GetUser(name)
|
||||
user, err := store.User(name)
|
||||
if err != nil {
|
||||
c.Fail(404, err)
|
||||
return
|
||||
|
@ -119,7 +119,7 @@ func DeleteUser(c *gin.Context) {
|
|||
return
|
||||
}
|
||||
|
||||
if err := ds.DeleteUser(user); err != nil {
|
||||
if err := store.DelUser(user); err != nil {
|
||||
c.Fail(400, err)
|
||||
} else {
|
||||
c.Writer.WriteHeader(204)
|
||||
|
|
Loading…
Reference in a new issue