about summary refs log tree commit diff stats
path: root/internal/storage/files/file.go
blob: a71811cd38519b323a81d8ea09f73c8cb9c05100 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package files

import (
	"fmt"
	"hash/fnv"
	"io"
	"mime"
	"os"
	"path/filepath"
	"strings"

	"gitlab.com/tozd/go/errors"
	"go.alanpearce.eu/website/internal/storage"
)

type File struct {
	storage.File
}

var encodings = map[string]string{
	"br":   ".br",
	"gzip": ".gz",
}

func (r *Reader) OpenFile(path string, filename string) (*File, error) {
	f, err := os.Open(filename)
	if err != nil {
		return nil, errors.WithMessage(err, "could not open file for reading")
	}
	stat, err := f.Stat()
	if err != nil {
		return nil, errors.WithMessage(err, "could not stat file")
	}

	etag, err := etag(f)
	if err != nil {
		return nil, errors.WithMessage(err, "could not calculate etag")
	}

	file := &File{
		File: storage.File{
			Path:         path,
			ContentType:  mime.TypeByExtension(filepath.Ext(filename)),
			LastModified: stat.ModTime(),
			Etag:         etag,
			Encodings: map[string]io.ReadSeekCloser{
				"identity": f,
			},
		},
	}

	for enc, suffix := range encodings {
		_, err := os.Stat(filename + suffix)
		if err != nil {
			if errors.Is(err, os.ErrNotExist) {
				continue
			}

			return nil, errors.WithMessagef(err, "could not stat file %s", filename+suffix)
		}
		file.Encodings[enc], err = os.Open(filename + suffix)
		if err != nil {
			return nil, errors.WithMessagef(err, "could not read file %s", filename+suffix)
		}
	}

	return file, nil
}

func etag(f io.Reader) (string, error) {
	hash := fnv.New64a()
	if _, err := io.Copy(hash, f); err != nil {
		return "", errors.WithMessage(err, "could not hash file")
	}

	return fmt.Sprintf(`W/"%x"`, hash.Sum(nil)), nil
}

func (f *File) Close() error {
	var errs []error
	for _, enc := range f.Encodings {
		if err := enc.Close(); err != nil {
			errs = append(errs, err)
		}
	}

	return errors.Join(errs...)
}

func pathNameToFileName(pathname string) string {
	if strings.HasSuffix(pathname, "/") {
		pathname = pathname + "index.html"
	}

	return pathname
}

func fileNameToPathName(filename string) string {
	pathname, _ := strings.CutSuffix(filename, "index.html")

	return pathname
}