package main

import (
   "github.com/gorilla/mux"
   "encoding/json"
   "net/http"
   "log"
   "structs"
   "database/sql"
   _ "github.com/go-sql-driver/mysql"
   "io/ioutil"
   "strconv"
   //"fmt"
   //"os"
   //"strings"
   //
   //"io"
)

var categs, recipes = structs.InitAll()
var conf = structs.GetConfig()
var db, err = sql.Open("mysql", conf["login"] + ":" + conf["password"] + "@/" + conf["dbname"])

func main() {

   if err != nil {
     panic(err)
   }
   defer db.Close()

   r := mux.NewRouter()

   r.HandleFunc("/", MainHandler)
   r.HandleFunc("/categories", CategoriesGetHandler)
   r.HandleFunc("/category/{id}", CategRecipeGetHandler)
   r.HandleFunc("/recipe/{id}", RecipeGetHandler)
   r.HandleFunc("/recipes", RecipesGetHandler)
   r.HandleFunc("/recipe/{id}/postcomment", CommentPostHandler)

   log.Println("Listen port " + conf["port"])
   http.ListenAndServe(":" + conf["port"], r)
}

func CommentPostHandler(iWrt http.ResponseWriter, iReq *http.Request) {
    var resp structs.PostForm
    vars := mux.Vars(iReq)

    id, err := strconv.Atoi(vars["id"])

    if err != nil {
      panic(err)
    }

    body, err := ioutil.ReadAll(iReq.Body)

    if err != nil {
      panic(err)
    } else {
      err = json.Unmarshal(body, &resp)
      if err != nil {
        panic(err)
      }
    }

    _, err = db.Exec("insert into serverDB.comments (id, text) values (?, ?)", id, resp.Text)

    if err != nil {
      panic(err)
    } else {
      iWrt.WriteHeader(http.StatusCreated)
    }
}

func RecipesGetHandler(iWrt http.ResponseWriter, iReq *http.Request) {
  var result []structs.Frecipe

  rows, error := db.Query("select * from " + conf["dbname"] + ".recipe")
  if error != nil {
    panic(error)
  }

  for rows.Next() {
    rec := structs.Frecipe{}
    err := rows.Scan(&rec.Recipe_id, &rec.User_id, &rec.Recipe_title, &rec.Time_of_cooking, &rec.Recipe,
                     &rec.Recipe_image, &rec.Recipe_type, &rec.Calories, &rec.Proteins, &rec.Carbohydrates,
                     &rec.Portions, &rec.Fats)
    if err != nil {
      panic(err)
      continue
    }
    result = append(result, rec)

  }

  if len(result) == 0 {
    iWrt.WriteHeader(http.StatusNotFound)
  } else {
    outJson, err := json.Marshal(result)

    if err != nil {
      panic(err)
    }

    iWrt.Write(outJson)
  }
}

func MainHandler(iWrt http.ResponseWriter, iReq *http.Request) {
   iWrt.WriteHeader(http.StatusNotFound)
}

func RecipeGetHandler(iWrt http.ResponseWriter, iReq *http.Request){
 vars := mux.Vars(iReq)

 recipe := vars["id"]

 row, err := db.Query("select * from serverDB.recipe where id = " + recipe)

 if err != nil {
   panic(err)
 }

 rec := structs.Frecipe{}

 err = row.Scan(&rec.Recipe_id, &rec.User_id, &rec.Recipe_title, &rec.Time_of_cooking, &rec.Recipe,
                  &rec.Recipe_image, &rec.Recipe_type, &rec.Calories, &rec.Proteins, &rec.Carbohydrates,
                  &rec.Portions, &rec.Fats)

 if err != nil || rec.Recipe == "" {
   iWrt.WriteHeader(http.StatusNotFound)
 } else {

   outJson, _ := json.Marshal(rec)
   iWrt.Write(outJson)
 }
}

func CategoriesGetHandler(iWrt http.ResponseWriter, iReq *http.Request){
 var result []structs.Categ

 rows, err := db.Query("select * from serverDB.recipe_types")

 if err != nil {
   panic(err)
 }

 for rows.Next() {
   var categ = structs.Categ{}

   err = rows.Scan(&categ.Recipe_type, &categ.Type_name, &categ.Type_image)

   if err != nil {
     panic(err)
   }

   result = append(result, categ)
 }

 if len(result) != 0 {
  outJson, _ := json.Marshal(result)
  iWrt.Write(outJson)
 } else {
   iWrt.WriteHeader(http.StatusNotFound)
 }
}

func CategRecipeGetHandler(iWrt http.ResponseWriter, iReq *http.Request){
 vars := mux.Vars(iReq)
 category := vars["id"]

 var result []structs.Frecipe

 rows, err := db.Query("select * from serverDB.recipe")

 if err != nil {
   panic(err)
 }

 for rows.Next(){
   rec := structs.Frecipe{}

   err = rows.Scan(&rec.Recipe_id, &rec.User_id, &rec.Recipe_title, &rec.Time_of_cooking, &rec.Recipe,
                    &rec.Recipe_image, &rec.Recipe_type, &rec.Calories, &rec.Proteins, &rec.Carbohydrates,
                    &rec.Portions, &rec.Fats)

   if err != nil {
     panic(err)
   }

   if structs.IntToString(rec.Recipe_id) == category {
     result = append(result, rec)
   }
 }
 if len(result) == 0 {
   iWrt.WriteHeader(http.StatusNotFound)
 } else {
   outJson, err := json.Marshal(result)

   if err != nil {
     panic(err)
   }

   iWrt.Write(outJson)
 }
}