73 lines
1.2 KiB
Go
73 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
)
|
|
|
|
func getComic(i int) []byte {
|
|
url := fmt.Sprintf("https://xkcd.com/%d/info.0.json", i)
|
|
resp, err := http.Get(url)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "Can't reach: %s", err)
|
|
os.Exit(-1)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
|
|
fmt.Fprintf(os.Stderr, "Skipping %d: got %d\n", i, resp.StatusCode)
|
|
return nil
|
|
}
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "invalid body %s\n", err)
|
|
}
|
|
return body
|
|
}
|
|
|
|
func main() {
|
|
var (
|
|
output io.WriteCloser = os.Stdout
|
|
err error
|
|
cnt int
|
|
fails int
|
|
data []byte
|
|
)
|
|
|
|
if len(os.Args) > 1 {
|
|
output, err = os.Create(os.Args[1])
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(-1)
|
|
}
|
|
|
|
defer output.Close()
|
|
|
|
}
|
|
fmt.Fprint(output, "[")
|
|
defer fmt.Fprint(output, "]")
|
|
for i := 1; fails < 2; i++ {
|
|
if data = getComic(i); data == nil {
|
|
fails++
|
|
continue
|
|
}
|
|
if cnt > 0 {
|
|
fmt.Fprint(output, ",")
|
|
}
|
|
|
|
_, err := io.Copy(output, bytes.NewBuffer(data))
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "STOPPED: %s\n", err)
|
|
os.Exit(-1)
|
|
}
|
|
fails = 0
|
|
cnt++
|
|
}
|
|
fmt.Fprintf(os.Stderr, "Read %d comics\n", cnt)
|
|
}
|