storj/cmd/internal/asset/inmemory.go
Egon Elbre 080ba47a06 all: fix dots
Change-Id: I6a419c62700c568254ff67ae5b73efed2fc98aa2
2020-07-16 14:58:28 +00:00

118 lines
2.6 KiB
Go

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
package asset
import (
"bytes"
"errors"
"net/http"
"os"
"path"
"time"
)
var _ http.FileSystem = (*InmemoryFileSystem)(nil)
// InmemoryFileSystem defines an inmemory http.FileSystem.
type InmemoryFileSystem struct {
Root *Asset
Index map[string]*Asset
}
// Inmemory creates an InmemoryFileSystem from.
func Inmemory(root *Asset) *InmemoryFileSystem {
fs := &InmemoryFileSystem{}
fs.Root = root
fs.Index = map[string]*Asset{}
fs.reindex("/", "", root)
return fs
}
// reindex inserts a node to the index.
func (fs *InmemoryFileSystem) reindex(prefix, name string, file *Asset) {
fs.Index[path.Join(prefix, name)] = file
for _, child := range file.Children {
fs.reindex(path.Join(prefix, name), child.Name, child)
}
}
// Open opens the file at the specified path.
func (fs *InmemoryFileSystem) Open(path string) (http.File, error) {
asset, ok := fs.Index[path]
if !ok {
return nil, os.ErrNotExist
}
return asset.File(), nil
}
// File opens the particular asset as a file.
func (asset *Asset) File() *File {
return &File{*bytes.NewReader(asset.Data), asset}
}
// File defines a readable file.
type File struct {
bytes.Reader
*Asset
}
// Readdir reads all file infos from the directory.
func (file *File) Readdir(count int) ([]os.FileInfo, error) {
if !file.Mode.IsDir() {
return nil, errors.New("not a directory")
}
if count > len(file.Children) {
count = len(file.Children)
}
infos := make([]os.FileInfo, 0, count)
for _, child := range file.Children {
infos = append(infos, child.stat())
}
return infos, nil
}
func (asset *Asset) stat() FileInfo {
return FileInfo{
name: asset.Name,
size: int64(len(asset.Data)),
mode: asset.Mode,
modTime: asset.ModTime,
}
}
// Stat returns stats about the file.
func (file *File) Stat() (os.FileInfo, error) { return file.stat(), nil }
// Close closes the file.
func (file *File) Close() error { return nil }
// FileInfo implements file info.
type FileInfo struct {
name string
size int64
mode os.FileMode
modTime time.Time
}
// Name implements os.FileInfo.
func (info FileInfo) Name() string { return info.name }
// Size implements os.FileInfo.
func (info FileInfo) Size() int64 { return info.size }
// Mode implements os.FileInfo.
func (info FileInfo) Mode() os.FileMode { return info.mode }
// ModTime implements os.FileInfo.
func (info FileInfo) ModTime() time.Time { return info.modTime }
// IsDir implements os.FileInfo.
func (info FileInfo) IsDir() bool { return info.mode.IsDir() }
// Sys implements os.FileInfo.
func (info FileInfo) Sys() interface{} { return nil }