Started seperating the main.go package into multiple packages

2017-12-17 21:49:32 -05:00
parent c408801447
commit b7c5032c37
3 changed files with 5 additions and 5 deletions
--- a/storage/storage.go
+++ b/storage/storage.go
@@ -0,0 +1,135 @@
+package main
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/boltdb/bolt"
+)
+
+type TorrentLocal struct { //local storage of the torrents for readd on server restart
+	Hash            string
+	DateAdded       string
+	StoragePath     string
+	TorrentName     string
+	TorrentStatus   string
+	TorrentType     string //magnet or .torrent file
+	TorrentFileName string
+}
+
+func readInTorrents(torrentStorage *bolt.DB) (torrentLocalArray []*TorrentLocal) {
+
+	torrentLocalArray = []*TorrentLocal{}
+
+	torrentStorage.View(func(tx *bolt.Tx) error {
+
+		tx.ForEach(func(name []byte, b *bolt.Bucket) error {
+			torrentLocal := new(TorrentLocal) //create a struct to store to an array //TODO clean this the fuck up just read the struct into something that converts them all the byte arrays
+			var Dateadded []byte
+			var StoragePath []byte
+			var Hash []byte
+			var TorrentName []byte
+			var TorrentStatus []byte
+			var TorrentType []byte
+			var TorrentFileName []byte
+			Dateadded = b.Get([]byte("Date"))
+			if Dateadded == nil {
+				fmt.Println("Date added error!")
+				Dateadded = []byte(time.Now().Format("Jan _2 2006"))
+			}
+			StoragePath = b.Get([]byte("StoragePath"))
+			if StoragePath == nil {
+				fmt.Println("StoragePath error!")
+				StoragePath = []byte("downloads")
+			}
+			Hash = b.Get([]byte("InfoHash"))
+			if Hash == nil {
+				fmt.Println("Hash error!")
+			}
+			TorrentName = b.Get([]byte("TorrentName"))
+			if TorrentName == nil {
+				fmt.Println("Torrent Name not found")
+				TorrentName = []byte("Not Found!")
+			}
+			TorrentStatus = b.Get([]byte("TorrentStatus"))
+			if TorrentStatus == nil {
+				fmt.Println("Torrent Status not found in local storage")
+				TorrentStatus = []byte("")
+			}
+			TorrentType = b.Get([]byte("TorrentType"))
+			if TorrentType == nil {
+				fmt.Println("Torrent Type not found in local storage")
+				TorrentStatus = []byte("")
+			}
+			TorrentFileName = b.Get([]byte("TorrentFileName"))
+			if TorrentFileName == nil {
+				fmt.Println("Torrent File Name not found in local storage")
+				TorrentFileName = []byte("")
+			}
+
+			torrentLocal.DateAdded = string(Dateadded)
+			torrentLocal.StoragePath = string(StoragePath)
+			torrentLocal.Hash = string(Hash) //Converting the byte slice back into the full hash
+			torrentLocal.TorrentName = string(TorrentName)
+			torrentLocal.TorrentStatus = string(TorrentStatus)
+			torrentLocal.TorrentType = string(TorrentType)
+			torrentLocal.TorrentFileName = string(TorrentFileName)
+
+			fmt.Println("Torrentlocal list: ", torrentLocal)
+			torrentLocalArray = append(torrentLocalArray, torrentLocal) //dumping it into the array
+			return nil
+		})
+		return nil
+	})
+
+	return torrentLocalArray //all done, return the entire Array to add to the torrent client
+}
+
+func addTorrentLocalStorage(torrentStorage *bolt.DB, local *TorrentLocal) {
+	println("Adding Local storage information")
+
+	torrentStorage.Update(func(tx *bolt.Tx) error {
+		b, err := tx.CreateBucketIfNotExists([]byte(local.Hash)) //translating hash into bytes for storage
+		if err != nil {
+			return fmt.Errorf("create bucket %s", err)
+		}
+		err = b.Put([]byte("Date"), []byte(local.DateAdded)) //TODO error checking  marshall into JSON
+		if err != nil {
+			return err
+		}
+		err = b.Put([]byte("StoragePath"), []byte(local.StoragePath))
+		if err != nil {
+			return err
+		}
+		err = b.Put([]byte("InfoHash"), []byte(local.Hash))
+		if err != nil {
+			return err
+		}
+		err = b.Put([]byte("TorrentName"), []byte(local.TorrentName))
+		if err != nil {
+			return err
+		}
+		err = b.Put([]byte("TorrentType"), []byte(local.TorrentType))
+		if err != nil {
+			return err
+		}
+		err = b.Put([]byte("TorrentFileName"), []byte(local.TorrentFileName))
+		if err != nil {
+			return err
+		}
+		return nil
+	})
+}
+
+func delTorrentLocalStorage(torrentStorage *bolt.DB, local *TorrentLocal) { //deleting a torrent by hash
+	println("Deleting torrent", local.TorrentFileName)
+
+	torrentStorage.Update(func(tx *bolt.Tx) error {
+		err := tx.DeleteBucket([]byte(local.Hash))
+		if err != nil {
+			return err
+		}
+		return nil
+	})
+
+}