package gitea import ( "bytes" "fmt" "io" "io/fs" "net/http" "net/url" "strings" "sync" gclient "code.gitea.io/sdk/gitea" "github.com/spf13/viper" ) type Client struct { serverURL string token string giteapages string gc *gclient.Client } func NewClient(serverURL, token, giteapages string) (*Client, error) { if giteapages == "" { giteapages = "gitea-pages" } gc, err := gclient.NewClient(serverURL, gclient.SetToken(token), gclient.SetGiteaVersion("")) if err != nil { return nil, err } return &Client{ serverURL: serverURL, token: token, gc: gc, giteapages: giteapages, }, nil } func (c *Client) Open(name, ref string) (fs.File, error) { if ref == "" { ref = c.giteapages } owner, repo, filepath := splitName(name) // if repo is empty they want to have the gitea-pages repo if repo == "" { repo = c.giteapages filepath = "index.html" } // if filepath is empty they want to have the index.html if filepath == "" { filepath = "index.html" } // we need to check if the repo exists (and allows access) if !c.allowsPages(owner, repo) { // if we're checking the gitea-pages and it doesn't exist, return 404 if repo == c.giteapages && !c.hasRepoBranch(owner, repo, c.giteapages) { return nil, fs.ErrNotExist } // the repo didn't exist but maybe it's a filepath in the gitea-pages repo // so we need to check if the gitea-pages repo exists filepath = repo repo = c.giteapages if !c.allowsPages(owner, repo) || !c.hasRepoBranch(owner, repo, c.giteapages) { return nil, fs.ErrNotExist } } hasConfig := true if err := c.readConfig(owner, repo); err != nil { // we don't need a config for gitea-pages // no config is only exposing the gitea-pages branch if repo != c.giteapages { return nil, err } hasConfig = false } // if we don't have a config and the repo is the gitea-pages // always overwrite the ref to the gitea-pages branch if !hasConfig && (repo == c.giteapages || ref == c.giteapages) { ref = c.giteapages } else if !validRefs(ref) { return nil, fs.ErrNotExist } res, err := c.getRawFileOrLFS(owner, repo, filepath, ref) if err != nil { return nil, err } if strings.HasSuffix(filepath, ".md") { res, err = handleMD(res) if err != nil { return nil, err } } return &openFile{ content: res, name: filepath, }, nil } func (c *Client) getRawFileOrLFS(owner, repo, filepath, ref string) ([]byte, error) { var ( giteaURL string err error ) // TODO: make pr for go-sdk // gitea sdk doesn't support "media" type for lfs/non-lfs giteaURL, err = url.JoinPath(c.serverURL+"/api/v1/repos/", owner, repo, "media", filepath) if err != nil { return nil, err } giteaURL += "?ref=" + url.QueryEscape(ref) req, err := http.NewRequest(http.MethodGet, giteaURL, nil) if err != nil { return nil, err } req.Header.Add("Authorization", "token "+c.token) resp, err := http.DefaultClient.Do(req) if err != nil { return nil, err } switch resp.StatusCode { case http.StatusNotFound: return nil, fs.ErrNotExist case http.StatusOK: default: return nil, fmt.Errorf("unexpected status code '%d'", resp.StatusCode) } res, err := io.ReadAll(resp.Body) if err != nil { return nil, err } defer resp.Body.Close() return res, nil } var bufPool = sync.Pool{ New: func() any { return new(bytes.Buffer) }, } func handleMD(res []byte) ([]byte, error) { meta, resbody, err := extractFrontMatter(string(res)) if err != nil { return nil, err } resmd, err := markdown([]byte(resbody)) if err != nil { return nil, err } res = append([]byte("\n\n\n

"), []byte(meta["title"].(string))...) res = append(res, []byte("

")...) res = append(res, resmd...) res = append(res, []byte("")...) return res, nil } func (c *Client) repoTopics(owner, repo string) ([]string, error) { repos, _, err := c.gc.ListRepoTopics(owner, repo, gclient.ListRepoTopicsOptions{}) return repos, err } func (c *Client) hasRepoBranch(owner, repo, branch string) bool { b, _, err := c.gc.GetRepoBranch(owner, repo, branch) if err != nil { return false } return b.Name == branch } func (c *Client) allowsPages(owner, repo string) bool { topics, err := c.repoTopics(owner, repo) if err != nil { return false } for _, topic := range topics { if topic == c.giteapages { return true } } return false } func (c *Client) readConfig(owner, repo string) error { cfg, err := c.getRawFileOrLFS(owner, repo, c.giteapages+".toml", c.giteapages) if err != nil { return err } viper.SetConfigType("toml") return viper.ReadConfig(bytes.NewBuffer(cfg)) } func splitName(name string) (string, string, string) { parts := strings.Split(name, "/") // parts contains: ["owner", "repo", "filepath"] switch len(parts) { case 1: return parts[0], "", "" case 2: return parts[0], parts[1], "" default: return parts[0], parts[1], strings.Join(parts[2:], "/") } } func validRefs(ref string) bool { validrefs := viper.GetStringSlice("allowedrefs") for _, r := range validrefs { if r == ref { return true } if r == "*" { return true } } return false }