/
bapteme.go
121 lines (100 loc) · 2.32 KB
/
bapteme.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
120
121
package main
import (
"encoding/base64"
"flag"
"fmt"
"net/http"
"strconv"
"strings"
"github.com/dchest/uniuri"
"github.com/mssola/user_agent"
"github.com/op/go-logging"
)
func prefix(r *http.Request) string {
ua := new(user_agent.UserAgent)
ua.Parse(r.UserAgent())
os := ua.OS()
var ret string
if strings.Contains(os, "Linux") {
ret = "lin"
} else if strings.Contains(os, "Windows") {
ret = "win"
} else {
ret = "srv"
}
return ret
}
func randomName(length int) string {
return uniuri.NewLen(length)
}
func hashName(id string) string {
// h := md5.New()
// h.Write([]byte(id))
// return base64.URLEncoding.EncodeToString(h.Sum(nil))
return base64.URLEncoding.EncodeToString([]byte(id))
}
func handler(w http.ResponseWriter, r *http.Request) {
id := r.FormValue("id")
var err error
var size int
formSize := r.FormValue("size")
if formSize != "" {
size, err = strconv.Atoi(formSize)
if err != nil {
fmt.Println(err)
return
}
} else {
size = *flagSize
}
var name string
pre := r.FormValue("prefix")
if len(pre) != 0 {
name = pre
} else {
name = prefix(r)
}
log.Debug("Prefix = %s", name)
instance := r.FormValue("instance")
name = strings.Join([]string{name, instance}, "")
if len(name) >= size {
http.Error(w, "name too long", 500)
return
}
var suffix string
if len(id) != 0 {
ts := hashName(id)
if len(ts) >= size-len(name) {
suffix = ts[0 : size-len(name)]
} else {
suffix = ts
}
} else {
suffix = randomName(size - len(name))
}
log.Debug("Suffix = %s", suffix)
name = strings.Join([]string{name, suffix}, "")
fmt.Fprintf(w, "%s", name)
}
var (
flagBind = flag.String("bind", "", "Address to bind. Format IP:PORT")
flagSize = flag.Int("size", 10, "Default final hostname size")
flagDebug = flag.Bool("d", false, "turn on debug info")
)
var log = logging.MustGetLogger("bapteme")
func main() {
flag.Parse()
var format = logging.MustStringFormatter("%{level} %{message}")
logging.SetFormatter(format)
if *flagDebug {
logging.SetLevel(logging.DEBUG, "bapteme")
} else {
logging.SetLevel(logging.INFO, "bapteme")
}
log.Info("Bind to %s", *flagBind)
http.HandleFunc("/", handler)
// http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
// handler(w, r, *size)
// })
http.ListenAndServe(*flagBind, nil)
}