package storage import ( "encoding/hex" "errors" "git.lumeweb.com/LumeWeb/libs5-go/encoding" "git.lumeweb.com/LumeWeb/libs5-go/types" "git.lumeweb.com/LumeWeb/portal/db/models" "io" "time" ) type FileImpl struct { reader io.ReadCloser hash []byte storage *StorageServiceDefault record *models.Upload cid *encoding.CID read bool } func NewFile(hash []byte, storage *StorageServiceDefault) *FileImpl { return &FileImpl{hash: hash, storage: storage, read: false} } func (f *FileImpl) Exists() bool { exists, _ := f.storage.FileExists(f.hash) return exists } func (f *FileImpl) Read(p []byte) (n int, err error) { err = f.init(0) if err != nil { return 0, err } f.read = true return f.reader.Read(p) } func (f *FileImpl) Seek(offset int64, whence int) (int64, error) { switch whence { case io.SeekStart: if !f.read { return 0, nil } if f.reader != nil { err := f.reader.Close() if err != nil { return 0, err } f.reader = nil } err := f.init(offset) if err != nil { return 0, err } case io.SeekCurrent: return 0, errors.New("not supported") case io.SeekEnd: return int64(f.Size()), nil default: return 0, errors.New("invalid whence") } return 0, nil } func (f *FileImpl) Close() error { if f.reader != nil { r := f.reader f.reader = nil return r.Close() } return nil } func (f *FileImpl) init(offset int64) error { if f.reader == nil { reader, _, err := f.storage.GetFile(f.hash, offset) if err != nil { return err } f.reader = reader f.read = false } return nil } func (f *FileImpl) Record() (*models.Upload, error) { if f.record == nil { exists, record := f.storage.FileExists(f.hash) if !exists { return nil, errors.New("file does not exist") } f.record = &record } return f.record, nil } func (f *FileImpl) Hash() []byte { hashStr := f.HashString() if hashStr == "" { return nil } str, err := hex.DecodeString(hashStr) if err != nil { return nil } return str } func (f *FileImpl) HashString() string { record, err := f.Record() if err != nil { return "" } return record.Hash } func (f *FileImpl) Name() string { cid, _ := f.CID().ToString() return cid } func (f *FileImpl) Modtime() time.Time { record, err := f.Record() if err != nil { return time.Unix(0, 0) } return record.CreatedAt } func (f *FileImpl) Size() uint64 { record, err := f.Record() if err != nil { return 0 } return record.Size } func (f *FileImpl) CID() *encoding.CID { if f.cid == nil { multihash := encoding.MultihashFromBytes(f.Hash(), types.HashTypeBlake3) cid := encoding.NewCID(types.CIDTypeRaw, *multihash, f.Size()) f.cid = cid } return f.cid } func (f *FileImpl) Mime() string { record, err := f.Record() if err != nil { return "" } return record.MimeType }