main.go 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155
  1. package main
  2. import (
  3. "log"
  4. "os"
  5. "time"
  6. "fmt"
  7. "net/http"
  8. "html/template"
  9. "github.com/gorilla/mux"
  10. "github.com/gorilla/sessions"
  11. )
  12. type Page struct {
  13. Title string
  14. Body []byte
  15. }
  16. var (
  17. logger *log.Logger
  18. dBConnector connection
  19. sessionsStore = sessions.NewCookieStore([]byte("mysecretcookie"))
  20. indexTemplate = template.Must(template.ParseFiles("www\\index.html"))
  21. logInTemplate = template.Must(template.ParseFiles("www\\login.html"))
  22. signInTemplate = template.Must(template.ParseFiles("www\\signin.html"))
  23. )
  24. func indexHandler(w http.ResponseWriter, r *http.Request) {
  25. data := struct {
  26. User string
  27. Title string
  28. Items []string
  29. }{
  30. User: "",
  31. Title: "My page",
  32. Items: []string{
  33. "My photos",
  34. "My blog",
  35. "More",
  36. },
  37. }
  38. session, _ := sessionsStore.Get(r, "session")
  39. var (
  40. password string
  41. ok3 bool
  42. )
  43. untyped, ok := session.Values["username"]
  44. if ok {
  45. username, ok1 := untyped.(string)
  46. if ok1 {
  47. data.User = username
  48. untyped, ok2 := session.Values["password"]
  49. if ok2 {
  50. password, ok3 = untyped.(string)
  51. }
  52. }
  53. }
  54. indexTemplate.Execute(w, data)
  55. if ok3 {
  56. w.Write([]byte(password))
  57. }
  58. }
  59. func indexPostHandler(w http.ResponseWriter, r *http.Request) {
  60. fmt.Print("POST")
  61. }
  62. func logInGetHandler(w http.ResponseWriter, r *http.Request) {
  63. logInTemplate.Execute(w, nil)
  64. }
  65. func logInPostHandler(w http.ResponseWriter, r *http.Request) {
  66. r.ParseForm()
  67. username := r.PostForm.Get("username")
  68. password := r.PostForm.Get("password")
  69. if (dBConnector.LogIn(username, password)){
  70. session, _ := sessionsStore.Get(r, "session")
  71. session.Values["username"] = username
  72. session.Values["password"] = password
  73. session.Save(r,w)
  74. fmt.Printf("%s\n%s\n",username,password)
  75. http.Redirect(w, r, "/", http.StatusFound)
  76. } else {
  77. logInTemplate.Execute(w, nil)
  78. }
  79. }
  80. func signInHandler(w http.ResponseWriter, r *http.Request) {
  81. signInTemplate.Execute(w, nil)
  82. }
  83. func pressHandler(w http.ResponseWriter, r *http.Request) {
  84. logger.Print("click")
  85. http.Redirect(w, r, "/", http.StatusFound)
  86. }
  87. func createLogger() {
  88. startTime := time.Now()
  89. logFileName := "logs/go-site_log_" + startTime.Format("2006-01-02_15-04-05") + ".txt"
  90. file, err := os.OpenFile(logFileName, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0666)
  91. logger = log.New(file, "logger: ", log.Lshortfile)
  92. if err != nil {
  93. log.Fatal(err)
  94. }
  95. }
  96. type user struct{
  97. idusers int
  98. username string
  99. password string
  100. }
  101. func main() {
  102. createLogger()
  103. logger.Print("Hello, log file!")
  104. requestRouter := mux.NewRouter()
  105. dBConnector.Init("config.json")
  106. requestRouter.HandleFunc("/", indexHandler).Methods("GET")
  107. requestRouter.HandleFunc("/", indexPostHandler).Methods("POST")
  108. requestRouter.HandleFunc("/login/", logInGetHandler).Methods("GET")
  109. requestRouter.HandleFunc("/login/", logInPostHandler).Methods("POST")
  110. requestRouter.HandleFunc("/signin/", signInHandler).Methods("GET")
  111. //requestRouter.HandleFunc("/", pressHandler).Methods("POST")
  112. http.Handle("/", requestRouter)
  113. logger.Fatal(http.ListenAndServe(":8080", nil))
  114. }