83 lignes
1.8 KiB
Go
83 lignes
1.8 KiB
Go
package models
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"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]
|
|
}
|
|
|
|
// FilenameWithoutExtension : Returns the filename of the file without the extension
|
|
func (f *File) FilenameWithoutExtension() string {
|
|
path := f.Path()
|
|
fileName := path[len(path)-1]
|
|
index := strings.LastIndex(fileName, ".")
|
|
|
|
if index == -1 {
|
|
return fileName
|
|
}
|
|
|
|
return fileName[:index]
|
|
}
|
|
|
|
// FilenameExtension : Returns the extension of a filename, or an empty string
|
|
func (f *File) FilenameExtension() string {
|
|
path := f.Path()
|
|
fileName := path[len(path)-1]
|
|
index := strings.LastIndex(fileName, ".")
|
|
|
|
if index == -1 {
|
|
return ""
|
|
}
|
|
|
|
return fileName[index:]
|
|
}
|