mirror of
https://github.com/bestnite/igdb-database.git
synced 2025-06-17 10:13:18 +08:00
start
This commit is contained in:
185
collector/webhook.go
Normal file
185
collector/webhook.go
Normal file
@ -0,0 +1,185 @@
|
||||
package collector
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"igdb-database/config"
|
||||
"igdb-database/db"
|
||||
"igdb-database/model"
|
||||
"io"
|
||||
"log"
|
||||
"net"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"slices"
|
||||
|
||||
"github.com/bestnite/go-igdb"
|
||||
"github.com/bestnite/go-igdb/endpoint"
|
||||
"go.mongodb.org/mongo-driver/v2/mongo"
|
||||
)
|
||||
|
||||
func StartWebhookServer(client *igdb.Client) {
|
||||
baseUrl, err := url.Parse(config.C().ExtralUrl)
|
||||
if err != nil {
|
||||
log.Fatalf("failed to parse url: %v", err)
|
||||
}
|
||||
|
||||
http.HandleFunc(webhook(client.AgeRatingCategories))
|
||||
http.HandleFunc(webhook(client.AgeRatingContentDescriptions))
|
||||
http.HandleFunc(webhook(client.AgeRatingContentDescriptionsV2))
|
||||
http.HandleFunc(webhook(client.AgeRatingOrganizations))
|
||||
http.HandleFunc(webhook(client.AgeRatings))
|
||||
http.HandleFunc(webhook(client.AlternativeNames))
|
||||
http.HandleFunc(webhook(client.Artworks))
|
||||
http.HandleFunc(webhook(client.CharacterGenders))
|
||||
http.HandleFunc(webhook(client.CharacterMugShots))
|
||||
http.HandleFunc(webhook(client.Characters))
|
||||
http.HandleFunc(webhook(client.CharacterSpecies))
|
||||
http.HandleFunc(webhook(client.CollectionMemberships))
|
||||
http.HandleFunc(webhook(client.CollectionMembershipTypes))
|
||||
http.HandleFunc(webhook(client.CollectionRelations))
|
||||
http.HandleFunc(webhook(client.CollectionRelationTypes))
|
||||
http.HandleFunc(webhook(client.Collections))
|
||||
http.HandleFunc(webhook(client.CollectionTypes))
|
||||
http.HandleFunc(webhook(client.Companies))
|
||||
http.HandleFunc(webhook(client.CompanyLogos))
|
||||
http.HandleFunc(webhook(client.CompanyStatuses))
|
||||
http.HandleFunc(webhook(client.CompanyWebsites))
|
||||
http.HandleFunc(webhook(client.Covers))
|
||||
http.HandleFunc(webhook(client.DateFormats))
|
||||
http.HandleFunc(webhook(client.EventLogos))
|
||||
http.HandleFunc(webhook(client.EventNetworks))
|
||||
http.HandleFunc(webhook(client.Events))
|
||||
http.HandleFunc(webhook(client.ExternalGames))
|
||||
http.HandleFunc(webhook(client.ExternalGameSources))
|
||||
http.HandleFunc(webhook(client.Franchises))
|
||||
http.HandleFunc(webhook(client.GameEngineLogos))
|
||||
http.HandleFunc(webhook(client.GameEngines))
|
||||
http.HandleFunc(webhook(client.GameLocalizations))
|
||||
http.HandleFunc(webhook(client.GameModes))
|
||||
http.HandleFunc(webhook(client.GameReleaseFormats))
|
||||
http.HandleFunc(webhook(client.Games))
|
||||
http.HandleFunc(webhook(client.GameStatuses))
|
||||
http.HandleFunc(webhook(client.GameTimeToBeats))
|
||||
http.HandleFunc(webhook(client.GameTypes))
|
||||
http.HandleFunc(webhook(client.GameVersionFeatures))
|
||||
http.HandleFunc(webhook(client.GameVersionFeatureValues))
|
||||
http.HandleFunc(webhook(client.GameVersions))
|
||||
http.HandleFunc(webhook(client.GameVideos))
|
||||
http.HandleFunc(webhook(client.Genres))
|
||||
http.HandleFunc(webhook(client.InvolvedCompanies))
|
||||
http.HandleFunc(webhook(client.Keywords))
|
||||
http.HandleFunc(webhook(client.Languages))
|
||||
http.HandleFunc(webhook(client.LanguageSupports))
|
||||
http.HandleFunc(webhook(client.LanguageSupportTypes))
|
||||
http.HandleFunc(webhook(client.MultiplayerModes))
|
||||
http.HandleFunc(webhook(client.NetworkTypes))
|
||||
http.HandleFunc(webhook(client.PlatformFamilies))
|
||||
http.HandleFunc(webhook(client.PlatformLogos))
|
||||
http.HandleFunc(webhook(client.Platforms))
|
||||
http.HandleFunc(webhook(client.PlatformTypes))
|
||||
http.HandleFunc(webhook(client.PlatformVersionCompanies))
|
||||
http.HandleFunc(webhook(client.PlatformVersionReleaseDates))
|
||||
http.HandleFunc(webhook(client.PlatformVersions))
|
||||
http.HandleFunc(webhook(client.PlatformWebsites))
|
||||
http.HandleFunc(webhook(client.PlayerPerspectives))
|
||||
http.HandleFunc(webhook(client.PopularityTypes))
|
||||
http.HandleFunc(webhook(client.Regions))
|
||||
http.HandleFunc(webhook(client.ReleaseDateRegions))
|
||||
http.HandleFunc(webhook(client.ReleaseDates))
|
||||
http.HandleFunc(webhook(client.ReleaseDateStatuses))
|
||||
http.HandleFunc(webhook(client.Screenshots))
|
||||
http.HandleFunc(webhook(client.Themes))
|
||||
http.HandleFunc(webhook(client.Websites))
|
||||
http.HandleFunc(webhook(client.WebsiteTypes))
|
||||
|
||||
enabledEndpoint := endpoint.AllEndpoints
|
||||
enabledEndpoint = slices.DeleteFunc(enabledEndpoint, func(e endpoint.EndpointName) bool {
|
||||
return e == endpoint.EPWebhooks || e == endpoint.EPSearch || e == endpoint.EPPopularityPrimitives
|
||||
})
|
||||
|
||||
ip := net.ParseIP(baseUrl.Hostname())
|
||||
if baseUrl.Hostname() == "localhost" || (ip != nil && ip.IsLoopback()) {
|
||||
log.Printf("extral url is localhost. webhook will not be registered")
|
||||
} else {
|
||||
for _, endp := range enabledEndpoint {
|
||||
Url := baseUrl.JoinPath(fmt.Sprintf("/webhook/%s", string(endp)))
|
||||
log.Printf("registering webhook \"%s\" to \"%s\"", endp, Url.String())
|
||||
err = client.Webhooks.Register(endp, config.C().WebhookSecret, Url.String())
|
||||
if err != nil {
|
||||
log.Fatalf("failed to register webhook \"%s\": %v", endp, err)
|
||||
}
|
||||
log.Printf("webhook \"%s\" registered", endp)
|
||||
}
|
||||
if err != nil {
|
||||
log.Fatalf("failed to active webhook \"%s\": %v", endpoint.EPGames, err)
|
||||
}
|
||||
}
|
||||
|
||||
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(200)
|
||||
if _, err := w.Write([]byte("Hello World!")); err != nil {
|
||||
log.Printf("failed to write response: %v", err)
|
||||
}
|
||||
})
|
||||
|
||||
err = http.ListenAndServe(config.C().Address, nil)
|
||||
if err != nil {
|
||||
log.Fatalf("failed to start webhook server: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func webhook[T any](
|
||||
e endpoint.EntityEndpoint[T],
|
||||
) (string, func(w http.ResponseWriter, r *http.Request)) {
|
||||
return fmt.Sprintf("/webhook/%s", e.GetEndpointName()), func(w http.ResponseWriter, r *http.Request) {
|
||||
secret := r.Header.Get("X-Secret")
|
||||
if secret != config.C().WebhookSecret {
|
||||
w.WriteHeader(401)
|
||||
return
|
||||
}
|
||||
data := struct {
|
||||
ID uint64 `json:"id"`
|
||||
}{}
|
||||
jsonBytes, err := io.ReadAll(r.Body)
|
||||
if err != nil {
|
||||
log.Printf("failed to read request body: %v", err)
|
||||
w.WriteHeader(500)
|
||||
return
|
||||
}
|
||||
err = json.Unmarshal(jsonBytes, &data)
|
||||
if err != nil {
|
||||
log.Printf("failed to unmarshal request body: %v", err)
|
||||
w.WriteHeader(500)
|
||||
return
|
||||
}
|
||||
if data.ID == 0 {
|
||||
w.WriteHeader(400)
|
||||
return
|
||||
}
|
||||
item, err := e.GetByID(data.ID)
|
||||
if err != nil {
|
||||
log.Printf("failed to get %s: %v", e.GetEndpointName(), err)
|
||||
w.WriteHeader(500)
|
||||
return
|
||||
}
|
||||
oldItem, err := db.GetItemByIGDBID[T](e.GetEndpointName(), data.ID)
|
||||
if err != nil && err != mongo.ErrNoDocuments {
|
||||
log.Printf("failed to get %s: %v", e.GetEndpointName(), err)
|
||||
w.WriteHeader(500)
|
||||
return
|
||||
}
|
||||
newItem := model.NewItem(item)
|
||||
if oldItem != nil {
|
||||
newItem.MId = oldItem.MId
|
||||
}
|
||||
err = db.SaveItem(e.GetEndpointName(), newItem)
|
||||
if err != nil {
|
||||
log.Printf("failed to save %s: %v", e.GetEndpointName(), err)
|
||||
w.WriteHeader(500)
|
||||
return
|
||||
}
|
||||
log.Printf("%s %d saved", e.GetEndpointName(), data.ID)
|
||||
w.WriteHeader(200)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user