package main import ( "bytes" "encoding/json" "fmt" "log" "net/http" "time" "github.com/dustin/go-humanize" "github.com/joho/godotenv" "github.com/namsral/flag" ) type GraphQLPayload struct { Query string `json:"query"` Variables map[string]interface{} `json:"variables,omitempty"` } type Dupe struct { Id int `json:"id,string"` Url string `json:"url"` Title string `json:"title"` } type DupesResponse struct { Data struct { Dupes []Dupe `json:"dupes"` } `json:"data"` } type Item struct { Id int `json:"id,string"` } type UpsertLinkResponse struct { Data struct { UpsertLink Item `json:"upsertLink"` } `json:"data"` } var ( SnApiUrl string SnApiToken string ) func init() { SnApiUrl = "https://stacker.news/api/graphql" err := godotenv.Load() if err != nil { log.Fatal("Error loading .env file") } flag.StringVar(&SnApiToken, "NEXT_AUTH_CSRF_TOKEN", "", "Token required for authorizing requests to stacker.news/api/graphql") flag.Parse() if SnApiToken == "" { log.Fatal("NEXT_AUTH_CSRF_TOKEN not set") } } func MakeStackerNewsRequest(body GraphQLPayload) *http.Response { bodyJSON, err := json.Marshal(body) if err != nil { log.Fatal("Error during json.Marshal:", err) } req, err := http.NewRequest("POST", SnApiUrl, bytes.NewBuffer(bodyJSON)) if err != nil { panic(err) } req.Header.Set("Content-Type", "application/json") req.Header.Set("Cookie", fmt.Sprintf("__Host-next-auth.csrf-token=%s", SnApiToken)) client := http.DefaultClient resp, err := client.Do(req) if err != nil { panic(err) } log.Printf("POST %s %d\n", SnApiUrl, resp.StatusCode) return resp } func CurateContentForStackerNews(stories *[]Story) *[]Story { // TODO: filter by relevance slice := (*stories)[0:1] return &slice } func FetchStackerNewsDupes(url string) *[]Dupe { body := GraphQLPayload{ Query: ` query Dupes($url: String!) { dupes(url: $url) { id url title } }`, Variables: map[string]interface{}{ "url": url, }, } resp := MakeStackerNewsRequest(body) defer resp.Body.Close() var dupesResp DupesResponse err := json.NewDecoder(resp.Body).Decode(&dupesResp) if err != nil { log.Fatal("Error decoding dupes JSON:", err) } return &dupesResp.Data.Dupes } func PostStoryToStackerNews(story *Story) { dupes := FetchStackerNewsDupes(story.Url) if len(*dupes) > 0 { return } body := GraphQLPayload{ Query: ` mutation upsertLink($url: String!, $title: String!) { upsertLink(url: $url, title: $title) { id } }`, Variables: map[string]interface{}{ "url": story.Url, "title": story.Title, }, } resp := MakeStackerNewsRequest(body) defer resp.Body.Close() var upsertLinkResp UpsertLinkResponse err := json.NewDecoder(resp.Body).Decode(&upsertLinkResp) if err != nil { log.Fatal("Error decoding dupes JSON:", err) } parentId := upsertLinkResp.Data.UpsertLink.Id comment := fmt.Sprintf( "This link was posted by [%s](%s) %s on [HN](%s). It received %d points and %d comments.", story.By, HackerNewsUserLink(story.By), humanize.Time(time.Unix(int64(story.Time), 0)), HackerNewsItemLink(story.ID), story.Score, story.Descendants, ) CommentStackerNewsPost(comment, parentId) } func CommentStackerNewsPost(text string, parentId int) { body := GraphQLPayload{ Query: ` mutation createComment($text: String!, $parentId: ID!) { createComment(text: $text, parentId: $parentId) { id } }`, Variables: map[string]interface{}{ "text": text, "parentId": parentId, }, } resp := MakeStackerNewsRequest(body) defer resp.Body.Close() }