add: get/update user endpoint

bot/handler/token.go

@@ -0,0 +1,47 @@
+package handler
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"net/http"
+
+	"github.com/claustra01/calendeye/db"
+)
+
+type TokenResponseBody struct {
+	RefreshToken string `json:"refresh_token"`
+}
+
+func UpdateRefreshToken(w http.ResponseWriter, req *http.Request) {
+	defer func() { _ = req.Body.Close() }()
+
+	id := req.URL.Query().Get("id")
+	body, err := ioutil.ReadAll(req.Body)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte(err.Error()))
+		return
+	}
+
+	var token TokenResponseBody
+	err = json.Unmarshal(body, &token)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte(err.Error()))
+		return
+	}
+
+	err = db.UpdateRefreshToken(id, token.RefreshToken)
+	if err == db.ErrNoRecord {
+		w.WriteHeader(http.StatusNotFound)
+		w.Write([]byte(err.Error()))
+		return
+	} else if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte(err.Error()))
+		return
+	}
+
+	w.WriteHeader(http.StatusOK)
+	w.Write([]byte("Refresh token updated"))
+}

bot/handler/user.go

@@ -0,0 +1,40 @@
+package handler
+
+import (
+	"encoding/json"
+	"net/http"
+
+	"github.com/claustra01/calendeye/db"
+)
+
+func GetUser(w http.ResponseWriter, req *http.Request) {
+	defer func() { _ = req.Body.Close() }()
+
+	id := req.URL.Query().Get("id")
+	if id == "" {
+		w.WriteHeader(http.StatusBadRequest)
+		w.Write([]byte("id is required"))
+		return
+	}
+
+	user, err := db.GetUser(id)
+	if err == db.ErrNoRecord {
+		w.WriteHeader(http.StatusNotFound)
+		w.Write([]byte(err.Error()))
+		return
+	} else if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte(err.Error()))
+		return
+	}
+
+	jsonUser, err := json.Marshal(user)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte(err.Error()))
+		return
+	}
+
+	w.WriteHeader(http.StatusOK)
+	w.Write([]byte(jsonUser))
+}

bot/server.go

@@ -48,6 +48,10 @@ func main() {
 		handler.Callback(w, req, bot, channelSecret)
 	})
 
+	// Setup HTTP Server for get/update user information
+	http.HandleFunc("/user", handler.GetUser)
+	http.HandleFunc("/token", handler.UpdateRefreshToken)
+
 	// This is just sample code.
 	// For actual use, you must support HTTPS by using `ListenAndServeTLS`, a reverse proxy or something else.
 	port := os.Getenv("PORT")