Something went wrong. Try again.
Monorepo for Tangled
Something went wrong. Try again.
4.7 kB · 165 lines
Go
at ci
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166package xrpc
import ( "context" "crypto/sha256" "fmt" "io" "net/http" "path/filepath" "slices" "strings"
"github.com/bluesky-social/indigo/atproto/atclient" "github.com/bluesky-social/indigo/atproto/syntax" "github.com/go-git/go-git/v5/plumbing/object" "tangled.org/core/knotmirror/xrpc/gitea")
func (x *Xrpc) GetBlob(w http.ResponseWriter, r *http.Request) { var ( repoQuery = r.URL.Query().Get("repo") ref = r.URL.Query().Get("ref") // ref can be empty (git.Open handles this) path = r.URL.Query().Get("path") )
repo, err := syntax.ParseDID(repoQuery) if err != nil { writeJson(w, http.StatusBadRequest, atclient.ErrorBody{Name: "BadRequest", Message: fmt.Sprintf("repo parameter invalid: %s", repoQuery)}) return }
l := x.logger.With("method", "git.getBlob", "repo", repo, "ref", ref, "path", path) l.Debug("request")
if path == "" { writeJson(w, http.StatusBadRequest, atclient.ErrorBody{Name: "BadRequest", Message: "missing path parameter"}) return }
size, reader, err := x.getFile(r.Context(), repo, ref, path) if err != nil { l.Warn("local mirror failed, trying proxy", "err", err) if x.proxyToKnot(w, r, repo) { return } writeJson(w, http.StatusInternalServerError, atclient.ErrorBody{Name: "InternalServerError", Message: "failed to get blob"}) return } defer reader.Close()
// default to octet-stream for large blobs if size > 1000*1000 { // 1MB w.Header().Set("Content-Type", "application/octet-stream") if _, err := io.Copy(w, reader); err != nil { l.Error("failed to serve the blob", "err", err) } return }
contents, err := io.ReadAll(reader) if err != nil { l.Error("failed to read blob content", "err", err) writeJson(w, http.StatusInternalServerError, atclient.ErrorBody{Name: "InternalServerError", Message: "failed to read the blob"}) return }
mimeType := http.DetectContentType(contents) // override MIME types for formats that http.DetectContentType does not recognize switch filepath.Ext(path) { case ".svg": mimeType = "image/svg+xml" case ".avif": mimeType = "image/avif" case ".jxl": mimeType = "image/jxl" case ".heic", ".heif": mimeType = "image/heif" }
switch { case strings.HasPrefix(mimeType, "image/"), strings.HasPrefix(mimeType, "video/"): eTag := fmt.Sprintf("\"%x\"", sha256.Sum256(contents)) if clientETag := r.Header.Get("If-None-Match"); clientETag == eTag { w.WriteHeader(http.StatusNotModified) return } w.Header().Set("ETag", eTag) w.Header().Set("Content-Type", mimeType)
case strings.HasPrefix(mimeType, "text/") || isTextualMimeType(mimeType): w.Header().Set("Cache-Control", "public, no-cache") // serve all text content as text/plain w.Header().Set("Content-Type", "text/plain; charset=utf-8")
default: l.Error("attempted to serve disallowed file type", "mimetype", mimeType) writeJson(w, http.StatusInternalServerError, atclient.ErrorBody{Name: "InvalidRequest", Message: "only image, video, and text files can be accessed directly"}) return } w.Write(contents)}
func (x *Xrpc) getFile(ctx context.Context, repo syntax.DID, ref, path string) (int64, io.ReadCloser, error) { repoPath, err := x.makeRepoPath(ctx, repo) if err != nil { return 0, nil, fmt.Errorf("resolving repo did: %w", err) }
rev := ref if rev == "" { rev = "HEAD" }
head, err := gitea.GetCommit(ctx, repoPath, rev) if err != nil { return 0, nil, fmt.Errorf("get head commit: %w", err) }
treePath := filepath.Dir(path) name := filepath.Base(path)
// find subTree subRev := head.Hash.String() + "^{tree}" if treePath != "." { subRev = head.Hash.String() + ":" + treePath } subTree, err := gitea.GetTree(ctx, repoPath, subRev) if err != nil { return 0, nil, fmt.Errorf("get subtree %s: %w", subRev, err) }
// find entry entry, err := func(subTree *object.Tree) (*object.TreeEntry, error) { for _, entry := range subTree.Entries { if entry.Name == name { return &entry, nil } } return nil, fmt.Errorf("object doesn't exist") }(subTree) if err != nil { return 0, nil, fmt.Errorf("get file: %w", err) }
x.logger.Debug("ReadBlob", "name", entry.Name, "mode", entry.Mode.String(), "hash", entry.Hash.String())
// find blob return gitea.ReadBlob(ctx, repoPath, entry.Hash)}
var textualMimeTypes = []string{ "application/json", "application/xml", "application/yaml", "application/x-yaml", "application/toml", "application/javascript", "application/ecmascript",}
// isTextualMimeType returns true if the MIME type represents textual content// that should be served as text/plain for security reasonsfunc isTextualMimeType(mimeType string) bool { return slices.Contains(textualMimeTypes, mimeType)}