77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
package app
|
|
|
|
import (
|
|
"encoding/json"
|
|
"math/rand"
|
|
"strings"
|
|
|
|
"git.macaw.me/skunky/devianter"
|
|
)
|
|
|
|
type API struct {
|
|
main *skunkyart
|
|
}
|
|
|
|
type info struct {
|
|
Version string `json:"version"`
|
|
Settings settingsParams `json:"settings"`
|
|
}
|
|
|
|
func (a API) Info() {
|
|
json, err := json.Marshal(info{
|
|
Version: a.main.Version,
|
|
Settings: settingsParams{
|
|
Nsfw: CFG.Nsfw,
|
|
Proxy: CFG.Proxy,
|
|
},
|
|
})
|
|
try(err)
|
|
a.main.Writer.Write(json)
|
|
}
|
|
|
|
func (a API) Error(description string, status int) {
|
|
a.main.Writer.WriteHeader(status)
|
|
var response strings.Builder
|
|
response.WriteString(`{"error":"`)
|
|
response.WriteString(description)
|
|
response.WriteString(`"}`)
|
|
wr(a.main.Writer, response.String())
|
|
}
|
|
|
|
func (a API) sendMedia(d *devianter.Deviation) {
|
|
mediaUrl, name := devianter.UrlFromMedia(d.Media)
|
|
a.main.SetFilename(name)
|
|
|
|
if len(mediaUrl) != 0 {
|
|
mediaUrl = mediaUrl[21:]
|
|
dot := strings.Index(mediaUrl, ".")
|
|
a.main.Writer.Header().Del("Content-Type")
|
|
a.main.DownloadAndSendMedia(mediaUrl[:dot], mediaUrl[dot+11:])
|
|
}
|
|
}
|
|
|
|
// TODO: сделать фильтры
|
|
func (a API) Random() {
|
|
for attempt := 1; ; {
|
|
if attempt > 3 {
|
|
a.Error("Sorry, butt NSFW on this are disabled, and the instance failed to find a random art without NSFW", 500)
|
|
}
|
|
|
|
s, err, daErr := devianter.PerformSearch(string(rand.Intn(999)), rand.Intn(30), 'a')
|
|
try(err)
|
|
if daErr.RAW != nil {
|
|
continue
|
|
}
|
|
|
|
deviation := &s.Results[rand.Intn(len(s.Results))]
|
|
|
|
if deviation.NSFW && !CFG.Nsfw {
|
|
attempt++
|
|
continue
|
|
}
|
|
|
|
a.sendMedia(deviation)
|
|
return
|
|
}
|
|
}
|