func (controller *contactsController) get(rw http.ResponseWriter, req *http.Request) (int, error) {
var contacts database.Contacts
// Get contacts.
err := contacts.GetContacts(controller.DB)
if err != nil {
return http.StatusInternalServerError, err
}
isAuthenticated, user := getCurrentUser(rw, req, controller.authorizer)
vm := viewmodels.GetContactsViewModel(contacts, isAuthenticated, user, err)
return http.StatusOK, controller.getTemplate.Execute(rw, vm)
}
func (controller *sitesController) newPost(rw http.ResponseWriter, req *http.Request) (int, error) {
err := req.ParseForm()
if err != nil {
return http.StatusInternalServerError, err
}
decoder := schema.NewDecoder()
// Ignore unknown keys to prevent errors from the CSRF token.
decoder.IgnoreUnknownKeys(true)
formSite := new(viewmodels.SitesEditViewModel)
err = decoder.Decode(formSite, req.PostForm)
if err != nil {
return http.StatusInternalServerError, err
}
valErrors := validateSiteForm(formSite)
if len(valErrors) > 0 {
isAuthenticated, user := getCurrentUser(rw, req, controller.authorizer)
var contacts database.Contacts
err = contacts.GetContacts(controller.DB)
if err != nil {
return http.StatusInternalServerError, err
}
vm := viewmodels.NewSiteViewModel(formSite, contacts, isAuthenticated, user, valErrors)
vm.CsrfField = csrf.TemplateField(req)
return http.StatusOK, controller.newTemplate.Execute(rw, vm)
}
site := database.Site{}
viewmodels.MapSiteVMtoDB(formSite, &site)
err = site.CreateSite(controller.DB)
if err != nil {
return http.StatusInternalServerError, err
}
//Add any selected contacts
for _, contactSelID := range formSite.SelectedContacts {
err = site.AddContactToSite(controller.DB, contactSelID)
if err != nil {
return http.StatusInternalServerError, err
}
}
// Refresh the pinger with the changes.
err = controller.pinger.UpdateSiteSettings()
if err != nil {
return http.StatusInternalServerError, err
}
http.Redirect(rw, req, "/settings", http.StatusSeeOther)
return http.StatusSeeOther, nil
}
// TestCreateAndGetUnattachedContacts tests the creation of contacts not associated with a site.
func TestCreateAndGetUnattachedContacts(t *testing.T) {
db, err := database.InitializeTestDB("")
if err != nil {
t.Fatal("Failed to create database:", err)
}
defer db.Close()
// Create first contact
c := database.Contact{Name: "Joe Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551212",
SmsActive: false, EmailActive: false, SiteCount: 0}
err = c.CreateContact(db)
if err != nil {
t.Fatal("Failed to create new contact:", err)
}
// Create second contact
c2 := database.Contact{Name: "Jack Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551213",
SmsActive: false, EmailActive: false, SiteCount: 0}
err = c2.CreateContact(db)
if err != nil {
t.Fatal("Failed to create new contact:", err)
}
// Create third contact with name conflict.
c3 := database.Contact{Name: "Jack Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551213",
SmsActive: false, EmailActive: false}
err = c3.CreateContact(db)
if err == nil {
t.Fatal("Conflicting contact should throw error.")
}
var contacts database.Contacts
err = contacts.GetContacts(db)
if err != nil {
t.Fatal("Failed to get all contacts.", err)
}
// Verify the first contact was Loaded as the last in list by sort order
if !reflect.DeepEqual(c, contacts[1]) {
t.Fatal("New contact saved not equal to input:\n", contacts[1], c)
}
// Verify the second contact was Loaded as the first in list by sort order
if !reflect.DeepEqual(c2, contacts[0]) {
t.Fatal("New contact saved not equal to input:\n", contacts[0], c2)
}
}
func (controller *sitesController) newGet(rw http.ResponseWriter, req *http.Request) (int, error) {
isAuthenticated, user := getCurrentUser(rw, req, controller.authorizer)
siteNew := new(viewmodels.SitesEditViewModel)
siteNew.IsActive = true
// Get all of the contacts to display in the table.
var contacts database.Contacts
err := contacts.GetContacts(controller.DB)
if err != nil {
return http.StatusInternalServerError, err
}
// These are strings in the ViewModel.
siteNew.PingIntervalSeconds = "60"
siteNew.TimeoutSeconds = "15"
siteNew.SelectedContacts = []int64{}
vm := viewmodels.NewSiteViewModel(siteNew, contacts, isAuthenticated, user, make(map[string]string))
vm.CsrfField = csrf.TemplateField(req)
return http.StatusOK, controller.newTemplate.Execute(rw, vm)
}
func (controller *sitesController) editGet(rw http.ResponseWriter, req *http.Request) (int, error) {
vars := mux.Vars(req)
siteID, err := strconv.ParseInt(vars["siteID"], 10, 64)
if err != nil {
return http.StatusInternalServerError, err
}
// Get the site to edit
site := new(database.Site)
err = site.GetSite(controller.DB, siteID)
if err != nil {
return http.StatusInternalServerError, err
}
// Get all of the contacts to display in the table.
var contacts database.Contacts
err = contacts.GetContacts(controller.DB)
if err != nil {
return http.StatusInternalServerError, err
}
// Also get the site contacts to display in a table.
err = site.GetSiteContacts(controller.DB, siteID)
if err != nil {
return http.StatusInternalServerError, err
}
selectedContacts := []int64{}
for _, contact := range site.Contacts {
selectedContacts = append(selectedContacts, contact.ContactID)
}
isAuthenticated, user := getCurrentUser(rw, req, controller.authorizer)
siteEdit := new(viewmodels.SitesEditViewModel)
viewmodels.MapSiteDBtoVM(site, siteEdit)
siteEdit.SelectedContacts = selectedContacts
vm := viewmodels.EditSiteViewModel(siteEdit, contacts, isAuthenticated, user, make(map[string]string))
vm.CsrfField = csrf.TemplateField(req)
return http.StatusOK, controller.editTemplate.Execute(rw, vm)
}
// TestDeleteContacts tests creating a site and two contacts
// in the database and then deleting one of the contacts.
func TestDeleteContact(t *testing.T) {
db, err := database.InitializeTestDB("")
if err != nil {
t.Fatal("Failed to create database:", err)
}
defer db.Close()
// First create a site to associate with the contacts.
// Note: SiteID is ignored for create but is used in the test comparison
s := database.Site{SiteID: 1, Name: "Test", IsActive: true, URL: "http://www.google.com", PingIntervalSeconds: 60, TimeoutSeconds: 30, IsSiteUp: true}
err = s.CreateSite(db)
if err != nil {
t.Fatal("Failed to create new site:", err)
}
// Create first contact - ContactID is for referencing the contact get test
c := database.Contact{Name: "Joe Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551212",
SmsActive: false, EmailActive: false, ContactID: 1}
err = c.CreateContact(db)
if err != nil {
t.Fatal("Failed to create new contact:", err)
}
// Associate to the site ID
err = s.AddContactToSite(db, c.ContactID)
if err != nil {
t.Fatal("Failed to associate contact with site:", err)
}
// Create second contact
c2 := database.Contact{Name: "Jill Contact", EmailAddress: "*****@*****.**", SmsNumber: "5125551213",
SmsActive: false, EmailActive: false}
err = c2.CreateContact(db)
if err != nil {
t.Fatal("Failed to create new site:", err)
}
// Associate the contact to the site
err = s.AddContactToSite(db, c2.ContactID)
if err != nil {
t.Error("Failed to associate contact2 with site:", err)
}
err = s.GetSiteContacts(db, s.SiteID)
if err != nil {
t.Error("Failed to retrieve site contacts:", err)
}
if len(s.Contacts) != 2 {
t.Error("There should two contacts before deletion.")
}
// Delete the second contact
err = c2.DeleteContact(db)
if err != nil {
t.Fatal("Failed to delete contact 2:", err)
}
// Verify that it was deleted OK and not associated with the site, and
// that contact1 is still there.
err = s.GetSiteContacts(db, s.SiteID)
if err != nil {
t.Error("Failed to retrieve site contacts:", err)
}
if len(s.Contacts) != 1 {
t.Error("There should only be one contact for the site after deletion.")
}
if !reflect.DeepEqual(c, s.Contacts[0]) {
t.Error("Remaining contact not equal to input:\n", s.Contacts[0], c)
}
// Also verify that the contacts are correct.
var contacts database.Contacts
err = contacts.GetContacts(db)
if err != nil {
t.Fatal("Failed to get all contacts.", err)
}
if len(contacts) != 1 {
t.Error("There should only be one contact in the DB after deletion.")
}
if contacts[0].SiteCount != 1 {
t.Error("Site count should be 1.")
}
}
func (controller *sitesController) editPost(rw http.ResponseWriter, req *http.Request) (int, error) {
err := req.ParseForm()
if err != nil {
return http.StatusInternalServerError, err
}
decoder := schema.NewDecoder()
// Ignore unknown keys to prevent errors from the CSRF token.
decoder.IgnoreUnknownKeys(true)
formSite := new(viewmodels.SitesEditViewModel)
err = decoder.Decode(formSite, req.PostForm)
if err != nil {
return http.StatusInternalServerError, err
}
valErrors := validateSiteForm(formSite)
if len(valErrors) > 0 {
isAuthenticated, user := getCurrentUser(rw, req, controller.authorizer)
var contacts database.Contacts
err = contacts.GetContacts(controller.DB)
if err != nil {
return http.StatusInternalServerError, err
}
vm := viewmodels.EditSiteViewModel(formSite, contacts, isAuthenticated, user, valErrors)
vm.CsrfField = csrf.TemplateField(req)
return http.StatusOK, controller.editTemplate.Execute(rw, vm)
}
// Get the site to edit
site := new(database.Site)
err = site.GetSite(controller.DB, formSite.SiteID)
if err != nil {
return http.StatusInternalServerError, err
}
err = viewmodels.MapSiteVMtoDB(formSite, site)
if err != nil {
return http.StatusInternalServerError, err
}
err = site.UpdateSite(controller.DB)
if err != nil {
return http.StatusInternalServerError, err
}
//Loop selected ones first and if it's not already in the site then add it.
for _, contactSelID := range formSite.SelectedContacts {
if !int64InSlice(int64(contactSelID), formSite.SiteContacts) {
err = site.AddContactToSite(controller.DB, contactSelID)
if err != nil {
return http.StatusInternalServerError, err
}
}
}
// Loop existing site contacts and if it's not in the selected items then remove it.
for _, contactSiteID := range formSite.SiteContacts {
if !int64InSlice(int64(contactSiteID), formSite.SelectedContacts) {
err = site.RemoveContactFromSite(controller.DB, contactSiteID)
if err != nil {
return http.StatusInternalServerError, err
}
}
}
// Refresh the pinger with the changes.
err = controller.pinger.UpdateSiteSettings()
if err != nil {
return http.StatusInternalServerError, err
}
http.Redirect(rw, req, "/settings", http.StatusSeeOther)
return http.StatusSeeOther, nil
}