noury/mirror-nourybot
1
0
Fork 0
mirror of https://github.com/lyx0/nourybot.git synced 2024-11-13 19:49:55 +01:00
Code Issues Packages Projects Releases Wiki Activity

escape query string for predb

Browse source
This commit is contained in:
lyx0 2023-12-13 23:20:05 +01:00
parent 3642ef8a30
commit 33f91a6333
2 changed files with 63 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cmd/nourybot/commands.go
View file

@ -156,6 +156,10 @@ func (app *application) handleCommand(message twitch.PrivateMessage) {
if userLevel >= 100 && len(message.Message) > 16 {
reply = app.PreDBSearch(message.Message[15:len(message.Message)])
}
case "group":
if userLevel >= 100 && len(message.Message) > 15 {
reply = app.PreDBGroup(message.Message[14:len(message.Message)])
}
}
case "phonetic":

62
cmd/nourybot/predb.go
View file

@ -5,6 +5,7 @@ import (
"fmt"
"io/ioutil"
"net/http"
"net/url"
"strings"
)
@ -107,10 +108,65 @@ type searchSingleResult struct {
}
func (app *application) PreDBSearch(title string) string {
baseUrl := fmt.Sprintf("https://api.predb.net/?q=%s&order_by=release&sort=asc&limit=100", title)
escaped := fmt.Sprintf("https://api.predb.net/?q=%s&order_by=release&sort=asc&limit=100", url.QueryEscape(title))
app.Log.Info(title)
resp, err := http.Get(baseUrl)
resp, err := http.Get(escaped)
if err != nil {
app.Log.Error(err)
}
app.Log.Info(resp)
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
app.Log.Error(err)
}
app.Log.Info(body)
var release searchFullResult
_ = json.Unmarshal([]byte(body), &release)
var ts []string
app.Log.Info(release)
for i := 0; i < release.Results; i++ {
t := fmt.Sprintf("ID: %v\nRelease timestamp: %v\nRelease: %v\nSection: %v\nFiles: %v\nSize: %v\nStatus: %v\nReason: %v\nRelease group: %v\nRelease genre: %v\npredb.net: %v\nNFO URL: %v\nNFO Image URL: %v\n\n",
release.Data[i].ID,
release.Data[i].PreTime,
release.Data[i].Release,
release.Data[i].Section,
release.Data[i].Files,
release.Data[i].Size,
release.Data[i].Status,
release.Data[i].Reason,
release.Data[i].Group,
release.Data[i].Genre,
fmt.Sprint("https://predb.net"+release.Data[i].URL),
release.Data[i].NFO,
release.Data[i].NFOImage,
)
ts = append(ts, t)
}
// reply, err := app.uploadPaste(strings.Join(ts, ""))
// if err != nil {
// app.Log.Errorw("Error trying to retrieve all timers from database", err)
// return ""
// }
reply := app.YafUploadString(strings.Join(ts, ""))
if err != nil {
app.Log.Errorw("Error trying to retrieve all timers from database", err)
return ""
}
return reply
}
func (app *application) PreDBGroup(group string) string {
escaped := fmt.Sprintf("https://api.predb.net/?group=%s&order_by=release&sort=asc&limit=100", url.QueryEscape(group))
resp, err := http.Get(escaped)
if err != nil {
app.Log.Error(err)
}