Initial commit

2022-10-30 22:46:54 +01:00
parent e6e5634d44
commit 4511c4c017
6 changed files with 1739 additions and 0 deletions
--- a/pkg/gitea/gitea.go
+++ b/pkg/gitea/gitea.go
@ -0,0 +1,238 @@
+package gitea
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"io/fs"
+	"net/http"
+	"net/url"
+	"strings"
+	"sync"
+
+	"github.com/spf13/viper"
+)
+
+type Client struct {
+	serverURL string
+	token     string
+}
+
+type pagesConfig struct {
+}
+
+type topicsResponse struct {
+	Topics []string `json:"topics"`
+}
+
+func NewClient(serverURL, token string) *Client {
+	return &Client{
+		serverURL: serverURL,
+		token:     token,
+	}
+}
+
+func (c *Client) Open(name, ref string) (fs.File, error) {
+	if ref == "" {
+		ref = "gitea-pages"
+	}
+
+	owner, repo, filepath, err := splitName(name)
+	if err != nil {
+		return nil, err
+	}
+
+	if !c.allowsPages(owner, repo) {
+		return nil, fs.ErrNotExist
+	}
+
+	if err := c.readConfig(owner, repo); err != nil {
+		return nil, err
+	}
+
+	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
+	)
+
+	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("<!DOCTYPE html>\n<html>\n<body>\n<h1>"), []byte(meta["title"].(string))...)
+	res = append(res, []byte("</h1>")...)
+	res = append(res, resmd...)
+	res = append(res, []byte("</body></html>")...)
+
+	return res, nil
+}
+
+func (c *Client) repoTopics(owner, repo string) ([]string, error) {
+	var (
+		giteaURL string
+		err      error
+	)
+
+	giteaURL, err = url.JoinPath(c.serverURL+"/api/v1/repos/", owner, repo, "topics")
+	if err != nil {
+		return nil, err
+	}
+
+	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
+	}
+
+	res, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	defer resp.Body.Close()
+
+	t := topicsResponse{}
+	json.Unmarshal(res, &t)
+
+	return t.Topics, nil
+}
+
+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 == "gitea-pages" {
+			return true
+		}
+	}
+
+	return false
+}
+
+func (c *Client) readConfig(owner, repo string) error {
+	cfg, err := c.getRawFileOrLFS(owner, repo, "gitea-pages.toml", "gitea-pages")
+	if err != nil && !errors.Is(err, fs.ErrNotExist) {
+		return err
+	}
+
+	viper.SetConfigType("toml")
+	viper.ReadConfig(bytes.NewBuffer(cfg))
+
+	return nil
+}
+
+func splitName(name string) (string, string, string, error) {
+	parts := strings.Split(name, "/")
+
+	// parts contains: ["owner", "repo", "filepath"]
+	// return invalid path if not enough parts
+	if len(parts) < 3 {
+		return "", "", "", fs.ErrInvalid
+	}
+
+	owner := parts[0]
+	repo := parts[1]
+	filepath := strings.Join(parts[2:], "/")
+
+	return owner, repo, filepath, nil
+}
+
+func validRefs(ref string) bool {
+	validrefs := viper.GetStringSlice("allowedrefs")
+	for _, r := range validrefs {
+		if r == ref {
+			return true
+		}
+
+		if r == "*" {
+			return true
+		}
+	}
+
+	return false
+}