Refactor to use a Twitter Struct

scraper.go

@@ -3,7 +3,6 @@ package main
 import (
 	"WorstCaptcha"
 	"fmt"
-	"github.com/kurrik/twittergo"
 	"gopkg.in/gcfg.v1"
 	"os"
 	"strings"
@@ -11,7 +10,7 @@ import (
 
 func main() {
 	var config WorstCaptcha.Config
-	var twitter *twittergo.Client
+	var twitter WorstCaptcha.Twitter
 
 	// Parse config file
 	if err := gcfg.ReadFileInto(&config, "worstcaptcha.gcfg"); err != nil {
@@ -23,7 +22,7 @@ func main() {
 	twitter = WorstCaptcha.NewTwitter(config)
 
 	// Do twitter search and loop
-	for _, tweet := range WorstCaptcha.DoSearch(twitter) {
+	for _, tweet := range twitter.DoSearch() {
 		fmt.Printf("\n=================\n")
 
 		// Type assert entities to be a map

src/WorstCaptcha/Twitter.go

@@ -9,7 +9,7 @@ import (
 	"os"
 )
 
-func NewTwitter(config Config) *twittergo.Client {
+func NewTwitter(config Config) Twitter {
 	twitterConfig := &oauth1a.ClientConfig{
 		ConsumerKey:    config.Twitter.ConsumerKey,
 		ConsumerSecret: config.Twitter.ConsumerSecret,
@@ -20,10 +20,14 @@ func NewTwitter(config Config) *twittergo.Client {
 		config.Twitter.AccessTokenSecret,
 	)
 
-	return twittergo.NewClient(twitterConfig, user)
+	return Twitter{twittergo.NewClient(twitterConfig, user)}
 }
 
-func DoSearch(client *twittergo.Client) []twittergo.Tweet {
+type Twitter struct {
+	client *twittergo.Client
+}
+
+func (t *Twitter) DoSearch() []twittergo.Tweet {
 	var (
 		err     error
 		req     *http.Request
@@ -46,7 +50,7 @@ func DoSearch(client *twittergo.Client) []twittergo.Tweet {
 	}
 
 	// Sign and send request
-	if resp, err = client.SendRequest(req); err != nil {
+	if resp, err = t.client.SendRequest(req); err != nil {
 		fmt.Printf("Could not send request: %v\n", err)
 		os.Exit(1)
 	}