mirror of https://github.com/usememos/memos
update restful api
steven
3 years ago
parent
bdaeb3a68b
commit
a08ad0ebab
@ -1,4 +1,4 @@
|
||||
package error
|
||||
package e
|
||||
|
||||
var Codes = map[string]int{
|
||||
"NOT_AUTH": 20001,
|
||||
@ -0,0 +1,110 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"memos/api/e"
|
||||
"memos/store"
|
||||
"net/http"
|
||||
|
||||
"github.com/gorilla/mux"
|
||||
)
|
||||
|
||||
func handleGetMyQueries(w http.ResponseWriter, r *http.Request) {
|
||||
userId, _ := GetUserIdInCookie(r)
|
||||
|
||||
queries, err := store.GetQueriesByUserId(userId)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "DATABASE_ERROR", err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
json.NewEncoder(w).Encode(Response{
|
||||
Succeed: true,
|
||||
Message: "",
|
||||
Data: queries,
|
||||
})
|
||||
}
|
||||
|
||||
type QueryPut struct {
|
||||
Title string `json:"title"`
|
||||
Querystring string `json:"querystring"`
|
||||
}
|
||||
|
||||
func handleCreateQuery(w http.ResponseWriter, r *http.Request) {
|
||||
userId, _ := GetUserIdInCookie(r)
|
||||
|
||||
queryPut := QueryPut{}
|
||||
err := json.NewDecoder(r.Body).Decode(&queryPut)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "REQUEST_BODY_ERROR", "Bad request")
|
||||
return
|
||||
}
|
||||
|
||||
query, err := store.CreateNewQuery(queryPut.Title, queryPut.Querystring, userId)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "DATABASE_ERROR", err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
json.NewEncoder(w).Encode(Response{
|
||||
Succeed: true,
|
||||
Message: "",
|
||||
Data: query,
|
||||
})
|
||||
}
|
||||
|
||||
func handleUpdateQuery(w http.ResponseWriter, r *http.Request) {
|
||||
vars := mux.Vars(r)
|
||||
queryId := vars["id"]
|
||||
|
||||
queryPatch := store.QueryPatch{}
|
||||
err := json.NewDecoder(r.Body).Decode(&queryPatch)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "REQUEST_BODY_ERROR", "Bad request")
|
||||
return
|
||||
}
|
||||
|
||||
query, err := store.UpdateQuery(queryId, &queryPatch)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "DATABASE_ERROR", err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
json.NewEncoder(w).Encode(Response{
|
||||
Succeed: true,
|
||||
Message: "",
|
||||
Data: query,
|
||||
})
|
||||
}
|
||||
|
||||
func handleDeleteQuery(w http.ResponseWriter, r *http.Request) {
|
||||
vars := mux.Vars(r)
|
||||
queryId := vars["id"]
|
||||
|
||||
_, err := store.DeleteQuery(queryId)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "DATABASE_ERROR", err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
json.NewEncoder(w).Encode(Response{
|
||||
Succeed: true,
|
||||
Message: "",
|
||||
Data: nil,
|
||||
})
|
||||
}
|
||||
|
||||
func RegisterQueryRoutes(r *mux.Router) {
|
||||
queryRouter := r.PathPrefix("/api/query").Subrouter()
|
||||
|
||||
queryRouter.HandleFunc("/all", handleGetMyQueries).Methods("GET")
|
||||
queryRouter.HandleFunc("/", handleCreateQuery).Methods("PUT")
|
||||
queryRouter.HandleFunc("/{id}", handleUpdateQuery).Methods("PATCH")
|
||||
queryRouter.HandleFunc("/{id}", handleDeleteQuery).Methods("DELETE")
|
||||
}
|
||||
@ -0,0 +1,34 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
type SPAHandler struct {
|
||||
StaticPath string
|
||||
IndexPath string
|
||||
}
|
||||
|
||||
func (h SPAHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||
path, err := filepath.Abs(r.URL.Path)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
path = filepath.Join(h.StaticPath, path)
|
||||
|
||||
_, err = os.Stat(path)
|
||||
if os.IsNotExist(err) {
|
||||
// file does not exist, serve index.html
|
||||
http.ServeFile(w, r, filepath.Join(h.StaticPath, h.IndexPath))
|
||||
return
|
||||
} else if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
http.FileServer(http.Dir(h.StaticPath)).ServeHTTP(w, r)
|
||||
}
|
||||
@ -1,3 +1,6 @@
|
||||
/*
|
||||
* Re-create tables and insert initial data(todo)
|
||||
*/
|
||||
CREATE TABLE `users` (
|
||||
`id` TEXT NOT NULL PRIMARY KEY,
|
||||
`username` TEXT NOT NULL,
|
||||
Binary file not shown.
@ -0,0 +1,99 @@
|
||||
package store
|
||||
|
||||
import (
|
||||
"memos/common"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Query struct {
|
||||
Id string `json:"id"`
|
||||
UserId string `json:"userId"`
|
||||
Title string `json:"title"`
|
||||
Querystring string `json:"querystring"`
|
||||
PinnedAt string `json:"pinnedAt"`
|
||||
CreatedAt string `json:"createdAt"`
|
||||
UpdatedAt string `json:"updatedAt"`
|
||||
}
|
||||
|
||||
func CreateNewQuery(title string, querystring string, userId string) (Query, error) {
|
||||
nowDateTimeStr := common.GetNowDateTimeStr()
|
||||
newQuery := Query{
|
||||
Id: common.GenUUID(),
|
||||
Title: title,
|
||||
Querystring: querystring,
|
||||
UserId: userId,
|
||||
PinnedAt: "",
|
||||
CreatedAt: nowDateTimeStr,
|
||||
UpdatedAt: nowDateTimeStr,
|
||||
}
|
||||
|
||||
query := `INSERT INTO queries (id, title, querystring, user_id, pinned_at, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?)`
|
||||
_, err := DB.Exec(query, newQuery.Id, newQuery.Title, newQuery.Querystring, newQuery.UserId, newQuery.PinnedAt, newQuery.CreatedAt, newQuery.UpdatedAt)
|
||||
|
||||
return newQuery, err
|
||||
}
|
||||
|
||||
type QueryPatch struct {
|
||||
Title *string
|
||||
Querystring *string
|
||||
PinnedAt *string
|
||||
}
|
||||
|
||||
func UpdateQuery(id string, queryPatch *QueryPatch) (Query, error) {
|
||||
query, _ := GetQueryById(id)
|
||||
set, args := []string{}, []interface{}{}
|
||||
|
||||
if v := queryPatch.Title; v != nil {
|
||||
query.Title = *v
|
||||
set, args = append(set, "title=?"), append(args, *v)
|
||||
}
|
||||
if v := queryPatch.Querystring; v != nil {
|
||||
query.Querystring = *v
|
||||
set, args = append(set, "querystring=?"), append(args, *v)
|
||||
}
|
||||
if v := queryPatch.PinnedAt; v != nil {
|
||||
query.PinnedAt = *v
|
||||
set, args = append(set, "pinned_at=?"), append(args, *v)
|
||||
}
|
||||
set, args = append(set, "updated_at=?"), append(args, common.GetNowDateTimeStr())
|
||||
args = append(args, id)
|
||||
|
||||
sqlQuery := `UPDATE queries SET ` + strings.Join(set, ",") + ` WHERE id=?`
|
||||
_, err := DB.Exec(sqlQuery, args...)
|
||||
|
||||
return query, err
|
||||
}
|
||||
|
||||
func DeleteQuery(queryId string) (error, error) {
|
||||
query := `DELETE FROM queries WHERE id=?`
|
||||
_, err := DB.Exec(query, queryId)
|
||||
|
||||
return nil, err
|
||||
}
|
||||
|
||||
func GetQueryById(queryId string) (Query, error) {
|
||||
sqlQuery := `SELECT id, title, querystring, user_id, pinned_at, created_at, updated_at FROM queries WHERE id=?`
|
||||
query := Query{}
|
||||
err := DB.QueryRow(sqlQuery, queryId).Scan(&query.Id, &query.Title, &query.Querystring, &query.UserId, &query.PinnedAt, &query.CreatedAt, &query.UpdatedAt)
|
||||
return query, err
|
||||
}
|
||||
|
||||
func GetQueriesByUserId(userId string) ([]Query, error) {
|
||||
query := `SELECT id, title, querystring, user_id, pinned_at, created_at, updated_at FROM queries WHERE user_id=?`
|
||||
|
||||
rows, _ := DB.Query(query, userId)
|
||||
defer rows.Close()
|
||||
|
||||
queries := []Query{}
|
||||
|
||||
for rows.Next() {
|
||||
query := Query{}
|
||||
rows.Scan(&query.Id, &query.Title, &query.Querystring, &query.UserId, &query.PinnedAt, &query.CreatedAt, &query.UpdatedAt)
|
||||
|
||||
queries = append(queries, query)
|
||||
}
|
||||
|
||||
err := rows.Err()
|
||||
|
||||
return queries, err
|
||||
}
|
||||
Loading…
Reference in New Issue