roulette/types/text/text.go

/*
Copyright © 2024 Seednode <seednode@seedno.de>
*/

package text

import (
	"errors"
	"fmt"
	"net/http"
	"os"
	"strings"
	"unicode/utf8"

	"seedno.de/seednode/roulette/types"
)

type Format struct{}

func (t Format) CSP(w http.ResponseWriter) string {
	nonce := types.GetNonce()

	w.Header().Add("Content-Security-Policy", fmt.Sprintf("default-src 'self' 'nonce-%s';", nonce))

	return nonce
}

func (t Format) CSS() string {
	var css strings.Builder

	css.WriteString(`html,body{margin:0;padding:0;height:100%;}`)
	css.WriteString(`a{color:inherit;display:block;height:100%;width:100%;text-decoration:none;overflow:hidden;}`)
	css.WriteString(`textarea{border:none;caret-color:transparent;outline:none;margin:.5rem;`)
	css.WriteString(`height:99%;width:99%;white-space:pre;overflow:auto;}`)

	return css.String()
}

func (t Format) Title(rootUrl, fileUri, filePath, fileName, prefix, mime string) (string, error) {
	return fmt.Sprintf(`<title>%s</title>`, fileName), nil
}

func (t Format) Body(rootUrl, fileUri, filePath, fileName, prefix, mime, nonce string) (string, error) {
	body, err := os.ReadFile(filePath)
	if err != nil {
		body = []byte{}
	}

	return fmt.Sprintf(`<a href="%s"><textarea autofocus readonly>%s</textarea></a>`,
		rootUrl,
		body), nil
}

func (t Format) Extensions() map[string]string {
	return map[string]string{
		`.csv`: `text/csv`,
		`.txt`: `text/plain`,
	}
}

func (t Format) MediaType(extension string) string {
	extensions := t.Extensions()

	value, exists := extensions[extension]
	if exists {
		return value
	}

	return ""
}

func (t Format) Validate(filePath string) bool {
	file, err := os.Open(filePath)
	switch {
	case errors.Is(err, os.ErrNotExist):
		return false
	case err != nil:
		return false
	}
	defer file.Close()

	head := make([]byte, 512)
	file.Read(head)

	return utf8.Valid(head)
}

func (t Format) Type() string {
	return "inline"
}

func init() {
	types.SupportedFormats.Register(Format{})
}