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 giteapagesAllowAll string gc *gclient.Client } func NewClient(serverURL, token, giteapages, giteapagesAllowAll string) (*Client, error) { if giteapages == "" { giteapages = "gitea-pages" } if giteapagesAllowAll == "" { giteapagesAllowAll = "gitea-pages-allowall" } 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, giteapagesAllowAll: giteapagesAllowAll, }, nil } func (c *Client) Open(name, ref string) (fs.File, error) { owner, repo, filepath := splitName(name) if repo == "" { repo = owner + ".fluffy.pw" filepath = "index.html" } if filepath == "" { filepath = "index.html" } isFluffyPagesRepo := strings.HasSuffix(repo, ".fluffy.pw") limited, allowall := c.allowsPages(owner, repo) if !limited && !allowall && !isFluffyPagesRepo { return nil, fs.ErrNotExist } hasConfig := true if err := c.readConfig(owner, repo); err != nil { if !isFluffyPagesRepo && !allowall { return nil, err } hasConfig = false } if !hasConfig && !validRefs(ref, allowall || isFluffyPagesRepo) { 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 } title, _ := meta["title"].(string) res = append([]byte("\n\n\n

"), []byte(title)...) 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, bool) { if strings.HasSuffix(repo, ".fluffy.pw") { return true, true } topics, err := c.repoTopics(owner, repo) if err != nil { return false, false } for _, topic := range topics { if topic == c.giteapagesAllowAll { return true, true } } for _, topic := range topics { if topic == c.giteapages { return true, false } } return false, 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, allowall bool) bool { if allowall { return true } validrefs := viper.GetStringSlice("allowedrefs") for _, r := range validrefs { if r == ref { return true } if r == "*" { return true } } return false }