package model import ( "github.com/NyaaPantsu/nyaa/config" "github.com/zeebo/bencode" ) // File model type File struct { ID uint `gorm:"column:file_id;primary_key"` TorrentID uint `gorm:"column:torrent_id;unique_index:idx_tid_path"` // this path is bencode'd, call Path() to obtain BencodedPath string `gorm:"column:path;unique_index:idx_tid_path"` Filesize int64 `gorm:"column:filesize"` } // TableName : Return the name of files table func (f File) TableName() string { return config.Conf.Models.FilesTableName } // Size : Returns the total size of memory allocated for this struct func (f File) Size() int { return (2 + len(f.BencodedPath) + 1) * 8 } // Path : Returns the path to the file func (f *File) Path() (out []string) { bencode.DecodeString(f.BencodedPath, &out) return } // SetPath : Set the path of the file func (f *File) SetPath(path []string) error { encoded, err := bencode.EncodeString(path) if err != nil { return err } f.BencodedPath = encoded return nil } // Filename : Returns the filename of the file func (f *File) Filename() string { path := f.Path() return path[len(path)-1] }