80 lines
1.4 KiB
Go
80 lines
1.4 KiB
Go
package cmd
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/lbryio/reflector.go/server/peer"
|
|
"github.com/lbryio/reflector.go/store"
|
|
|
|
"github.com/lbryio/lbry.go/v2/stream"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func init() {
|
|
var cmd = &cobra.Command{
|
|
Use: "getstream ADDRESS:PORT SDHASH",
|
|
Short: "Get a stream from a reflector server",
|
|
Args: cobra.ExactArgs(2),
|
|
Run: getStreamCmd,
|
|
}
|
|
rootCmd.AddCommand(cmd)
|
|
}
|
|
|
|
func getStreamCmd(cmd *cobra.Command, args []string) {
|
|
addr := args[0]
|
|
sdHash := args[1]
|
|
|
|
s := store.NewCachingStore(
|
|
"getstream",
|
|
peer.NewStore(peer.StoreOpts{Address: addr}),
|
|
store.NewDiskStore("/tmp/lbry_downloaded_blobs", 2),
|
|
)
|
|
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
var sd stream.SDBlob
|
|
|
|
sdb, _, err := s.Get(sdHash)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = sd.FromBlob(sdb)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
filename := sd.SuggestedFileName
|
|
if filename == "" {
|
|
filename = "stream_" + time.Now().Format("20060102_150405")
|
|
}
|
|
|
|
f, err := os.Create(wd + "/" + filename)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
for i := 0; i < len(sd.BlobInfos)-1; i++ {
|
|
b, _, err := s.Get(hex.EncodeToString(sd.BlobInfos[i].BlobHash))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
data, err := b.Plaintext(sd.Key, sd.BlobInfos[i].IV)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
_, err = f.Write(data)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
}
|