Query hasNewNotes and forward

This commit is contained in:
ekzyis 2023-06-01 01:58:07 +02:00
parent 64799bfa10
commit aaa89408d1
2 changed files with 90 additions and 0 deletions

28
main.go
View File

@ -13,7 +13,35 @@ func WaitUntilNextHour() {
time.Sleep(dur) time.Sleep(dur)
} }
func WaitUntilNextMinute() {
now := time.Now()
dur := now.Truncate(time.Minute).Add(time.Minute).Sub(now)
log.Println("sleeping for", dur.Round(time.Second))
time.Sleep(dur)
}
func CheckNotifications() {
var prevHasNewNotes bool
for {
hasNewNotes, err := FetchHasNewNotes()
if err != nil {
SendErrorToDiscord(err)
} else {
if !prevHasNewNotes && hasNewNotes {
// only send embed on "rising edge"
SendNotificationsEmbedToDiscord()
log.Println("Forwarded to monitoring")
} else if hasNewNotes {
log.Println("Already forwarded")
}
}
prevHasNewNotes = hasNewNotes
WaitUntilNextMinute()
}
}
func main() { func main() {
go CheckNotifications()
for { for {
stories, err := FetchHackerNewsTopStories() stories, err := FetchHackerNewsTopStories()

62
sn.go
View File

@ -95,6 +95,13 @@ type ItemsResponse struct {
} `json:"data"` } `json:"data"`
} }
type HasNewNotesResponse struct {
Errors []GraphQLError `json:"errors"`
Data struct {
HasNewNotes bool `json:"hasNewNotes"`
} `json:"data"`
}
var ( var (
StackerNewsUrl = "https://stacker.news" StackerNewsUrl = "https://stacker.news"
SnApiUrl = "https://stacker.news/api/graphql" SnApiUrl = "https://stacker.news/api/graphql"
@ -318,3 +325,58 @@ func SendStackerNewsEmbedToDiscord(title string, id int) {
} }
SendEmbedToDiscord(&embed) SendEmbedToDiscord(&embed)
} }
func SendNotificationsEmbedToDiscord() {
Timestamp := time.Now().Format(time.RFC3339)
color := 0xffc107
embed := discordgo.MessageEmbed{
Title: "new notifications",
URL: "https://stacker.news/hn/posts",
Color: color,
Footer: &discordgo.MessageEmbedFooter{
Text: "Stacker News",
IconURL: "https://stacker.news/favicon-notify.png",
},
Timestamp: Timestamp,
}
SendEmbedToDiscord(&embed)
}
func FetchHasNewNotes() (bool, error) {
log.Println("Checking notifications ...")
body := GraphQLPayload{
Query: `
{
hasNewNotes
}`,
}
resp, err := MakeStackerNewsRequest(body)
if err != nil {
return false, err
}
defer resp.Body.Close()
var hasNewNotesResp HasNewNotesResponse
err = json.NewDecoder(resp.Body).Decode(&hasNewNotesResp)
if err != nil {
err = fmt.Errorf("error decoding SN hasNewNotes: %w", err)
return false, err
}
err = CheckForErrors(hasNewNotesResp.Errors)
if err != nil {
return false, err
}
hasNewNotes := hasNewNotesResp.Data.HasNewNotes
msg := "Checking notifications ... OK - "
if hasNewNotes {
msg += "NEW"
} else {
msg += "NONE"
}
log.Println(msg)
return hasNewNotes, nil
}