nyaa-pantsu/models/file.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:]
}