mirror of https://git.jolheiser.com/ugit.git
110 lines
2.6 KiB
Go
110 lines
2.6 KiB
Go
package http
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"io/fs"
|
|
"mime"
|
|
"net/http"
|
|
"path/filepath"
|
|
|
|
"go.jolheiser.com/ugit/internal/git"
|
|
"go.jolheiser.com/ugit/internal/html"
|
|
"go.jolheiser.com/ugit/internal/http/httperr"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-git/go-git/v5/plumbing/object"
|
|
)
|
|
|
|
func (rh repoHandler) repoTree(ref, path string) http.HandlerFunc {
|
|
return httperr.Handler(func(w http.ResponseWriter, r *http.Request) error {
|
|
repoName := chi.URLParam(r, "repo")
|
|
repo, err := git.NewRepo(rh.s.RepoDir, repoName)
|
|
if err != nil {
|
|
httpErr := http.StatusInternalServerError
|
|
if errors.Is(err, fs.ErrNotExist) {
|
|
httpErr = http.StatusNotFound
|
|
}
|
|
return httperr.Status(err, httpErr)
|
|
}
|
|
if repo.Meta.Private {
|
|
return httperr.Status(errors.New("could not get git repo"), http.StatusNotFound)
|
|
}
|
|
|
|
if ref == "" {
|
|
ref, err = repo.DefaultBranch()
|
|
if err != nil {
|
|
return httperr.Error(err)
|
|
}
|
|
}
|
|
|
|
tree, err := repo.Dir(ref, path)
|
|
if err != nil {
|
|
if errors.Is(err, object.ErrDirectoryNotFound) {
|
|
return rh.repoFile(w, r, repo, ref, path)
|
|
}
|
|
return httperr.Error(err)
|
|
}
|
|
|
|
readmeContent, err := html.Readme(repo, ref, path)
|
|
if err != nil {
|
|
return httperr.Error(err)
|
|
}
|
|
|
|
var back string
|
|
if path != "" {
|
|
back = filepath.Dir(path)
|
|
}
|
|
if err := html.RepoTree(html.RepoTreeContext{
|
|
Description: repo.Meta.Description,
|
|
BaseContext: rh.baseContext(),
|
|
RepoHeaderComponentContext: rh.repoHeaderContext(repo, r),
|
|
RepoTreeComponentContext: html.RepoTreeComponentContext{
|
|
Repo: repoName,
|
|
Ref: ref,
|
|
Tree: tree,
|
|
Back: back,
|
|
},
|
|
ReadmeComponentContext: html.ReadmeComponentContext{
|
|
Markdown: readmeContent,
|
|
},
|
|
}).Render(r.Context(), w); err != nil {
|
|
return httperr.Error(err)
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func (rh repoHandler) repoFile(w http.ResponseWriter, r *http.Request, repo *git.Repo, ref, path string) error {
|
|
content, err := repo.FileContent(ref, path)
|
|
if err != nil {
|
|
return httperr.Error(err)
|
|
}
|
|
|
|
if r.URL.Query().Has("raw") {
|
|
if r.URL.Query().Has("pretty") {
|
|
ext := filepath.Ext(path)
|
|
w.Header().Set("Content-Type", mime.TypeByExtension(ext))
|
|
}
|
|
w.Write([]byte(content))
|
|
return nil
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
if err := html.Code.Convert([]byte(content), filepath.Base(path), &buf); err != nil {
|
|
return httperr.Error(err)
|
|
}
|
|
|
|
if err := html.RepoFile(html.RepoFileContext{
|
|
BaseContext: rh.baseContext(),
|
|
RepoHeaderComponentContext: rh.repoHeaderContext(repo, r),
|
|
Code: buf.String(),
|
|
Path: path,
|
|
}).Render(r.Context(), w); err != nil {
|
|
return httperr.Error(err)
|
|
}
|
|
|
|
return nil
|
|
}
|