mirror of
https://github.com/usememos/memos.git
synced 2025-06-05 22:09:59 +02:00
feat: support open api with webhooks
This commit is contained in:
129
api/auth.go
129
api/auth.go
@@ -1,14 +1,9 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"memos/api/e"
|
||||
"memos/config"
|
||||
"memos/store"
|
||||
"memos/utils"
|
||||
"net/http"
|
||||
|
||||
"github.com/gorilla/mux"
|
||||
@@ -38,7 +33,7 @@ func handleUserSignUp(w http.ResponseWriter, r *http.Request) {
|
||||
return
|
||||
}
|
||||
|
||||
user, err := store.CreateNewUser(userSignup.Username, userSignup.Password, "")
|
||||
user, err := store.CreateNewUser(userSignup.Username, userSignup.Password)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "DATABASE_ERROR", err.Error())
|
||||
@@ -107,127 +102,6 @@ func handleUserSignOut(w http.ResponseWriter, r *http.Request) {
|
||||
})
|
||||
}
|
||||
|
||||
func handleGithubAuthCallback(w http.ResponseWriter, r *http.Request) {
|
||||
code := r.URL.Query().Get("code")
|
||||
|
||||
requestBody := map[string]string{
|
||||
"client_id": config.GITHUB_CLIENTID,
|
||||
"client_secret": config.GITHUB_SECRET,
|
||||
"code": code,
|
||||
}
|
||||
|
||||
requestJSON, _ := json.Marshal(requestBody)
|
||||
|
||||
// POST request to get access_token
|
||||
req, err := http.NewRequest(
|
||||
"POST",
|
||||
"https://github.com/login/oauth/access_token",
|
||||
bytes.NewBuffer(requestJSON),
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "REQUEST_BODY_ERROR", "Error in request github api")
|
||||
return
|
||||
}
|
||||
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
req.Header.Set("Accept", "application/json")
|
||||
|
||||
resp, err := http.DefaultClient.Do(req)
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "REQUEST_BODY_ERROR", "Error in request github api")
|
||||
return
|
||||
}
|
||||
|
||||
// Response body converted to stringified JSON
|
||||
respBody, _ := ioutil.ReadAll(resp.Body)
|
||||
|
||||
// Represents the response received from Github
|
||||
type GithubAccessTokenResponse struct {
|
||||
AccessToken string `json:"access_token"`
|
||||
TokenType string `json:"token_type"`
|
||||
Scope string `json:"scope"`
|
||||
}
|
||||
|
||||
ghResp := GithubAccessTokenResponse{}
|
||||
json.Unmarshal(respBody, &ghResp)
|
||||
|
||||
githubAccessToken := ghResp.AccessToken
|
||||
|
||||
// Get request to a set URL
|
||||
req, err = http.NewRequest(
|
||||
"GET",
|
||||
"https://api.github.com/user",
|
||||
nil,
|
||||
)
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "REQUEST_BODY_ERROR", "Error in request github api")
|
||||
return
|
||||
}
|
||||
|
||||
authorizationHeaderValue := fmt.Sprintf("token %s", githubAccessToken)
|
||||
req.Header.Set("Authorization", authorizationHeaderValue)
|
||||
|
||||
resp, err = http.DefaultClient.Do(req)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "REQUEST_BODY_ERROR", "Error in request github api")
|
||||
return
|
||||
}
|
||||
|
||||
respBody, _ = ioutil.ReadAll(resp.Body)
|
||||
|
||||
githubData := string(respBody)
|
||||
|
||||
type GithubUser struct {
|
||||
Login string `json:"login"`
|
||||
Name string `json:"name"`
|
||||
}
|
||||
|
||||
githubUser := GithubUser{}
|
||||
json.Unmarshal([]byte(githubData), &githubUser)
|
||||
|
||||
session, _ := SessionStore.Get(r, "session")
|
||||
userId := fmt.Sprintf("%v", session.Values["user_id"])
|
||||
|
||||
if userId != "" {
|
||||
githubNameUsable, err := store.CheckGithubNameUsable(githubUser.Login)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "DATABASE_ERROR", "Error in CheckGithubNameUsable")
|
||||
return
|
||||
}
|
||||
|
||||
if !githubNameUsable {
|
||||
e.ErrorHandler(w, "DATABASE_ERROR", "Error in CheckGithubNameUsable")
|
||||
return
|
||||
}
|
||||
|
||||
userPatch := store.UserPatch{
|
||||
GithubName: &githubUser.Login,
|
||||
}
|
||||
|
||||
store.UpdateUser(userId, &userPatch)
|
||||
}
|
||||
|
||||
user, err := store.GetUserByGithubName(githubUser.Login)
|
||||
|
||||
if err != nil {
|
||||
username := githubUser.Name
|
||||
usernameUsable, _ := store.CheckUsernameUsable(username)
|
||||
for !usernameUsable {
|
||||
username = githubUser.Name + utils.GenUUID()
|
||||
usernameUsable, _ = store.CheckUsernameUsable(username)
|
||||
}
|
||||
user, _ = store.CreateNewUser(username, username, githubUser.Login)
|
||||
}
|
||||
|
||||
session.Values["user_id"] = user.Id
|
||||
session.Save(r, w)
|
||||
|
||||
http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
|
||||
}
|
||||
|
||||
func RegisterAuthRoutes(r *mux.Router) {
|
||||
authRouter := r.PathPrefix("/api/auth").Subrouter()
|
||||
|
||||
@@ -236,5 +110,4 @@ func RegisterAuthRoutes(r *mux.Router) {
|
||||
authRouter.HandleFunc("/signup", handleUserSignUp).Methods("POST")
|
||||
authRouter.HandleFunc("/signin", handleUserSignIn).Methods("POST")
|
||||
authRouter.HandleFunc("/signout", handleUserSignOut).Methods("POST")
|
||||
authRouter.HandleFunc("/github", handleGithubAuthCallback).Methods("GET")
|
||||
}
|
||||
|
28
api/user.go
28
api/user.go
@@ -29,16 +29,16 @@ func handleGetMyUserInfo(w http.ResponseWriter, r *http.Request) {
|
||||
func handleUpdateMyUserInfo(w http.ResponseWriter, r *http.Request) {
|
||||
userId, _ := GetUserIdInSession(r)
|
||||
|
||||
userPatch := store.UserPatch{}
|
||||
err := json.NewDecoder(r.Body).Decode(&userPatch)
|
||||
updateUserPatch := store.UpdateUserPatch{}
|
||||
err := json.NewDecoder(r.Body).Decode(&updateUserPatch)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "REQUEST_BODY_ERROR", "Bad request")
|
||||
return
|
||||
}
|
||||
|
||||
if userPatch.Username != nil {
|
||||
usernameUsable, _ := store.CheckUsernameUsable(*userPatch.Username)
|
||||
if updateUserPatch.Username != nil {
|
||||
usernameUsable, _ := store.CheckUsernameUsable(*updateUserPatch.Username)
|
||||
if !usernameUsable {
|
||||
json.NewEncoder(w).Encode(Response{
|
||||
Succeed: false,
|
||||
@@ -49,19 +49,7 @@ func handleUpdateMyUserInfo(w http.ResponseWriter, r *http.Request) {
|
||||
}
|
||||
}
|
||||
|
||||
if userPatch.GithubName != nil {
|
||||
githubNameUsable, _ := store.CheckGithubNameUsable(*userPatch.GithubName)
|
||||
if !githubNameUsable {
|
||||
json.NewEncoder(w).Encode(Response{
|
||||
Succeed: false,
|
||||
Message: "GitHub name is existed",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
user, err := store.UpdateUser(userId, &userPatch)
|
||||
user, err := store.UpdateUser(userId, &updateUserPatch)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "DATABASE_ERROR", err.Error())
|
||||
@@ -75,10 +63,10 @@ func handleUpdateMyUserInfo(w http.ResponseWriter, r *http.Request) {
|
||||
})
|
||||
}
|
||||
|
||||
func handleRefreshUserOpenId(w http.ResponseWriter, r *http.Request) {
|
||||
func handleResetUserOpenId(w http.ResponseWriter, r *http.Request) {
|
||||
userId, _ := GetUserIdInSession(r)
|
||||
|
||||
openId, err := store.UpdateUserOpenId(userId)
|
||||
openId, err := store.ResetUserOpenId(userId)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "DATABASE_ERROR", err.Error())
|
||||
@@ -155,7 +143,7 @@ func RegisterUserRoutes(r *mux.Router) {
|
||||
|
||||
userRouter.HandleFunc("/me", handleGetMyUserInfo).Methods("GET")
|
||||
userRouter.HandleFunc("/me", handleUpdateMyUserInfo).Methods("PATCH")
|
||||
userRouter.HandleFunc("/open_id/new", handleRefreshUserOpenId).Methods("POST")
|
||||
userRouter.HandleFunc("/open_id/new", handleResetUserOpenId).Methods("POST")
|
||||
userRouter.HandleFunc("/checkusername", handleCheckUsername).Methods("POST")
|
||||
userRouter.HandleFunc("/validpassword", handleValidPassword).Methods("POST")
|
||||
}
|
||||
|
55
api/webhooks.go
Normal file
55
api/webhooks.go
Normal file
@@ -0,0 +1,55 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"memos/api/e"
|
||||
"memos/store"
|
||||
"net/http"
|
||||
|
||||
"github.com/gorilla/mux"
|
||||
)
|
||||
|
||||
func handleCreateMemoByWH(w http.ResponseWriter, r *http.Request) {
|
||||
vars := mux.Vars(r)
|
||||
openId := vars["openId"]
|
||||
|
||||
type CreateMemoDataBody struct {
|
||||
Content string `json:"content"`
|
||||
}
|
||||
|
||||
createMemo := CreateMemoDataBody{}
|
||||
err := json.NewDecoder(r.Body).Decode(&createMemo)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "REQUEST_BODY_ERROR", "Bad request")
|
||||
return
|
||||
}
|
||||
|
||||
user, err := store.GetUserByOpenId(openId)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "DATABASE_ERROR", err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
memo, err := store.CreateNewMemo(createMemo.Content, user.Id)
|
||||
|
||||
if err != nil {
|
||||
e.ErrorHandler(w, "DATABASE_ERROR", err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
json.NewEncoder(w).Encode(Response{
|
||||
Succeed: true,
|
||||
Message: "",
|
||||
Data: memo,
|
||||
})
|
||||
}
|
||||
|
||||
func RegisterWebHooksRoutes(r *mux.Router) {
|
||||
memoRouter := r.PathPrefix("/api/whs").Subrouter()
|
||||
|
||||
memoRouter.Use(JSONResponseMiddleWare)
|
||||
|
||||
memoRouter.HandleFunc("/memo/{openId}", handleCreateMemoByWH).Methods("POST")
|
||||
}
|
Reference in New Issue
Block a user