package main

import (
  "text/template"
  "fmt"
  "net/http"
  "encoding/json"
  "strings"
  "log"
  "os"
  "path/filepath"
  "gitler.moe/suwako/goliblocale"
)

type (
  Page struct {
    Tit, Err, Url, Dom, Lan, Ver, Ves string
    i18n map[string]string
  }
  Api struct {
    Cod int `json:"code"`
    Err string `json:"error"`
    Url string `json:"url"`
    Mot string `json:"origin"`
    New bool `json:"isnew"`
  }
  Stat struct {
    Url string `json:"url"`
    Ver string `json:"version"`
  }
)

func (p Page) T(key string) string {
  return p.i18n[key]
}

func initloc(r *http.Request) string {
  supportLang := map[string]bool{
    "ja": true,
    "en": true,
  }

  cookie, err := r.Cookie("lang")
  if err != nil {
    return "ja"
  }

  if _, ok := supportLang[cookie.Value]; ok {
    return cookie.Value
  } else {
    return "ja"
  }
}

func serv(cnf Config, port int) {
  dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
  if err != nil {
    log.Fatal(err)
  }
  err = os.Chdir(dir)
  if err != nil {
    log.Fatal(err)
  }

  http.Handle(
    "/static/",
    http.StripPrefix("/static/", http.FileServer(http.Dir(cnf.webpath + "/static"))),
  )

  ftmpl := []string{
    cnf.webpath + "/view/index.html",
    cnf.webpath + "/view/header.html",
    cnf.webpath + "/view/footer.html",
  }

  http.HandleFunc("/api", func(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Content-Type", "application/json; charset=UTF-8")
    w.WriteHeader(200)
    buf, _ := json.MarshalIndent(&Stat{Url: cnf.domain, Ver: version}, "", "  ")
    _, _ = w.Write(buf)
  })

  http.HandleFunc("/api/lolify", func(w http.ResponseWriter, r *http.Request) {
    lang := initloc(r)
    i18n, err := goliblocale.GetLocale(cnf.webpath + "/locale/" + lang)
    if err != nil {
      fmt.Printf("liblocaleエラー：%v", err)
      return
    }

    w.Header().Set("Content-Type", "application/json; charset=UTF-8")
    w.WriteHeader(200)
    res := &Api{Cod: 500, Err: "未対応"}
    if r.Method == "POST" {
      err := r.ParseForm()
      if err != nil {
        fmt.Println(err)
        res.Err = "失敗"
        return
      }

      if r.PostForm.Get("url") != "" {
        addurl := r.PostForm.Get("url")
        chkprx := checkprefix(addurl)
        chklim := checkcharlim(addurl)
        if !chkprx {
          res = &Api{Cod: 400, Err: i18n["errfusei"]}
          buf, _ := json.MarshalIndent(res, "", "  ")
          _, _ = w.Write(buf)
          return
        }
        if !chklim {
          res = &Api{Cod: 400, Err: i18n["errcharlim"]}
          buf, _ := json.MarshalIndent(res, "", "  ")
          _, _ = w.Write(buf)
          return
        }

        if !chklim || !chkprx {
          res = &Api{Cod: 400, Err: i18n["errurlent"]}
          buf, _ := json.MarshalIndent(res, "", "  ")
          _, _ = w.Write(buf)
          return
        }

        chkfn, key := geturl(addurl, cnf.linkpath, true)
        if chkfn != "" {
          res = &Api{
            Cod: 200,
            Url: cnf.domain + "/" + key,
            Mot: addurl,
            New: false,
          }
        } else {
          res = &Api{
            Cod: 200,
            Url: cnf.domain + "/" + insertjson(addurl, cnf.linkpath),
            Mot: addurl,
            New: true,
          }
        }
      }
    }

    buf, _ := json.MarshalIndent(res, "", "  ")
    _, _ = w.Write(buf)
  })

  http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
    data := &Page{Ver: version, Ves: strings.ReplaceAll(version, ".", "")}
    uri := r.URL.Path
    lang := initloc(r)

    i18n, err := goliblocale.GetLocale(cnf.webpath + "/locale/" + lang)
    if err != nil {
      fmt.Printf("liblocaleエラー：%v", err)
      return
    }
    data.i18n = i18n
    data.Lan = lang

    // デフォルトページ＝未検出
    data.Tit = i18n["mikensyutu"]
    data.Err = i18n["errurlnai"]
    ftmpl[0] = cnf.webpath + "/view/404.html"
    tmpl := template.Must(template.ParseFiles(ftmpl[0], ftmpl[1], ftmpl[2]))

    if r.Method == "POST" {
      err := r.ParseForm()
      if err != nil { fmt.Println(err) }
      if r.PostForm.Get("sosin") != "" {
        if r.PostForm.Get("newadd") == "" {
          data.Err = i18n["errurlent"]
          return
        }

        addurl := r.PostForm.Get("newadd")
        chkprx := checkprefix(addurl)
        chklim := checkcharlim(addurl)

        if !chkprx {
          data.Tit = i18n["fuseiurl"]
          data.Err = i18n["errfusei"]
          return
        }

        if !chklim {
          data.Tit = i18n["fuseiurl"]
          data.Err = i18n["errcharlim"]
          return
        }

        chkfn, _ := geturl(addurl, cnf.linkpath, true)
        if chkfn != "" {
          http.Redirect(w, r, addurl, http.StatusSeeOther)
          return
        }

        data.Url = insertjson(addurl, cnf.linkpath)
        data.Dom = cnf.domain
        data.Tit = i18n["tansyukuzumi"]
        ftmpl[0] = cnf.webpath + "/view/submitted.html"
      } else if r.PostForm.Get("langchange") != "" {
        lang := r.PostForm.Get("lang")
        http.SetCookie(
          w,
          &http.Cookie{Name: "lang", Value: lang, MaxAge: 31536000, Path: "/"},
        )
        http.Redirect(w, r, "/", http.StatusSeeOther)
        return
      }
    } else { // r.Method == "GET"
      if uri == "/" {
        data.Tit = i18n["top"]
        ftmpl[0] = cnf.webpath + "/view/index.html"
      } else {
        red, _ := geturl(uri[1:], cnf.linkpath, false)
        if red != "" {
          http.Redirect(w, r, red, http.StatusSeeOther)
          return
        }
      }
    } // r.Method

    tmpl = template.Must(template.ParseFiles(ftmpl[0], ftmpl[1], ftmpl[2]))
    tmpl.Execute(w, data)
    data = nil
  })

  fmt.Println(fmt.Sprint(
    "http://" + cnf.ip + ":",
    port,
    " でサーバーを実行中。終了するには、CTRL+Cを押して下さい。",
  ))
  http.ListenAndServe(fmt.Sprint(cnf.ip + ":", port), nil)
}