package main import ( "fmt" "io/ioutil" "math/rand" "net/http" "strings" "time" "github.com/layeh/gumble/gumble" ) var client_token = make(map[string]string) var token_client = make(map[string]string) var external_ip = "" type Page struct { Title string Body []byte } func (p *Page) save() error { filename := p.Title + ".txt" return ioutil.WriteFile(filename, p.Body, 0600) } func loadPage(title string) (*Page, error) { filename := title + ".txt" body, err := ioutil.ReadFile(filename) if err != nil { return nil, err } return &Page{Title: title, Body: body}, nil } func handler(w http.ResponseWriter, r *http.Request) { var uname = token_client[r.URL.Path[1:]] if uname == "" { fmt.Fprintf(w, "I don't know you") } else { fmt.Fprintf(w, "Hi there, I love %s!", uname) } } func Webserver() { http.HandleFunc("/", handler) http.ListenAndServe(":9563", nil) rand.Seed(time.Now().UnixNano()) } func GetWebAddress(user *gumble.User) { if client_token[user.Name] != "" { token_client[client_token[user.Name]] = "" } // dealing with collisions var firstLoop = true for !firstLoop && token_client[client_token[user.Name]] == "" { client_token[user.Name] = randSeq(10) firstLoop = false } token_client[client_token[user.Name]] = user.Name dj.SendPrivateMessage(user, fmt.Sprintf(WEB_ADDRESS, getIP(), client_token[user.Name], getIP(), client_token[user.Name])) } func getIP() string { if external_ip != "" { return external_ip } else { if response, err := http.Get("http://myexternalip.com/raw"); err == nil { defer response.Body.Close() if response.StatusCode == 200 { if body, err := ioutil.ReadAll(response.Body); err == nil { external_ip = strings.TrimSpace(string(body)) } } } return external_ip } } var letters = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ") func randSeq(n int) string { b := make([]rune, n) for i := range b { b[i] = letters[rand.Intn(len(letters))] } return string(b) }