diff options
Diffstat (limited to 'system')
| -rw-r--r-- | system/db/item.go | 8 | ||||
| -rw-r--r-- | system/db/query.go | 156 | ||||
| -rw-r--r-- | system/server.go | 85 |
3 files changed, 235 insertions, 14 deletions
diff --git a/system/db/item.go b/system/db/item.go deleted file mode 100644 index 23e8553..0000000 --- a/system/db/item.go +++ /dev/null @@ -1,8 +0,0 @@ -// Package db ... -package db - -// Item should only be embedded into content type structs. -// Helper for DB-related actions -type Item struct { - ID int `json:"_id"` -} diff --git a/system/db/query.go b/system/db/query.go index c9cf37c..06e9c62 100644 --- a/system/db/query.go +++ b/system/db/query.go @@ -1,17 +1,161 @@ package db +import ( + "bytes" + "encoding/json" + "fmt" + "log" + "net/url" + "strconv" + "strings" + + "github.com/boltdb/bolt" + "github.com/gorilla/schema" + "github.com/nilslice/cms/content" +) + +var store *bolt.DB + +func init() { + var err error + store, err = bolt.Open("store.db", 0666, nil) + if err != nil { + log.Fatal(err) + } +} + // Set inserts or updates values in the database. -// The `key` argument is a string made up of namespace:id (string:int) -func Set(key string) error { +// The `target` argument is a string made up of namespace:id (string:int) +func Set(target string, data url.Values) (int, error) { + t := strings.Split(target, ":") + ns, id := t[0], t[1] - return nil + // check if content has an id, and if not get new one from target bucket + if len(id) == 0 { + return insert(ns, data) + } + + return update(ns, id, data) +} + +func update(ns, id string, data url.Values) (int, error) { + cid, err := strconv.Atoi(id) + if err != nil { + return 0, err + } + + err = store.Update(func(tx *bolt.Tx) error { + b, err := tx.CreateBucketIfNotExists([]byte(ns)) + if err != nil { + return err + } + + j, err := toJSON(ns, data) + if err != nil { + return err + } + + err = b.Put([]byte(fmt.Sprintf("%d", cid)), j) + if err != nil { + return err + } + + return nil + }) + if err != nil { + return 0, nil + } + + return cid, nil +} + +func insert(ns string, data url.Values) (int, error) { + var effectedID int + err := store.Update(func(tx *bolt.Tx) error { + b, err := tx.CreateBucketIfNotExists([]byte(ns)) + if err != nil { + return err + } + + // get the next available ID and convert to string + // also set effectedID to int of ID + id, err := b.NextSequence() + if err != nil { + return err + } + cid := strconv.FormatUint(id, 10) + effectedID, err = strconv.Atoi(cid) + if err != nil { + return err + } + data.Add("id", cid) + + j, err := toJSON(ns, data) + if err != nil { + return err + } + + err = b.Put([]byte(cid), j) + if err != nil { + return err + } + + return nil + }) + if err != nil { + return 0, err + } + + return effectedID, nil +} + +func toJSON(ns string, data url.Values) ([]byte, error) { + // find the content type and decode values into it + t, ok := content.Types[ns] + if !ok { + return nil, fmt.Errorf(content.ErrTypeNotRegistered, ns) + } + post := t() + + dec := schema.NewDecoder() + dec.SetAliasTag("json") // allows simpler struct tagging when creating a content type + dec.IgnoreUnknownKeys(true) // will skip over form values submitted, but not in struct + err := dec.Decode(post, data) + if err != nil { + return nil, err + } + + // marshall content struct to json for db storage + j, err := json.Marshal(post) + if err != nil { + return nil, err + } + + return j, nil } // Get retrives one item from the database. Non-existent values will return an empty []byte -// The `key` argument is a string made up of namespace:id (string:int) -func Get(key string) []byte { +// The `target` argument is a string made up of namespace:id (string:int) +func Get(target string) ([]byte, error) { + t := strings.Split(target, ":") + ns, id := t[0], t[1] - return nil + val := &bytes.Buffer{} + err := store.View(func(tx *bolt.Tx) error { + b := tx.Bucket([]byte(ns)) + _, err := val.Write(b.Get([]byte(id))) + if err != nil { + fmt.Println(err) + return err + } + + return nil + }) + if err != nil { + return nil, err + } + + return val.Bytes(), nil } // GetAll retrives all items from the database within the provided namespace diff --git a/system/server.go b/system/server.go new file mode 100644 index 0000000..4a83706 --- /dev/null +++ b/system/server.go @@ -0,0 +1,85 @@ +package main + +import ( + "encoding/json" + "fmt" + "net/http" + + "github.com/nilslice/cms/content" + "github.com/nilslice/cms/management/editor" + "github.com/nilslice/cms/management/manager" + "github.com/nilslice/cms/system/db" +) + +func main() { + http.HandleFunc("/admin/edit", func(res http.ResponseWriter, req *http.Request) { + switch req.Method { + case http.MethodGet: + q := req.URL.Query() + i := q.Get("id") + t := q.Get("type") + contentType, ok := content.Types[t] + if !ok { + fmt.Fprintf(res, content.ErrTypeNotRegistered, t) + return + } + post := contentType() + + if i != "" { + fmt.Println("Need to show post id:", i, "(", t, ")") + + data, err := db.Get(t + ":" + i) + if err != nil { + fmt.Println(err) + res.WriteHeader(http.StatusInternalServerError) + return + } + + err = json.Unmarshal(data, post) + if err != nil { + fmt.Println(err) + res.WriteHeader(http.StatusInternalServerError) + return + } + } + + view, err := manager.Manage(post.(editor.Editable), t) + if err != nil { + fmt.Println(err) + res.WriteHeader(http.StatusInternalServerError) + return + } + res.Header().Set("Content-Type", "text/html") + res.Write(view) + + case http.MethodPost: + err := req.ParseForm() + if err != nil { + fmt.Println(err) + res.WriteHeader(http.StatusBadRequest) + return + } + + cid := req.FormValue("id") + t := req.FormValue("type") + fmt.Println("query data: t=", t, "id=", cid) + + id, err := db.Set(t+":"+cid, req.PostForm) + if err != nil { + fmt.Println(err) + res.WriteHeader(http.StatusInternalServerError) + return + } + + fmt.Println(t, "post created:", id) + scheme := req.URL.Scheme + host := req.URL.Host + path := req.URL.Path + desURL := scheme + host + path + "?type=" + t + "&id=" + fmt.Sprintf("%d", id) + http.Redirect(res, req, desURL, http.StatusFound) + } + }) + + http.ListenAndServe(":8080", nil) + +} |