Merge pull request goplus#21 from wu-jj/feat/mvp-20240119

refactor login service

cmd/gopcomm/community_yap.gox

@@ -27,7 +27,7 @@ logger, _ := zap.NewProduction()
 defer logger.Sync()
 zlog := logger.Sugar()
 
-static "/"
+// static "/"
 
 get "/success", ctx => {
 	ctx.yap "2xx", {}
@@ -355,10 +355,59 @@ post "/upload", ctx => {
 	ctx.JSON(200,id)
 }
 
+get "/login", ctx => {
+	// Get URL from query string
+	redirectURL := ctx.URL.Query().Get("redirect_url")
+	loginURL := community.RedirectToCasdoor(redirectURL)
+	ctx.Redirect loginURL, http.StatusFound
+}
+
+
+get "/callback", ctx => {
+	code :=ctx.URL.Query().Get("code")
+	state :=ctx.URL.Query().Get("state")
+
+	token, error := community.GetAccessToken(code, state)
+	if error != nil {
+		zlog.Error("err",error) // Redirect to login
+	}
+
+	cookie := http.Cookie{
+		Name: "token",
+		Value: token.AccessToken,
+		Path: "/",
+		MaxAge: 3600,
+	}
+	http.SetCookie(ctx.ResponseWriter, &cookie)
+
+	// Redirect to home page
+	// TODO: Get redirect URL from state
+	http.Redirect(ctx.ResponseWriter, ctx.Request, fmt.Sprintf("http://localhost:8080?token=%s", token.AccessToken), http.StatusFound)
+}
+
 conf := &core.Config{}
 community, _ = core.New(todo, conf)
 trans = translation.New(os.Getenv("NIUTRANS_API_KEY"), "", "")
 core.CasdoorConfigInit()
 
+// 404
+handle "/",ctx => {
+	ctx.yap "4xx", {}
+}
+
 zlog.Info "Started in endpoint: ", endpoint
-run endpoint
+// run endpoint
+
+// 500
+run(endpoint, func(h http.Handler) http.Handler {
+
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		defer func() {
+			if err := recover(); err != nil {
+				http.Redirect(w, r, "/failed", http.StatusFound)
+			}
+		}()
+
+		h.ServeHTTP(w, r)
+	})
+})