forked from tangled.org/core
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363package pages
import ( "context" "crypto/hmac" "crypto/sha256" "encoding/hex" "errors" "fmt" "html" "html/template" "log" "math" "net/url" "path/filepath" "reflect" "strings" "time"
"github.com/dustin/go-humanize" "github.com/go-enry/go-enry/v2" "tangled.sh/tangled.sh/core/appview/filetree" "tangled.sh/tangled.sh/core/appview/pages/markup" "tangled.sh/tangled.sh/core/crypto")
func (p *Pages) funcMap() template.FuncMap { return template.FuncMap{ "split": func(s string) []string { return strings.Split(s, "\n") }, "resolve": func(s string) string { identity, err := p.resolver.ResolveIdent(context.Background(), s)
if err != nil { return s }
if identity.Handle.IsInvalidHandle() { return "handle.invalid" }
return "@" + identity.Handle.String() }, "truncateAt30": func(s string) string { if len(s) <= 30 { return s } return s[:30] + "…" }, "splitOn": func(s, sep string) []string { return strings.Split(s, sep) }, "int64": func(a int) int64 { return int64(a) }, "add": func(a, b int) int { return a + b }, "now": func() time.Time { return time.Now() }, // the absolute state of go templates "add64": func(a, b int64) int64 { return a + b }, "sub": func(a, b int) int { return a - b }, "f64": func(a int) float64 { return float64(a) }, "addf64": func(a, b float64) float64 { return a + b }, "subf64": func(a, b float64) float64 { return a - b }, "mulf64": func(a, b float64) float64 { return a * b }, "divf64": func(a, b float64) float64 { if b == 0 { return 0 } return a / b }, "negf64": func(a float64) float64 { return -a }, "cond": func(cond any, a, b string) string { if cond == nil { return b }
if boolean, ok := cond.(bool); boolean && ok { return a }
return b }, "didOrHandle": func(did, handle string) string { if handle != "" { return fmt.Sprintf("@%s", handle) } else { return did } }, "assoc": func(values ...string) ([][]string, error) { if len(values)%2 != 0 { return nil, fmt.Errorf("invalid assoc call, must have an even number of arguments") } pairs := make([][]string, 0) for i := 0; i < len(values); i += 2 { pairs = append(pairs, []string{values[i], values[i+1]}) } return pairs, nil }, "append": func(s []string, values ...string) []string { s = append(s, values...) return s }, "commaFmt": humanize.Comma, "relTimeFmt": humanize.Time, "shortRelTimeFmt": func(t time.Time) string { return humanize.CustomRelTime(t, time.Now(), "", "", []humanize.RelTimeMagnitude{ {time.Second, "now", time.Second}, {2 * time.Second, "1s %s", 1}, {time.Minute, "%ds %s", time.Second}, {2 * time.Minute, "1min %s", 1}, {time.Hour, "%dmin %s", time.Minute}, {2 * time.Hour, "1hr %s", 1}, {humanize.Day, "%dhrs %s", time.Hour}, {2 * humanize.Day, "1d %s", 1}, {20 * humanize.Day, "%dd %s", humanize.Day}, {8 * humanize.Week, "%dw %s", humanize.Week}, {humanize.Year, "%dmo %s", humanize.Month}, {18 * humanize.Month, "1y %s", 1}, {2 * humanize.Year, "2y %s", 1}, {humanize.LongTime, "%dy %s", humanize.Year}, {math.MaxInt64, "a long while %s", 1}, }) }, "longTimeFmt": func(t time.Time) string { return t.Format("Jan 2, 2006, 3:04 PM MST") }, "iso8601DateTimeFmt": func(t time.Time) string { return t.Format("2006-01-02T15:04:05-07:00") }, "iso8601DurationFmt": func(duration time.Duration) string { days := int64(duration.Hours() / 24) hours := int64(math.Mod(duration.Hours(), 24)) minutes := int64(math.Mod(duration.Minutes(), 60)) seconds := int64(math.Mod(duration.Seconds(), 60)) return fmt.Sprintf("P%dD%dH%dM%dS", days, hours, minutes, seconds) }, "durationFmt": func(duration time.Duration) string { return durationFmt(duration, [4]string{"d", "hr", "min", "s"}) }, "longDurationFmt": func(duration time.Duration) string { return durationFmt(duration, [4]string{"days", "hours", "minutes", "seconds"}) }, "byteFmt": humanize.Bytes, "length": func(slice any) int { v := reflect.ValueOf(slice) if v.Kind() == reflect.Slice || v.Kind() == reflect.Array { return v.Len() } return 0 }, "splitN": func(s, sep string, n int) []string { return strings.SplitN(s, sep, n) }, "escapeHtml": func(s string) template.HTML { if s == "" { return template.HTML("<br>") } return template.HTML(s) }, "unescapeHtml": func(s string) string { return html.UnescapeString(s) }, "nl2br": func(text string) template.HTML { return template.HTML(strings.ReplaceAll(template.HTMLEscapeString(text), "\n", "<br>")) }, "unwrapText": func(text string) string { paragraphs := strings.Split(text, "\n\n")
for i, p := range paragraphs { lines := strings.Split(p, "\n") paragraphs[i] = strings.Join(lines, " ") }
return strings.Join(paragraphs, "\n\n") }, "sequence": func(n int) []struct{} { return make([]struct{}, n) }, // take atmost N items from this slice "take": func(slice any, n int) any { v := reflect.ValueOf(slice) if v.Kind() != reflect.Slice && v.Kind() != reflect.Array { return nil } if v.Len() == 0 { return nil } return v.Slice(0, min(n, v.Len())).Interface() }, "markdown": func(text string) template.HTML { p.rctx.RendererType = markup.RendererTypeDefault htmlString := p.rctx.RenderMarkdown(text) sanitized := p.rctx.SanitizeDefault(htmlString) return template.HTML(sanitized) }, "description": func(text string) template.HTML { p.rctx.RendererType = markup.RendererTypeDefault htmlString := p.rctx.RenderMarkdown(text) sanitized := p.rctx.SanitizeDescription(htmlString) return template.HTML(sanitized) }, "isNil": func(t any) bool { // returns false for other "zero" values return t == nil }, "list": func(args ...any) []any { return args }, "dict": func(values ...any) (map[string]any, error) { if len(values)%2 != 0 { return nil, errors.New("invalid dict call") } dict := make(map[string]any, len(values)/2) for i := 0; i < len(values); i += 2 { key, ok := values[i].(string) if !ok { return nil, errors.New("dict keys must be strings") } dict[key] = values[i+1] } return dict, nil }, "deref": func(v any) any { val := reflect.ValueOf(v) if val.Kind() == reflect.Ptr && !val.IsNil() { return val.Elem().Interface() } return nil }, "i": func(name string, classes ...string) template.HTML { data, err := icon(name, classes) if err != nil { log.Printf("icon %s does not exist", name) data, _ = icon("airplay", classes) } return template.HTML(data) }, "cssContentHash": CssContentHash, "fileTree": filetree.FileTree, "pathEscape": func(s string) string { return url.PathEscape(s) }, "pathUnescape": func(s string) string { u, _ := url.PathUnescape(s) return u },
"tinyAvatar": func(handle string) string { return p.avatarUri(handle, "tiny") }, "fullAvatar": func(handle string) string { return p.avatarUri(handle, "") }, "langColor": enry.GetColor, "layoutSide": func() string { return "col-span-1 md:col-span-2 lg:col-span-3" }, "layoutCenter": func() string { return "col-span-1 md:col-span-8 lg:col-span-6" },
"normalizeForHtmlId": func(s string) string { // TODO: extend this to handle other cases? return strings.ReplaceAll(s, ":", "_") }, "sshFingerprint": func(pubKey string) string { fp, err := crypto.SSHFingerprint(pubKey) if err != nil { return "error" } return fp }, }}
func (p *Pages) avatarUri(handle, size string) string { handle = strings.TrimPrefix(handle, "@")
secret := p.avatar.SharedSecret h := hmac.New(sha256.New, []byte(secret)) h.Write([]byte(handle)) signature := hex.EncodeToString(h.Sum(nil))
sizeArg := "" if size != "" { sizeArg = fmt.Sprintf("size=%s", size) } return fmt.Sprintf("%s/%s/%s?%s", p.avatar.Host, signature, handle, sizeArg)}
func icon(name string, classes []string) (template.HTML, error) { iconPath := filepath.Join("static", "icons", name)
if filepath.Ext(name) == "" { iconPath += ".svg" }
data, err := Files.ReadFile(iconPath) if err != nil { return "", fmt.Errorf("icon %s not found: %w", name, err) }
// Convert SVG data to string svgStr := string(data)
svgTagEnd := strings.Index(svgStr, ">") if svgTagEnd == -1 { return "", fmt.Errorf("invalid SVG format for icon %s", name) }
classTag := ` class="` + strings.Join(classes, " ") + `"`
modifiedSVG := svgStr[:svgTagEnd] + classTag + svgStr[svgTagEnd:] return template.HTML(modifiedSVG), nil}
func durationFmt(duration time.Duration, names [4]string) string { days := int64(duration.Hours() / 24) hours := int64(math.Mod(duration.Hours(), 24)) minutes := int64(math.Mod(duration.Minutes(), 60)) seconds := int64(math.Mod(duration.Seconds(), 60))
chunks := []struct { name string amount int64 }{ {names[0], days}, {names[1], hours}, {names[2], minutes}, {names[3], seconds}, }
parts := []string{}
for _, chunk := range chunks { if chunk.amount != 0 { parts = append(parts, fmt.Sprintf("%d%s", chunk.amount, chunk.name)) } }
return strings.Join(parts, " ")}