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
|
package main
import (
"crypto/sha1"
"encoding/hex"
"errors"
"io/ioutil"
"net/http"
"os"
"regexp"
"sort"
)
type Storage struct {
Path string
}
var (
tokenPattern = regexp.MustCompile(`[a-fA-F0-9]{10,}`)
)
func (s *Storage) MustInit() error {
os.Mkdir(s.Path, 0755)
os.Mkdir(s.Path+"/battles", 0755)
os.Mkdir(s.Path+"/tokens", 0700)
return nil
}
func (s *Storage) CanPost(token string) (bool, error) {
if !tokenPattern.MatchString(token) {
return false, errors.New("invalid token")
}
_, err := os.Stat(s.Path + "/tokens/" + token)
return err == nil, nil
}
func (s *Storage) AddBattle(data []byte) (string, error) {
hash := sha1.Sum(data)
id := hex.EncodeToString(hash[:])
return id, ioutil.WriteFile(s.Path+"/battles/"+id, data, 0644)
}
func (s *Storage) ServeBattle(id string, w http.ResponseWriter, r *http.Request) {
w.Header().Add("Content-Type", "application/json")
http.ServeFile(w, r, s.Path+"/battles/"+id)
}
func (s *Storage) ListRecentBattles(count int) ([]Battle, error) {
files, err := ioutil.ReadDir(s.Path + "/battles")
if err != nil {
return nil, err
}
sort.Slice(files, func(lhs, rhs int) bool {
return files[lhs].ModTime().After(files[rhs].ModTime())
})
battles := make([]Battle, 0, count)
for i := 0; i < count && i < len(files); i++ {
battles = append(battles, Battle{Id: files[i].Name(), Date: files[i].ModTime()})
}
return battles, nil
}
|