Rewrite in Go

1 files changed, 36 insertions, 0 deletions

diff --git a/pkg/site/fs.go b/pkg/site/fs.go
new file mode 100644
index 0000000..a77da63
--- /dev/null
+++ b/pkg/site/fs.go
@@ -0,0 +1,36 @@
+package site
+
+import (
+	"net/http"
+	"path/filepath"
+)
+
+type siteFS struct {
+	fs http.FileSystem
+}
+
+func (sfs siteFS) Open(path string) (http.File, error) {
+	f, err := sfs.fs.Open(path)
+	if err != nil {
+		return nil, err
+	}
+
+	s, err := f.Stat()
+	if err != nil {
+		return nil, err
+	}
+
+	if s.IsDir() {
+		index := filepath.Join(path, "index.html")
+		if _, err := sfs.fs.Open(index); err != nil {
+			closeErr := f.Close()
+			if closeErr != nil {
+				return nil, closeErr
+			}
+
+			return nil, err
+		}
+	}
+
+	return f, nil
+}