RievoGitAdmin/discordBots
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add more image urls

Browse source
This commit is contained in:
Seraphim Strub 2024-05-30 22:01:13 +00:00
parent 0b97f282e2
commit 3a076998f9
5 changed files with 74 additions and 75 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

85
cmd/dealsbot/api/gog.go
View file

@ -6,7 +6,6 @@ import (
"log/slog"
"net/http"
"regexp"
"strings"
)
type GogStruct struct {
@ -67,7 +66,7 @@ func (e GogStruct) Load() error {
return
case tt == html.StartTagToken:
t := bodyStore.Token()
if t.Data != "a" {
if t.Data != "a" { // only <a> tag
continue
}
for _, a := range t.Attr {
@ -105,44 +104,60 @@ func (e GogStruct) Load() error {
}
bodyGame := html.NewTokenizer(resGame.Body)
func() {
for {
tt := bodyGame.Next()
switch {
case tt == html.ErrorToken:
// file end or error
return
case tt == html.StartTagToken:
t := bodyGame.Token()
if t.Data != "h1" {
continue
}
for _, a := range t.Attr {
if !(a.Key == "class" && a.Val == "productcard-basics__title") {
continue
}
if tt = bodyGame.Next(); tt != html.TextToken {
continue
}
id := fmt.Sprintf("%v%v", e.idPrefix, appID)
title := strings.TrimSpace(bodyGame.Token().Data)
url := fmt.Sprintf("%v%v", e.baseUrl, appID)
e.deals[id] = Deal{
Id: id,
Title: title,
Url: url,
}
}
}
}
}()
title, image := gogParseMeta(bodyGame)
url := fmt.Sprintf("%v%v", e.baseUrl, appID)
id := fmt.Sprintf("%v%v", e.idPrefix, appID)
e.deals[id] = Deal{
Id: id,
Title: title,
Url: url,
Image: image,
}
}
return nil
}
func gogParseMeta(htmlBody *html.Tokenizer) (string, string) {
var title, image string
func() {
for {
tt := htmlBody.Next()
switch {
case tt == html.ErrorToken:
// file end or error
return
case tt == html.StartTagToken:
t := htmlBody.Token()
if t.Data != "meta" {
continue
}
for _, a := range t.Attr {
if a.Key == "property" && a.Val == "og:title" {
for _, c := range t.Attr {
if c.Key == "content" {
title = c.Val
}
}
}
if a.Key == "property" && a.Val == "og:image" {
for _, c := range t.Attr {
if c.Key == "content" {
image = c.Val
}
}
}
}
if title != "" && image != "" {
return
}
}
}
}()
return title, image
}
func (e GogStruct) Get() []Deal {
var deals []Deal
for _, deal := range e.deals {

45
cmd/dealsbot/api/gogfront.go
View file

@ -6,7 +6,6 @@ import (
"log/slog"
"net/http"
"regexp"
"strings"
)
type GogFrontStruct struct {
@ -67,7 +66,7 @@ func (e GogFrontStruct) Load() error {
return
case tt == html.StartTagToken:
t := bodyStore.Token()
if t.Data != "a" {
if t.Data != "a" { // only <a> tag
continue
}
for _, a := range t.Attr {
@ -105,39 +104,15 @@ func (e GogFrontStruct) Load() error {
}
bodyGame := html.NewTokenizer(resGame.Body)
func() {
for {
tt := bodyGame.Next()
switch {
case tt == html.ErrorToken:
// file end or error
return
case tt == html.StartTagToken:
t := bodyGame.Token()
if t.Data != "h1" {
continue
}
for _, a := range t.Attr {
if !(a.Key == "class" && a.Val == "productcard-basics__title") {
continue
}
if tt = bodyGame.Next(); tt != html.TextToken {
continue
}
id := fmt.Sprintf("%v%v", e.idPrefix, appID)
title := strings.TrimSpace(bodyGame.Token().Data)
url := fmt.Sprintf("%v%v", e.baseUrl, appID)
e.deals[id] = Deal{
Id: id,
Title: title,
Url: url,
}
}
}
}
}()
title, image := gogParseMeta(bodyGame)
url := fmt.Sprintf("%v%v", e.baseUrl, appID)
id := fmt.Sprintf("%v%v", e.idPrefix, appID)
e.deals[id] = Deal{
Id: id,
Title: title,
Url: url,
Image: image,
}
}
return nil

3
cmd/dealsbot/api/humblebundle.go
View file

@ -20,6 +20,7 @@ type HumbleBundleStruct struct {
}
func NewHumbleBundleApi(logger *slog.Logger) HumbleBundleStruct {
// TODO: does this still work and if yes, get image
humbleBundle := HumbleBundleStruct{
url: "https://www.humblebundle.com/",
baseUrl: "https://www.humblebundle.com/store/",
@ -80,7 +81,7 @@ func (e HumbleBundleStruct) Load() error {
return nil
case tt == html.StartTagToken:
t := bodyStore.Token()
if t.Data != "script" {
if t.Data != "script" { // only <script> tag
continue
}
for _, a := range t.Attr {

7
cmd/dealsbot/api/steam.go
View file

@ -42,6 +42,7 @@ type steamApiBodyGame struct {
Type string `json:"type"`
Name string `json:"name"`
IsFree bool `json:"is_free"`
HeaderImage string `json:"header_image"`
PriceOverview struct {
Currency string `json:"currency"`
Initial int `json:"initial"`
@ -95,7 +96,7 @@ func (e SteamStruct) Load() error {
return
case tt == html.StartTagToken:
t := bodyStore.Token()
if t.Data != "a" {
if t.Data != "a" { // only <a> tag
continue
}
for _, a := range t.Attr {
@ -139,6 +140,9 @@ func (e SteamStruct) Load() error {
if game.Data.Type != "game" {
continue
}
image := game.Data.HeaderImage
id := fmt.Sprintf("%v%v", e.idPrefix, appID)
title := strings.TrimSpace(game.Data.Name)
url := fmt.Sprintf("%v%v", e.baseUrl, appID)
@ -147,6 +151,7 @@ func (e SteamStruct) Load() error {
Id: id,
Title: title,
Url: url,
Image: image,
}
}
}

9
cmd/dealsbot/api/ubisoft.go
View file

@ -37,9 +37,10 @@ func NewUbsioftApi(logger *slog.Logger) UbisoftStruct {
type ubisoftApiBody struct {
News []struct {
Type string `json:"type"`
Title string `json:"title"`
Links []struct {
Type string `json:"type"`
Title string `json:"title"`
MediaUrl string `json:"mediaURL"`
Links []struct {
Param string `json:"param"`
} `json:"links"`
} `json:"news"`
@ -128,11 +129,13 @@ func (e UbisoftStruct) Load() error {
id := fmt.Sprintf("%v%v", e.idPrefix, idFromUrl[1])
url := news.Links[0].Param
image := news.MediaUrl
e.deals[id] = Deal{
Id: id,
Title: title,
Url: url,
Image: image,
}
}