713ab02450
* Added configor as a new library
Now config is a singleton. You only need to do config.Get() instead of doing config.Conf.
* Forgot godep save 🐤
* Fix accidental removal of }
54 lignes
1,2 Kio
Go
54 lignes
1,2 Kio
Go
package models
|
|
|
|
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"`
|
|
}
|
|
|
|
// FileJSON for file model in json
|
|
type FileJSON struct {
|
|
Path string `json:"path"`
|
|
Filesize int64 `json:"filesize"`
|
|
}
|
|
|
|
// TableName : Return the name of files table
|
|
func (f File) TableName() string {
|
|
return config.Get().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]
|
|
}
|