energylink/pkg/models/dashboards.go

package models

import (
	"encoding/json"
	"io"
	"regexp"
	"strings"
	"time"
)

var (
	GetDashboard    func(slug string, accountId int) (*Dashboard, error)
	SaveDashboard   func(dash *Dashboard) error
	DeleteDashboard func(slug string, accountId int) error
	SearchQuery     func(query string, acccountId int) ([]*SearchResult, error)
)

type Dashboard struct {
	Id                   string `gorethink:"id,omitempty"`
	Slug                 string
	AccountId            int
	LastModifiedByUserId string
	LastModifiedByDate   time.Time
	CreatedDate          time.Time

	Title string
	Tags  []string
	Data  map[string]interface{}
}

type SearchResult struct {
	Id    string `json:"id"`
	Title string `json:"title"`
	Slug  string `json:"slug"`
}

func NewDashboard(title string) *Dashboard {
	dash := &Dashboard{}
	dash.Id = ""
	dash.LastModifiedByDate = time.Now()
	dash.CreatedDate = time.Now()
	dash.LastModifiedByUserId = "123"
	dash.Data = make(map[string]interface{})
	dash.Data["title"] = title
	dash.Title = title
	dash.UpdateSlug()

	return dash
}

func NewFromJson(reader io.Reader) (*Dashboard, error) {
	dash := NewDashboard("temp")
	jsonParser := json.NewDecoder(reader)

	if err := jsonParser.Decode(&dash.Data); err != nil {
		return nil, err
	}

	return dash, nil
}

func (dash *Dashboard) GetString(prop string) string {
	return dash.Data[prop].(string)
}

func (dash *Dashboard) UpdateSlug() {
	title := strings.ToLower(dash.Data["title"].(string))
	re := regexp.MustCompile("[^\\w ]+")
	re2 := regexp.MustCompile("\\s")
	dash.Slug = re2.ReplaceAllString(re.ReplaceAllString(title, ""), "-")
}