updating the example to use the new Marshal and Unmarshal
This commit is contained in:
parent
0d6d4580ca
commit
8922de6783
1 changed files with 10 additions and 4 deletions
|
@ -5,6 +5,7 @@ import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/vbatts/go-bt/bencode"
|
"github.com/vbatts/go-bt/bencode"
|
||||||
"github.com/vbatts/go-bt/torrent"
|
"github.com/vbatts/go-bt/torrent"
|
||||||
|
"io/ioutil"
|
||||||
"os"
|
"os"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -27,19 +28,20 @@ func main() {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
data, err := bencode.Decode(fh)
|
buf, err := ioutil.ReadAll(fh)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Fprintf(os.Stderr, "%s\n", err)
|
fmt.Fprintf(os.Stderr, "%s\n", err)
|
||||||
fh.Close()
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
fh.Close()
|
fh.Close()
|
||||||
|
|
||||||
tf, err := torrent.DecocdeTorrentData(data)
|
tf := torrent.File{}
|
||||||
|
err = bencode.Unmarshal(buf, &tf)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Fprintf(os.Stderr, "%s\n", err)
|
fmt.Fprintf(os.Stderr, "%s\n", err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
fmt.Printf("Loaded: %s (%d files)\n", tf.Info.Name, len(tf.Info.Files))
|
fmt.Printf("Loaded: %s (%d files)\n", tf.Info.Name, len(tf.Info.Files))
|
||||||
|
|
||||||
if len(*flOutput) > 0 {
|
if len(*flOutput) > 0 {
|
||||||
|
@ -48,7 +50,11 @@ func main() {
|
||||||
fmt.Fprintf(os.Stderr, "%s\n", err)
|
fmt.Fprintf(os.Stderr, "%s\n", err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
err = bencode.Marshal(fhOutput, *tf)
|
buf, err = bencode.Marshal(tf)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Fprintf(os.Stderr, "%s\n", err)
|
||||||
|
}
|
||||||
|
_, err = fhOutput.Write(buf)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Fprintf(os.Stderr, "%s\n", err)
|
fmt.Fprintf(os.Stderr, "%s\n", err)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue