tstor/src/mounts/httpfs/httpfs.go

package httpfs

import (
	"io"
	"io/fs"
	"net/http"
	"os"
	"sync"

	"git.kmsign.ru/royalcat/tstor/src/host/vfs"
	"git.kmsign.ru/royalcat/tstor/src/iio"
)

var _ http.FileSystem = &HTTPFS{}

type HTTPFS struct {
	fs vfs.Filesystem
}

func NewHTTPFS(fs vfs.Filesystem) *HTTPFS {
	return &HTTPFS{fs: fs}
}

func (fs *HTTPFS) Open(name string) (http.File, error) {
	f, err := fs.fs.Open(name)
	if err != nil {
		return nil, err
	}

	fi := vfs.NewFileInfo(name, f.Size(), f.IsDir())

	// TODO make this lazy
	fis, err := fs.filesToFileInfo(name)
	if err != nil {
		return nil, err
	}

	return newHTTPFile(f, fis, fi), nil
}

func (fs *HTTPFS) filesToFileInfo(path string) ([]fs.FileInfo, error) {
	files, err := fs.fs.ReadDir(path)
	if err != nil {
		return nil, err
	}

	var out []os.FileInfo
	for n, f := range files {
		out = append(out, vfs.NewFileInfo(n, f.Size(), f.IsDir()))
	}

	return out, nil
}

var _ http.File = &httpFile{}

type httpFile struct {
	iio.ReaderSeeker

	mu sync.Mutex
	// dirPos is protected by mu.
	dirPos     int
	dirContent []os.FileInfo

	fi fs.FileInfo
}

func newHTTPFile(f vfs.File, fis []fs.FileInfo, fi fs.FileInfo) *httpFile {
	return &httpFile{
		dirContent: fis,
		fi:         fi,

		ReaderSeeker: iio.NewSeekerWrapper(f, f.Size()),
	}
}

func (f *httpFile) Readdir(count int) ([]fs.FileInfo, error) {
	f.mu.Lock()
	defer f.mu.Unlock()

	if !f.fi.IsDir() {
		return nil, os.ErrInvalid
	}

	old := f.dirPos
	if old >= len(f.dirContent) {
		// The os.File Readdir docs say that at the end of a directory,
		// the error is io.EOF if count > 0 and nil if count <= 0.
		if count > 0 {
			return nil, io.EOF
		}
		return nil, nil
	}
	if count > 0 {
		f.dirPos += count
		if f.dirPos > len(f.dirContent) {
			f.dirPos = len(f.dirContent)
		}
	} else {
		f.dirPos = len(f.dirContent)
		old = 0
	}

	return f.dirContent[old:f.dirPos], nil
}

func (f *httpFile) Stat() (fs.FileInfo, error) {
	return f.fi, nil
}