示例#1
1
文件: main.go 项目: Ragnis/autousts
func sync(db *bolt.DB) {
	tc := transmission.New("http://127.0.0.1:9091", "", "")
	if _, err := tc.GetTorrents(); err != nil {
		fmt.Println("Could not connect to Transmission RPC API: " + err.Error())
		return
	}

	var (
		shows   []*Show
		results []*search.Result
	)

	waiting := 0
	fin := make(chan bool)
	rec := make(chan *search.Result)

	db.View(func(tx *bolt.Tx) error {
		b := tx.Bucket([]byte("Shows"))
		c := b.Cursor()

		for k, v := c.First(); k != nil; k, v = c.Next() {
			show, err := ShowFromBytes(v)
			if err == nil {
				shows = append(shows, show)
			}
		}

		return nil
	})

	for _, show := range shows {
		if show.Paused {
			continue
		}

		waiting += 1
		go syncShow(show, rec, fin)
	}

	for waiting > 0 {
		select {
		case result := <-rec:
			results = append(results, result)

		case <-fin:
			waiting -= 1
		}
	}

	for _, result := range results {
		fmt.Printf("Found torrent: '%s'\n", result.Name)

		if _, err := tc.AddTorrentByFilename(result.MagnetURL, ""); err != nil {
			fmt.Println("Error adding torrent: " + err.Error())
			fmt.Println("Stopping...")
			return
		}
	}

	fmt.Printf("Found %d torrent(s)\n", len(results))

	err := db.Batch(func(tx *bolt.Tx) error {
		b := tx.Bucket([]byte("Shows"))

		for _, show := range shows {
			if err := b.Put([]byte(show.Name), show.Bytes()); err != nil {
				return err
			}
		}

		return nil
	})

	if err != nil {
		fmt.Printf("Error occurred while saving shows: %s", err.Error())
	}
}
示例#2
0
func downloadThing(url, dest string) error {
	if _, err := os.Stat(dest); err == nil {
		animu.LogVerbose(fmt.Sprintf("  File %s exists; skipping", dest))
		return dlSkipped
	}

	resp, err := http.Get(url)
	if err != nil {
		return fmt.Errorf("Unable to download '%s': %s", url, err)
	}
	defer resp.Body.Close()

	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return fmt.Errorf("Unable to read response: %s", err)
	}

	err = ioutil.WriteFile(dest, body, os.FileMode(0644))
	if err != nil {
		return fmt.Errorf("Unable to write to file '%s': %s", dest, err)
	}

	animu.LogMsg(fmt.Sprintf("Downloaded %s", path.Base(dest)))
	if rpcUrl != "" {
		client := gt.New(rpcUrl, rpcUser, rpcPass)
		t, err := client.AddTorrentByFile(dest, rpcDir)
		if err != nil {
			return fmt.Errorf("Error adding torrent to transmission: %s", err)
		}

		animu.LogVerbose(fmt.Sprintf("Torrent added to transmission: [%d] {%s} %s", t.ID, t.HashString, t.Name))
		animu.LogMsg(fmt.Sprintf("Added %s to transmission", path.Base(dest)))
	}

	return nil
}