Move crawler into the main package

minicrawler/minicrawler.go → crawler.go

@@ -1,7 +1,4 @@
-// minicrawler is a super simple crawler.
-// It's basically a wrapper around http.Client to make some things easier
-// It's sole purpose is to do the fetching in Picture Life Rescuer.
-package minicrawler
+package main
 
 import (
 	"fmt"
@@ -23,7 +20,7 @@ type Crawler struct {
 	UserAgent string
 }
 
-func New() Crawler {
+func NewCrawler() Crawler {
 	jar, _ := cookiejar.New(nil)
 	return Crawler{
 		Client:    http.Client{Jar: jar},

main.go

@@ -1,3 +1,4 @@
+// Rescue as much as possible from your Picturelife account
 package main
 
 import (
@@ -17,7 +18,6 @@ import (
 	"golang.org/x/crypto/ssh/terminal"
 
 	"github.com/cheggaaa/pb"
-	"github.com/morphar/rescuelife/minicrawler"
 )
 
 type Media struct {
@@ -126,7 +126,7 @@ func main() {
 	}
 
 	// Instantiate the crawler
-	client := minicrawler.New()
+	client := NewCrawler()
 
 	// Ask for email and password
 	signinValues := getCredentials()
@@ -330,7 +330,7 @@ func main() {
 	fmt.Println("\tFailed: ", fails)
 }
 
-func fetchMedia(client *minicrawler.Crawler, media *Media) {
+func fetchMedia(client *Crawler, media *Media) {
 	media.Retries += 1
 	media.Status = "started"