package router import ( "encoding/hex" "errors" "github.com/ewhal/nyaa/util" "github.com/ewhal/nyaa/util/metainfo" "github.com/zeebo/bencode" "net/http" "net/url" "strconv" "strings" ) // UploadForm serializing HTTP form for torrent upload type UploadForm struct { Name string Magnet string Infohash string Category string CategoryId int SubCategoryId int Description string } // TODO: these should be in another package (?) // form value for torrent name const UploadFormName = "name" // form value for torrent file const UploadFormTorrent = "torrent" // form value for magnet uri (?) const UploadFormMagnet = "magnet" // form value for category const UploadFormCategory = "c" // form value for description const UploadFormDescription = "desc" // error indicating a torrent is private var ErrPrivateTorrent = errors.New("torrent is private") // error indicating a torrent's name is invalid var ErrInvalidTorrentName = errors.New("torrent name is invalid") // error indicating a torrent's description is invalid var ErrInvalidTorrentDescription = errors.New("torrent description is invalid") // error indicating a torrent's category is invalid var ErrInvalidTorrentCategory = errors.New("torrent category is invalid") /** UploadForm.ExtractInfo takes an http request and computes all fields for this form */ func (f *UploadForm) ExtractInfo(r *http.Request) error { f.Name = r.FormValue(UploadFormName) f.Category = r.FormValue(UploadFormCategory) f.Description = r.FormValue(UploadFormDescription) f.Magnet = r.FormValue(UploadFormMagnet) // trim whitespaces f.Name = util.TrimWhitespaces(f.Name) f.Description = util.TrimWhitespaces(f.Description) f.Magnet = util.TrimWhitespaces(f.Magnet) if len(f.Name) == 0 { return ErrInvalidTorrentName } if len(f.Description) == 0 { return ErrInvalidTorrentDescription } catsSplit := strings.Split(f.Category, "_") // need this to prevent out of index panics if len(catsSplit) == 2 { CatId, err := strconv.Atoi(catsSplit[0]) if err != nil { return ErrInvalidTorrentCategory } SubCatId, err := strconv.Atoi(catsSplit[1]) if err != nil { return ErrInvalidTorrentCategory } f.CategoryId = CatId f.SubCategoryId = SubCatId } else { return ErrInvalidTorrentCategory } if len(f.Magnet) == 0 { // try parsing torrent file if provided if no magnet is specified tfile, _, err := r.FormFile(UploadFormTorrent) if err != nil { return err } var torrent metainfo.TorrentFile // decode torrent err = bencode.NewDecoder(tfile).Decode(&torrent) if err != nil { return metainfo.ErrInvalidTorrentFile } // check if torrent is private if torrent.IsPrivate() { return ErrPrivateTorrent } // generate magnet binInfohash := torrent.Infohash() f.Infohash = hex.EncodeToString(binInfohash[:]) f.Magnet = util.InfoHashToMagnet(f.Infohash, f.Name) f.Infohash = strings.ToUpper(f.Infohash) } else { magnetUrl, parseErr := url.Parse(f.Magnet) if parseErr != nil { return metainfo.ErrInvalidTorrentFile } exactTopic := magnetUrl.Query().Get("xt") if !strings.HasPrefix(exactTopic, "urn:btih:") { return metainfo.ErrInvalidTorrentFile } else { f.Infohash = strings.ToUpper(strings.TrimPrefix(exactTopic, "urn:btih:")) } } return nil } // NewUploadForm creates a new upload form given parameters as list func NewUploadForm(params ...string) (uploadForm UploadForm) { if len(params) > 1 { uploadForm.Category = params[0] } else { uploadForm.Category = "3_12" } if len(params) > 2 { uploadForm.Description = params[1] } else { uploadForm.Description = "Description" } return }