66 lines
1.2 KiB
Go
66 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"github.com/vbatts/go-bt/bencode"
|
|
"github.com/vbatts/go-bt/torrent"
|
|
"io/ioutil"
|
|
"os"
|
|
)
|
|
|
|
var (
|
|
flOutput = flag.String("o", "", "output the re-encoded torrent to file at this path")
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
if len(*flOutput) > 0 && flag.NArg() > 1 {
|
|
fmt.Fprintf(os.Stderr, "-o and multiple input files can not be used together")
|
|
os.Exit(1)
|
|
}
|
|
|
|
for _, arg := range flag.Args() {
|
|
fh, err := os.Open(arg)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", err)
|
|
continue
|
|
}
|
|
|
|
buf, err := ioutil.ReadAll(fh)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", err)
|
|
continue
|
|
}
|
|
fh.Close()
|
|
|
|
tf := torrent.File{}
|
|
err = bencode.Unmarshal(buf, &tf)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", err)
|
|
continue
|
|
}
|
|
|
|
fmt.Printf("Loaded: %s (%d files)\n", tf.Info.Name, len(tf.Info.Files))
|
|
|
|
if len(*flOutput) > 0 {
|
|
fhOutput, err := os.Create(*flOutput)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", err)
|
|
continue
|
|
}
|
|
buf, err = bencode.Marshal(tf)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", err)
|
|
}
|
|
_, err = fhOutput.Write(buf)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", err)
|
|
}
|
|
fmt.Printf("wrote: %s\n", fhOutput.Name())
|
|
fhOutput.Close()
|
|
}
|
|
}
|
|
}
|