126 lines
2.8 KiB
Go
126 lines
2.8 KiB
Go
package api
|
|
|
|
import (
|
|
"TheAdversary/config"
|
|
"TheAdversary/database"
|
|
"TheAdversary/schema"
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"go.uber.org/zap"
|
|
"gorm.io/gorm/clause"
|
|
"net/http"
|
|
"net/url"
|
|
"path"
|
|
"strconv"
|
|
)
|
|
|
|
var assetsPayload struct {
|
|
ArticleId int `json:"article_id"`
|
|
Content string `json:"content"`
|
|
}
|
|
|
|
func Assets(w http.ResponseWriter, r *http.Request) {
|
|
_, ok := authorizedSession(r)
|
|
if !ok {
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
return
|
|
}
|
|
|
|
switch r.Method {
|
|
case http.MethodGet:
|
|
assetsGet(w, r)
|
|
case http.MethodPost:
|
|
assetsPost(w, r)
|
|
case http.MethodDelete:
|
|
assetsDelete(w, r)
|
|
}
|
|
}
|
|
|
|
func assetsGet(w http.ResponseWriter, r *http.Request) {
|
|
query := r.URL.Query()
|
|
request := database.GetDB().Table("assets")
|
|
|
|
if query.Has("q") {
|
|
request.Where("LOWER(name) LIKE ?", "%"+query.Get("q")+"%")
|
|
}
|
|
limit := 20
|
|
if query.Has("limit") {
|
|
var err error
|
|
limit, err = strconv.Atoi(query.Get("limit"))
|
|
if err != nil {
|
|
ApiError{"invalid 'limit' parameter", http.StatusUnprocessableEntity}.Send(w)
|
|
return
|
|
}
|
|
}
|
|
request.Limit(limit)
|
|
|
|
var assets []schema.Asset
|
|
request.Find(&assets)
|
|
|
|
for _, asset := range assets {
|
|
asset.Link = "/" + path.Join(config.Prefix, "assets", asset.Link)
|
|
}
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(&assets)
|
|
}
|
|
|
|
type assetsPostPayload struct {
|
|
Name string `json:"name"`
|
|
Content string `json:"data"`
|
|
}
|
|
|
|
func assetsPost(w http.ResponseWriter, r *http.Request) {
|
|
var payload assetsPostPayload
|
|
if err := json.NewDecoder(r.Body).Decode(&payload); err != nil {
|
|
InvalidJson.Send(w)
|
|
return
|
|
}
|
|
|
|
rawData, err := base64.StdEncoding.DecodeString(payload.Content)
|
|
if err != nil {
|
|
zap.S().Warnf("Cannot decode base64")
|
|
ApiError{Message: "invalid base64 content", Code: http.StatusUnprocessableEntity}.Send(w)
|
|
return
|
|
}
|
|
|
|
tmpDatabaseSchema := struct {
|
|
Id int
|
|
Name string
|
|
Data []byte
|
|
Link string
|
|
}{Name: payload.Name, Data: rawData, Link: url.PathEscape(payload.Name)}
|
|
|
|
if database.GetDB().Table("assets").Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "name"}},
|
|
DoNothing: true,
|
|
}).Create(&tmpDatabaseSchema).RowsAffected == 0 {
|
|
w.WriteHeader(http.StatusConflict)
|
|
} else {
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(schema.Asset{
|
|
Id: tmpDatabaseSchema.Id,
|
|
Name: tmpDatabaseSchema.Name,
|
|
Link: "/" + path.Join(config.Prefix, "assets", tmpDatabaseSchema.Link),
|
|
})
|
|
}
|
|
}
|
|
|
|
type assetsDeletePayload struct {
|
|
Id int `json:"id"`
|
|
}
|
|
|
|
func assetsDelete(w http.ResponseWriter, r *http.Request) {
|
|
var payload assetsDeletePayload
|
|
if err := json.NewDecoder(r.Body).Decode(&payload); err != nil {
|
|
InvalidJson.Send(w)
|
|
return
|
|
}
|
|
|
|
if database.GetDB().Table("assets").Delete(schema.Asset{}, payload.Id).RowsAffected == 0 {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
} else {
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
}
|