Vroom

module/gitea/main.go

@@ -26,11 +26,11 @@ type Module struct {
 type giteaKeyType int
 
 const (
-	GITEA_KEY_OAUTH_CLIENT = giteaKeyType(iota)
+	KeyOauthClient = giteaKeyType(iota)
-	GITEA_KEY_USER
+	KeyUser
-	GITEA_KEY_STATE
+	KeyState
-	GITEA_KEY_ACCESS_TOKEN
+	KeyAccessToken
-	GITEA_KEY_REFRESH_TOKEN
+	KeyRefreshToken
 )
 
 func New() *Module {
@@ -56,7 +56,7 @@ func (m *Module) SetupModule(router *mux.Router, cb func(cb func(http.ResponseWr
 
 func (m *Module) getClient(cb func(http.ResponseWriter, *http.Request, *utils.State, *gitea.Client)) func(rw http.ResponseWriter, req *http.Request) {
 	return m.sessionWrapper(func(rw http.ResponseWriter, req *http.Request, state *utils.State) {
-		if v, ok := utils.GetStateValue[*gitea.Client](state, GITEA_KEY_OAUTH_CLIENT); ok {
+		if v, ok := utils.GetStateValue[*gitea.Client](state, KeyOauthClient); ok {
 			cb(rw, req, state, v)
 			return
 		}
@@ -67,22 +67,50 @@ func (m *Module) getClient(cb func(http.ResponseWriter, *http.Request, *utils.St
 func (m *Module) homepage(rw http.ResponseWriter, req *http.Request, state *utils.State, giteaClient *gitea.Client) {
 	myUser, _, err := giteaClient.GetMyUserInfo()
 	if err != nil {
-		state.Del(GITEA_KEY_OAUTH_CLIENT)
+		state.Del(KeyOauthClient)
 		http.Error(rw, err.Error(), http.StatusInternalServerError)
 		return
 	}
-	fmt.Println("Username:", myUser.UserName)
 	orgs, _, err := giteaClient.ListMyOrgs(gitea.ListOrgsOptions{})
 	if err != nil {
 		http.Error(rw, err.Error(), http.StatusInternalServerError)
 		return
 	}
-	fmt.Println("Orgs:", orgs)
-
 	orgSimple := make([]struct{ Name string }, len(orgs))
 	for i, j := range orgs {
 		orgSimple[i] = struct{ Name string }{j.UserName}
 	}
+	selOrg := ""
+	repoSimple := make([]struct{ Name string }, 0)
+	selRepo := ""
+	refSimple := make([]struct{ Name string }, 0)
+
+	q := req.URL.Query()
+	if q.Has("org") {
+		selOrg = q.Get("org")
+		repos, _, err := giteaClient.ListOrgRepos(selOrg, gitea.ListOrgReposOptions{ListOptions: gitea.ListOptions{Page: 0, PageSize: 100}})
+		if err != nil {
+			http.Error(rw, err.Error(), http.StatusInternalServerError)
+			return
+		}
+		repoSimple = make([]struct{ Name string }, len(repos))
+		for i, j := range repos {
+			repoSimple[i] = struct{ Name string }{Name: j.Name}
+		}
+
+		if q.Has("repo") {
+			selRepo = q.Get("repo")
+			refs, _, err := giteaClient.GetRepoRefs(selOrg, selRepo, "heads")
+			if err != nil {
+				http.Error(rw, err.Error(), http.StatusInternalServerError)
+				return
+			}
+			refSimple = make([]struct{ Name string }, len(refs))
+			for i, ref := range refs {
+				refSimple[i] = struct{ Name string }{Name: ref.Ref}
+			}
+		}
+	}
 
 	tmp, err := template.New("homepage").Parse(indexTemplate)
 	if err != nil {
@@ -91,12 +119,20 @@ func (m *Module) homepage(rw http.ResponseWriter, req *http.Request, state *util
 	}
 	err = tmp.Execute(rw, struct {
 		Username string
-		Orgs     []struct {
+		Orgs     []struct{ Name string }
-			Name string
+		Repos    []struct{ Name string }
-		}
+		SelOrg   string
+		ShowOrg  bool
+		SelRepo  string
+		ShowRepo bool
 	}{
 		Username: myUser.UserName,
 		Orgs:     orgSimple,
+		Repos:    repoSimple,
+		SelOrg:   selOrg,
+		SelRepo:  selRepo,
+		ShowOrg:  selOrg != "",
+		ShowRepo: selRepo != "",
 	})
 	if err != nil {
 		fmt.Println("Template execute error:", err)
@@ -105,14 +141,14 @@ func (m *Module) homepage(rw http.ResponseWriter, req *http.Request, state *util
 }
 
 func (m *Module) loginPage(rw http.ResponseWriter, req *http.Request, state *utils.State) {
-	if myUser, ok := utils.GetStateValue[*string](state, GITEA_KEY_USER); ok {
+	if myUser, ok := utils.GetStateValue[*string](state, KeyUser); ok {
 		if myUser != nil {
 			http.Redirect(rw, req, "/gitea", http.StatusTemporaryRedirect)
 			return
 		}
 	}
 
-	if flowState, ok := utils.GetStateValue[uuid.UUID](state, GITEA_KEY_STATE); ok {
+	if flowState, ok := utils.GetStateValue[uuid.UUID](state, KeyState); ok {
 		q := req.URL.Query()
 		if q.Has("code") && q.Has("state") {
 			if q.Get("state") == flowState.String() {
@@ -126,7 +162,9 @@ func (m *Module) loginPage(rw http.ResponseWriter, req *http.Request, state *uti
 					fmt.Println("Create client error:", err)
 					return
 				}
-				state.Put(GITEA_KEY_OAUTH_CLIENT, c)
+				state.Put(KeyOauthClient, c)
+				state.Put(KeyAccessToken, exchange.AccessToken)
+				state.Put(KeyRefreshToken, exchange.RefreshToken)
 				http.Redirect(rw, req, "/gitea", http.StatusTemporaryRedirect)
 				return
 			}
@@ -136,7 +174,7 @@ func (m *Module) loginPage(rw http.ResponseWriter, req *http.Request, state *uti
 	}
 
 	flowState := uuid.New()
-	state.Put(GITEA_KEY_STATE, flowState)
+	state.Put(KeyState, flowState)
 
 	http.Redirect(rw, req, m.oauthClient.AuthCodeURL(flowState.String(), oauth2.AccessTypeOffline), http.StatusTemporaryRedirect)
 }

module/gitea/pages/index.go.html

@@ -2,17 +2,48 @@
 <html>
 <head>
     <title>Gitea | Melon Tools</title>
+    <style>
+        .page-content {
+            display: flex;
+            flex-direction: row;
+            align-items: flex-start;
+        }
+    </style>
 </head>
 <body>
 <h1>Gitea | Melon Tools</h1>
-<div>
+<div class="page-content">
-    <p>Select repository source:</p>
+    <div>
-    <ul>
+        <p>Select repository source:</p>
-        <li>{{.Username}} (my user account)</li>
+        <ul>
-        {{range .Orgs}}
+            <li>
-            <li>{{.Name}}</li>
+                <a href="?org={{.Username}}">{{.Username}} (my user account)</a>
-        {{end}}
+            </li>
-    </ul>
+            {{range .Orgs}}
+                <li>
+                    <a href="?org={{.Name}}">{{.Name}}</a>
+                </li>
+            {{end}}
+        </ul>
+    </div>
+    {{if .ShowOrg}}
+        <div>
+            <p>Select repository:</p>
+            <ul>
+                {{range .Repos}}
+                <li>
+                    <a href="?org={{$.SelOrg}}&repo={{.Name}}">{{$.SelOrg}}/{{.Name}}</a>
+                </li>
+                {{end}}
+            </ul>
+        </div>
+    {{end}}
+    {{if .ShowRepo}}
+        <div>
+            <p>Repository details:</p>
+            <p>Go import: go get {{.SelRepo.Full}}</p>
+        </div>
+    {{end}}
 </div>
 </body>
 </html>