add flags for custom port, kamoji file and template file
This commit is contained in:
@@ -2,6 +2,7 @@ package main
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
|
"flag"
|
||||||
"html/template"
|
"html/template"
|
||||||
"log"
|
"log"
|
||||||
"math/rand"
|
"math/rand"
|
||||||
@@ -18,10 +19,10 @@ type Kamojis struct {
|
|||||||
Kamojis []Kamoji
|
Kamojis []Kamoji
|
||||||
}
|
}
|
||||||
|
|
||||||
func loadKamojis() Kamojis {
|
func loadKamojis(path string) Kamojis {
|
||||||
kamojis := Kamojis{}
|
kamojis := Kamojis{}
|
||||||
|
|
||||||
file, err := os.Open("kamojis.txt")
|
file, err := os.Open(path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -39,11 +40,16 @@ func loadKamojis() Kamojis {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
tmpl, err := template.ParseFiles("kamoji_template.html")
|
port := flag.String("port", "80", "http listening port")
|
||||||
|
kamojisPath := flag.String("kamojis", "kamojis.txt", "path to file with kamojis")
|
||||||
|
templatePath := flag.String("template", "kamoji_template.html", "path to HTML template file")
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
tmpl, err := template.ParseFiles(*templatePath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
allk := loadKamojis()
|
allk := loadKamojis(*kamojisPath)
|
||||||
timestamp := time.Now().Unix()
|
timestamp := time.Now().Unix()
|
||||||
randomNumber := rand.Intn(len(allk.Kamojis))
|
randomNumber := rand.Intn(len(allk.Kamojis))
|
||||||
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||||||
@@ -54,5 +60,5 @@ func main() {
|
|||||||
k := allk.Kamojis[randomNumber]
|
k := allk.Kamojis[randomNumber]
|
||||||
tmpl.Execute(w, k)
|
tmpl.Execute(w, k)
|
||||||
})
|
})
|
||||||
http.ListenAndServe(":80", nil)
|
http.ListenAndServe(":"+*port, nil)
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user