-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathcats.go
119 lines (100 loc) · 2.5 KB
/
cats.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
package main
import (
"errors"
"fmt"
"io/ioutil"
"log"
"log/syslog"
"math/rand"
"net"
"net/http"
"regexp"
"strconv"
"sync"
"time"
_ "net/http/pprof"
)
var imgsRe *regexp.Regexp
var mu sync.Mutex
var nextUpdate time.Time
var imgs [][]byte
func init() {
imgsRe = regexp.MustCompile(`(http://i.imgur.com/[^"]{5,20}|https://i.redditmedia.com/[^"]+\?s=[^"]+)|https://i.redd.it/[0-9a-z.]{10,30}`)
}
func maybeUpdateImgs() ([][]byte, error) {
mu.Lock()
defer mu.Unlock()
if time.Now().After(nextUpdate) {
req, err := http.NewRequest("GET", "http://www.reddit.com/r/catpics/hot.json", nil)
if err != nil {
return nil, err
}
req.Header.Add("User-Agent", "cats.go 0.1 (contact /u/eigma or [email protected])")
resp, err := (&http.Client{}).Do(req)
if err != nil {
return nil, err
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return nil, err
}
newImgs := imgsRe.FindAll(body, -1)
if len(newImgs) == 0 {
log.Printf("No cat pictures found: %s: %s", resp, body)
return nil, errors.New("No cat pictures found")
}
newImgs = append(newImgs, []byte("http://i.imgur.com/iHzbXfL.jpg"))
imgs = newImgs
nextUpdate = time.Now().Add(2 * time.Second)
}
return imgs, nil
}
func serveUrl(rw http.ResponseWriter, r *http.Request) {
rw.Header().Set("Content-Type", "text/plain")
imgs, e := maybeUpdateImgs()
if e != nil {
log.Print(e)
http.Error(rw, e.Error(), 503)
return
}
img := imgs[rand.Intn(len(imgs))]
rw.Header().Set("Content-Length", strconv.Itoa(len(img)))
rw.Write([]byte(img))
}
func serveRoot(rw http.ResponseWriter, r *http.Request) {
rw.Header().Set("Content-Type", "text/html")
imgs, e := maybeUpdateImgs()
if e != nil {
log.Print(e)
http.Error(rw, e.Error(), 503)
return
}
img := imgs[rand.Intn(len(imgs))]
html := fmt.Sprintf(
"<title>um, cats</title>"+
"<body background=http://104.131.51.57/static/bg.jpg>"+
"<div align=center style=\"width: 100%%; height: 100%%\">"+
"<img style=\"max-width: 90%%; max-height: 80%%\" src=\"%s\"/>"+
"</div>", img)
rw.Header().Set("Content-Length", strconv.Itoa(len(html)))
rw.Write([]byte(html))
}
func main() {
logger, e := syslog.New(syslog.LOG_NOTICE | syslog.LOG_DAEMON, "cats")
if e != nil {
log.Fatal(e)
}
log.SetOutput(logger)
http.HandleFunc("/url", serveUrl)
http.HandleFunc("/", serveRoot)
l, e := net.Listen("tcp", "127.0.0.1:8000")
if e != nil {
log.Fatal(e)
}
log.Print("serving")
e = http.Serve(l, nil)
if e != nil {
log.Fatal(e)
}
}