From 7c50f4dbb4421e61a5a8453684cb8f41348e3c68 Mon Sep 17 00:00:00 2001
From: Jordan <jmwright798@gmail.com>
Date: Thu, 27 Mar 2014 23:31:51 -0500
Subject: [PATCH] Initial commit of worker.go (Still work to do) Moved
 constants to models.go Changed Campaign.Template to be an actual template
 (will need to adjust all the methods to handle it) Added UpdateCampaignStatus
 function to update a campaign status

---
 controllers/api.go | 16 +++++++++-------
 models/campaign.go |  7 +++++--
 models/models.go   |  9 ++++++++-
 worker/worker.go   | 45 +++++++++++++++++++++++++++++++++++++++++++++
 4 files changed, 67 insertions(+), 10 deletions(-)
 create mode 100644 worker/worker.go

diff --git a/controllers/api.go b/controllers/api.go
index b535ff17..1b52b275 100644
--- a/controllers/api.go
+++ b/controllers/api.go
@@ -12,14 +12,15 @@ import (
 	"github.com/gorilla/mux"
 	"github.com/jordan-wright/gophish/auth"
 	"github.com/jordan-wright/gophish/models"
+	"github.com/jordan-wright/gophish/worker"
 )
 
-const (
-	IN_PROGRESS string = "In progress"
-	WAITING     string = "Waiting"
-	COMPLETE    string = "Completed"
-	ERROR       string = "Error"
-)
+var Worker *worker.Worker
+
+func init() {
+	Worker = worker.New()
+	go Worker.Start()
+}
 
 // API (/api) provides access to api documentation
 func API(w http.ResponseWriter, r *http.Request) {
@@ -79,12 +80,13 @@ func API_Campaigns(w http.ResponseWriter, r *http.Request) {
 		// Fill in the details
 		c.CreatedDate = time.Now()
 		c.CompletedDate = time.Time{}
-		c.Status = IN_PROGRESS
+		c.Status = models.QUEUED
 		c.UserId = ctx.Get(r, "user_id").(int64)
 		err = models.PostCampaign(&c, ctx.Get(r, "user_id").(int64))
 		if checkError(err, w, "Cannot insert campaign into database", http.StatusInternalServerError) {
 			return
 		}
+		Worker.Queue <- &c
 		cj, err := json.MarshalIndent(c, "", "  ")
 		if checkError(err, w, "Error creating JSON response", http.StatusInternalServerError) {
 			return
diff --git a/models/campaign.go b/models/campaign.go
index 46d91887..99a262a4 100644
--- a/models/campaign.go
+++ b/models/campaign.go
@@ -13,7 +13,7 @@ type Campaign struct {
 	Name          string    `json:"name" sql:"not null"`
 	CreatedDate   time.Time `json:"created_date"`
 	CompletedDate time.Time `json:"completed_date"`
-	Template      string    `json:"template"` //This may change
+	Template      Template  `json:"template"` //This may change
 	Status        string    `json:"status"`
 	Results       []Result  `json:"results,omitempty"`
 	Groups        []Group   `json:"groups,omitempty"`
@@ -39,7 +39,6 @@ func GetCampaigns(uid int64) ([]Campaign, error) {
 			fmt.Println(err)
 		}
 	}
-	fmt.Printf("%v", cs)
 	return cs, err
 }
 
@@ -90,6 +89,10 @@ func PostCampaign(c *Campaign, uid int64) error {
 	return nil
 }
 
+func UpdateCampaignStatus(c *Campaign, s string) error {
+	return db.Debug().Model(c).UpdateColumn("status", s).Error
+}
+
 //DeleteCampaign deletes the specified campaign
 func DeleteCampaign(id int64) error {
 	// Delete all the campaign results
diff --git a/models/models.go b/models/models.go
index 87cbc372..3b3aed0a 100644
--- a/models/models.go
+++ b/models/models.go
@@ -17,6 +17,13 @@ var err error
 var ErrUsernameTaken = errors.New("username already taken")
 var Logger = log.New(os.Stdout, " ", log.Ldate|log.Ltime|log.Lshortfile)
 
+const (
+	IN_PROGRESS string = "In progress"
+	QUEUED      string = "Queued"
+	COMPLETE    string = "Completed"
+	ERROR       string = "Error"
+)
+
 // Flash is used to hold flash information for use in templates.
 type Flash struct {
 	Type    string
@@ -27,7 +34,7 @@ type Flash struct {
 // It also populates the Gophish Config object
 func Setup() error {
 	db, err = gorm.Open("sqlite3", config.Conf.DBPath)
-	db.LogMode(true)
+	db.LogMode(false)
 	db.SetLogger(Logger)
 	if err != nil {
 		Logger.Println(err)
diff --git a/worker/worker.go b/worker/worker.go
new file mode 100644
index 00000000..64c5b177
--- /dev/null
+++ b/worker/worker.go
@@ -0,0 +1,45 @@
+package worker
+
+import (
+	"log"
+	"os"
+
+	"github.com/jordan-wright/gophish/models"
+)
+
+var Logger = log.New(os.Stdout, " ", log.Ldate|log.Ltime|log.Lshortfile)
+
+type Worker struct {
+	Queue chan *models.Campaign
+}
+
+func New() *Worker {
+	return &Worker{
+		Queue: make(chan *models.Campaign),
+	}
+}
+
+// Start launches the worker to monitor the database for any jobs.
+// If a job is found, it launches the job
+func (w *Worker) Start() {
+	Logger.Println("Background Worker Started Successfully - Waiting for Campaigns")
+	for {
+		processCampaign(<-w.Queue)
+	}
+}
+
+func processCampaign(c *models.Campaign) {
+	Logger.Printf("Worker received: %s", c.Name)
+	err := models.UpdateCampaignStatus(c, models.IN_PROGRESS)
+	if err != nil {
+		Logger.Println(err)
+	}
+	for _, t := range c.Results {
+		Logger.Println("Creating email using template")
+		/*e := email.Email{
+			Text: []byte(c.Template.Text),
+			HTML: []byte(c.Template.Html),
+		}*/
+		Logger.Printf("Sending Email to %s\n", t.Email)
+	}
+}