You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
1.5 KiB
Go
75 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"net/http"
|
|
"fmt"
|
|
"log"
|
|
"html/template"
|
|
"github.com/kreativmonkey/shrt/lib"
|
|
)
|
|
|
|
type Page struct {
|
|
Title string
|
|
Body string
|
|
}
|
|
|
|
var (
|
|
short *shrt.Storage
|
|
)
|
|
|
|
func init() {
|
|
var err error
|
|
short, err = shrt.Open()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func index(w http.ResponseWriter, r *http.Request) {
|
|
fmt.Println("method: ", r.Method) //get request method
|
|
fmt.Println("Path: ", r.URL.Path[1:])
|
|
switch r.Method {
|
|
case "GET":
|
|
var redirect string
|
|
if ok := short.Get(r.URL.Path[1:], &redirect); ok{
|
|
P := Page{Title: ToString(r.URL.Path[1:]), Body: redirect}
|
|
fmt.Printf("Redirect: %s \n", redirect)
|
|
t, _ := template.ParseFiles("template/index.gohtml")
|
|
t.Execute(w, P)
|
|
fmt.Println("Url by Token:", P)
|
|
} else {
|
|
t, _ := template.ParseFiles("template/index.gohtml")
|
|
t.Execute(w, nil)
|
|
}
|
|
case "POST":
|
|
r.ParseForm()
|
|
var test string
|
|
token, err := short.Add(ToString(r.Form["url"]), &test)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
P := Page{Title: ToString(r.Form["url"]), Body: token}
|
|
t, _ := template.ParseFiles("template/index.gohtml")
|
|
t.Execute(w, P)
|
|
fmt.Println("Token by Url:", P)
|
|
}
|
|
|
|
}
|
|
|
|
func handelPost(w http.ResponseWriter, r *http.Request){
|
|
|
|
}
|
|
|
|
func main() {
|
|
http.HandleFunc("/", index) // setting router rule
|
|
err := http.ListenAndServe(":9090", nil) // setting listening port
|
|
if err != nil {
|
|
log.Fatal("ListenAndServe: ", err)
|
|
}
|
|
}
|
|
|
|
// ToString convert the input to a string.
|
|
func ToString(obj interface{}) string {
|
|
res := fmt.Sprintf("%s", obj)
|
|
return string(res)
|
|
} |