Albirew/nyaa-pantsu
Archivé
1
0
Bifurcation 0
Ce dépôt a été archivé le 2022-05-07. Vous pouvez voir ses fichiers ou le cloner, mais pas ouvrir de ticket ou de demandes d'ajout, ni soumettre de changements.
nyaa-pantsu/models/file.go

83 lignes
1,8 Kio
Go
Brut Vue normale Historique

package models
import (
"strings"
"github.com/NyaaPantsu/nyaa/config"
"github.com/zeebo/bencode"
)
// File model
type File struct {
2017-05-24 09:11:13 +02:00
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 {
2017-05-24 09:11:13 +02:00
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:]
}