tools/module/gitea/main.go

package gitea

import (
	"code.gitea.io/sdk/gitea"
	"code.mrmelon54.xyz/sean/melon-tools/utils"
	"context"
	_ "embed"
	"encoding/gob"
	"fmt"
	"github.com/google/uuid"
	"github.com/gorilla/mux"
	"golang.org/x/oauth2"
	"html/template"
	"net/http"
	"os"
)

//go:embed pages/index.go.html
var indexTemplate string

type Module struct {
	oauthClient    *oauth2.Config
	sessionWrapper func(cb func(http.ResponseWriter, *http.Request, *utils.State)) func(rw http.ResponseWriter, req *http.Request)
}

type giteaKeyType int

const (
	KeyOauthClient = giteaKeyType(iota)
	KeyUser
	KeyState
	KeyAccessToken
	KeyRefreshToken
)

func New() *Module {
	gob.Register(new(giteaKeyType))
	return &Module{}
}

func (m *Module) SetupModule(router *mux.Router, cb func(cb func(http.ResponseWriter, *http.Request, *utils.State)) func(rw http.ResponseWriter, req *http.Request)) {
	m.sessionWrapper = cb
	m.oauthClient = &oauth2.Config{
		ClientID:     os.Getenv("GITEA_CLIENT_ID"),
		ClientSecret: os.Getenv("GITEA_CLIENT_SECRET"),
		Scopes:       []string{"openid"},
		Endpoint: oauth2.Endpoint{
			AuthURL:  os.Getenv("GITEA_AUTHORIZE_URL"),
			TokenURL: os.Getenv("GITEA_TOKEN_URL"),
		},
		RedirectURL: os.Getenv("GITEA_REDIRECT_URL"),
	}
	router.HandleFunc("/", m.getClient(m.homepage))
	router.HandleFunc("/login", m.sessionWrapper(m.loginPage))
}

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, KeyOauthClient); ok {
			cb(rw, req, state, v)
			return
		}
		http.Redirect(rw, req, "/gitea/login", http.StatusTemporaryRedirect)
	})
}

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(KeyOauthClient)
		http.Error(rw, err.Error(), http.StatusInternalServerError)
		return
	}
	orgs, _, err := giteaClient.ListMyOrgs(gitea.ListOrgsOptions{})
	if err != nil {
		http.Error(rw, err.Error(), http.StatusInternalServerError)
		return
	}
	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 {
		fmt.Println("Template parse error:", err)
		return
	}
	err = tmp.Execute(rw, struct {
		Username string
		Orgs     []struct{ 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)
		return
	}
}

func (m *Module) loginPage(rw http.ResponseWriter, req *http.Request, state *utils.State) {
	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, KeyState); ok {
		q := req.URL.Query()
		if q.Has("code") && q.Has("state") {
			if q.Get("state") == flowState.String() {
				exchange, err := m.oauthClient.Exchange(context.Background(), q.Get("code"))
				if err != nil {
					fmt.Println("Exchange token error:", err)
					return
				}
				c, err := gitea.NewClient(os.Getenv("GITEA_SERVER"), gitea.SetToken(exchange.AccessToken))
				if err != nil {
					fmt.Println("Create client error:", err)
					return
				}
				state.Put(KeyOauthClient, c)
				state.Put(KeyAccessToken, exchange.AccessToken)
				state.Put(KeyRefreshToken, exchange.RefreshToken)
				http.Redirect(rw, req, "/gitea", http.StatusTemporaryRedirect)
				return
			}
			http.Error(rw, "OAuth flow state doesn't match\n", http.StatusBadRequest)
			return
		}
	}

	flowState := uuid.New()
	state.Put(KeyState, flowState)

	http.Redirect(rw, req, m.oauthClient.AuthCodeURL(flowState.String(), oauth2.AccessTypeOffline), http.StatusTemporaryRedirect)
}

func (m *Module) fetchRepos(giteaClient *gitea.Client) {
	repos, _, err := giteaClient.ListOrgRepos("snow", gitea.ListOrgReposOptions{ListOptions: gitea.ListOptions{Page: 0, PageSize: 100}})
	if err != nil {
		fmt.Println(err)
		return
	}
	for _, myRepo := range repos {
		refs, _, err := giteaClient.GetRepoRefs("snow", myRepo.Name, "heads")
		if err != nil {
			fmt.Println(err)
			return
		}
		fmt.Println(len(refs))
		for _, myRef := range refs {
			if myRef.Ref == "refs/heads/"+myRepo.DefaultBranch {
				fmt.Println(myRef.Ref)
				return
			}
		}
		fmt.Println("Can't find default branch")
	}
}