server.go 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  1. package main
  2. import (
  3. "flag"
  4. "html/template"
  5. "log"
  6. "net/http"
  7. "time"
  8. )
  9. var host = flag.String("h", "localhost", "host")
  10. var port = flag.String("p", "80", "port")
  11. var url = flag.String("b", "", "API URL")
  12. var app Application
  13. var errorPage = template.Must(template.ParseFiles("./html/error.html"))
  14. // ServeIndex serves index page.
  15. func serveIndex(w http.ResponseWriter, r *http.Request) {
  16. t := template.Must(template.ParseFiles("./html/index.html"))
  17. t.Execute(w, app.Languages)
  18. }
  19. // DisplaySearchResults displays search results as HTML page.
  20. func displaySearchResults(w http.ResponseWriter, r *http.Request) {
  21. var info SearchInfo
  22. info.GetInfoFromRequest(r)
  23. if info.Phrase != "" {
  24. var searchResults SearchResults
  25. if info.LanguageCode == "" || app.CheckLanguage(info.LanguageCode) {
  26. searchResults = app.Search(app.GetTMs(info.LanguageCode), info.Phrase)
  27. info.ResultsServed = searchResults.TotalResults
  28. WriteLog(info)
  29. } else {
  30. errorPage.Execute(w, "Language not valid!")
  31. return
  32. }
  33. if len(searchResults.Results) > 0 {
  34. funcs := template.FuncMap{"add": add}
  35. t := template.Must(template.New("results.html").Funcs(funcs).ParseFiles("./html/results.html"))
  36. t.Execute(w, searchResults)
  37. } else {
  38. errorPage.Execute(w, "Nothing found!")
  39. }
  40. } else {
  41. errorPage.Execute(w, "You need to enter search phrase!")
  42. }
  43. }
  44. // DisplayTMs displays TMs as HTML page.
  45. func displayTMs(w http.ResponseWriter, r *http.Request) {
  46. var info SearchInfo
  47. info.GetInfoFromRequest(r)
  48. var TMList []TM
  49. if info.LanguageCode == "" || app.CheckLanguage(info.LanguageCode) {
  50. TMList = app.GetTMs(info.LanguageCode)
  51. info.ResultsServed = len(TMList)
  52. WriteLog(info)
  53. } else {
  54. errorPage.Execute(w, "Language not valid!")
  55. return
  56. }
  57. if len(TMList) > 0 {
  58. t := template.Must(template.New("tms.html").ParseFiles("./html/tms.html"))
  59. t.Execute(w, TMList)
  60. } else {
  61. errorPage.Execute(w, "No TMs to display!")
  62. }
  63. }
  64. func main() {
  65. flag.Parse()
  66. app.SetBaseURL(*url)
  67. if app.BaseURL == "" {
  68. log.Fatalln("Can't do anything without URL to API")
  69. }
  70. app.Login()
  71. app.LoadLanguages()
  72. app.Delay = time.Duration(20 * time.Second)
  73. hostname := *host + ":" + *port
  74. http.HandleFunc("/", serveIndex)
  75. http.HandleFunc("/q", displaySearchResults)
  76. http.HandleFunc("/tms", displayTMs)
  77. log.Fatal(http.ListenAndServe(hostname, nil))
  78. }