-
Notifications
You must be signed in to change notification settings - Fork 4
/
storinator.go
105 lines (86 loc) · 1.99 KB
/
storinator.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
package main
import (
"encoding/gob"
"bytes"
"encoding/base64"
"io/ioutil"
"os"
"strings"
)
var tokens = make(map[string]struct{})
func storeTokens() {
buffer := bytes.Buffer{}
encoder := gob.NewEncoder(&buffer)
err := encoder.Encode(tokens)
if err != nil {
panic("Error encoding gob!")
}
ioutil.WriteFile(tokensFile, buffer.Bytes(), 0644)
}
func loadTokens() {
fileBytes, err := ioutil.ReadFile(tokensFile)
if err != nil {
panic("Error reading tokens file!")
}
buffer := bytes.Buffer{}
buffer.Write(fileBytes)
decoder := gob.NewDecoder(&buffer)
err = decoder.Decode(&tokens)
if err != nil {
panic("Error decoding tokens file!")
}
}
func generateToken() string {
token := generateRandomString(64)
tokens[token] = struct{}{}
storeTokens()
return token
}
func getDataLoc(token string, key string) (string, string) {
dataFileDir := dataDir + token + "/"
dataFile := dataFileDir + base64.URLEncoding.EncodeToString([]byte(key)) + suffix
return dataFileDir, dataFile
}
func storeData(token string, key string, data []byte) {
dataFileDir, dataFile := getDataLoc(token, key)
os.MkdirAll(dataFileDir, os.ModePerm)
ioutil.WriteFile(dataFile, data, 0644)
}
func getData(token string, key string) []byte {
_, dataFile := getDataLoc(token, key)
fileBytes, err := ioutil.ReadFile(dataFile)
if err != nil {
return nil
}
return fileBytes
}
func deleteData(token string, key string) {
_, dataFile := getDataLoc(token, key)
os.Remove(dataFile)
}
func getKeys(token string) []string {
dataFileDir, _ := getDataLoc(token, "")
f, err := os.Open(dataFileDir)
if err != nil {
return nil
}
files, err := f.Readdir(-1)
f.Close()
if err != nil {
return nil
}
keys := make([]string, len(files))
for i, file := range files {
name := file.Name()
if !strings.HasSuffix(name, suffix) {
continue
}
name = strings.TrimSuffix(name, suffix)
decoded, err := base64.URLEncoding.DecodeString(name)
if err != nil {
continue
}
keys[i] = string(decoded)
}
return keys
}