go-bt/example_loader.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()
		}
	}
}