67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package lib
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
func Run(bind string, f *Forgejo) (*http.Server, error) {
|
|
if f == nil {
|
|
return nil, errors.New("forgejo is nil")
|
|
}
|
|
s := &http.Server{
|
|
Addr: bind,
|
|
Handler: http.HandlerFunc(f.handle),
|
|
}
|
|
return s, nil
|
|
}
|
|
|
|
func trySet(h http.Header, key string, src http.Header) {
|
|
value := src.Get(key)
|
|
if value == "" {
|
|
h.Del(key)
|
|
return
|
|
}
|
|
h.Set(key, value)
|
|
}
|
|
|
|
func (f *Forgejo) handle(w http.ResponseWriter, r *http.Request) {
|
|
arr := strings.Split(r.URL.Path[1:], "/")
|
|
if len(arr) < 2 {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
return
|
|
}
|
|
user, repo := arr[0], arr[1]
|
|
if user == "u" {
|
|
user, repo = repo, f.Branch
|
|
}
|
|
file := "index.html"
|
|
if l := len(arr); l > 2 {
|
|
if arr[l-1] == "" {
|
|
arr[l-1] = "index.html"
|
|
}
|
|
file = strings.Join(arr[2:], "/")
|
|
}
|
|
|
|
headers := map[string]string{}
|
|
headers["If-None-Match"] = r.Header.Get("If-None-Match")
|
|
headers["If-Modified-Since"] = r.Header.Get("If-Modified-Since")
|
|
headers["If-Range"] = r.Header.Get("If-Range")
|
|
headers["Range"] = r.Header.Get("Range")
|
|
resp, err := f.GetFile(r.Context(), headers, user, repo, f.Branch, file)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
trySet(w.Header(), "Etag", resp.Header)
|
|
trySet(w.Header(), "Last-Modified", resp.Header)
|
|
trySet(w.Header(), "Content-Length", resp.Header)
|
|
trySet(w.Header(), "Content-Range", resp.Header)
|
|
|
|
w.WriteHeader(resp.StatusCode)
|
|
io.Copy(w, resp.Body)
|
|
}
|