package main

import (
    "./models"
    "log"
    "github.com/gorilla/websocket"
    //"fmt"
    "net/http"
    "html/template"
    "github.com/gorilla/mux"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize: 1024,
    WriteBufferSize: 1024,
}

var templates *template.Template

func setupRoutes() {
    r := mux.NewRouter()
    templates = template.Must(template.ParseGlob("pages/*.html"))
    r.HandleFunc("/", models.IndexHandler).Methods("GET")
    r.HandleFunc("/", models.IndexHandler).Methods("POST")
    r.HandleFunc("/upgrade-click/", models.CUpgradeHandler).Methods("POST")
    r.HandleFunc("/add/", models.BalanceHandler).Methods("POST")
    r.HandleFunc("/auto-inc-one/", models.AutoIncOneHandler).Methods("POST")
    r.HandleFunc("/ws", wsEndpoint)
    //http.HandleFunc("/ws", wsEndpoint)
    r.PathPrefix("/static/").Handler(http.StripPrefix("/static", http.FileServer(http.Dir("./static/"))))
    http.Handle("/", r)
    http.ListenAndServe(":8080", nil)
}

func main() {
    go models.AutoIncOne()
    setupRoutes()
}

func reader(conn *websocket.Conn) {
    for {
        messageType, p, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            return
        }

        log.Println(string(p))

        if err := conn.WriteMessage(messageType, p); err != nil {
            log.Println(err)
            return
        }
    }
}

func wsEndpoint(w http.ResponseWriter, r *http.Request) {
    upgrader.CheckOrigin = func(r *http.Request) bool { return true }

    ws, err := upgrader.Upgrade(w, r, nil)
    if err != nil{
        log.Println(err)
    }

    log.Println("Client Successfully Connected...")

    reader(ws)
}