From cf926134ef03d4a1ccbb44178660181ff40b8fd5 Mon Sep 17 00:00:00 2001
From: utkarshcmu <utkarsh.cmu@gmail.com>
Date: Sat, 5 Mar 2016 13:15:49 -0800
Subject: [PATCH] Added savePreferencesAPI

---
 pkg/api/api.go                |  1 +
 pkg/api/user.go               | 21 ++++++++++++++++++++-
 pkg/models/preferences.go     | 28 ++++++++++++++++++++++++++++
 pkg/services/sqlstore/user.go | 25 +++++++++++++++++++++++++
 4 files changed, 74 insertions(+), 1 deletion(-)
 create mode 100644 pkg/models/preferences.go

diff --git a/pkg/api/api.go b/pkg/api/api.go
index ed029a5171a..3f805cd9c98 100644
--- a/pkg/api/api.go
+++ b/pkg/api/api.go
@@ -96,6 +96,7 @@ func Register(r *macaron.Macaron) {
 			r.Delete("/stars/dashboard/:id", wrap(UnstarDashboard))
 			r.Put("/password", bind(m.ChangeUserPasswordCommand{}), wrap(ChangeUserPassword))
 			r.Get("/quotas", wrap(GetUserQuotas))
+      r.Put("/prefs", bind(m.SavePreferenceCommand{}), wrap(SaveUserPreferences))
 		})
 
 		// users (admin permission required)
diff --git a/pkg/api/user.go b/pkg/api/user.go
index 5af243eeb22..0f972431426 100644
--- a/pkg/api/user.go
+++ b/pkg/api/user.go
@@ -5,6 +5,7 @@ import (
 	"github.com/grafana/grafana/pkg/middleware"
 	m "github.com/grafana/grafana/pkg/models"
 	"github.com/grafana/grafana/pkg/util"
+  "github.com/grafana/grafana/pkg/log"
 )
 
 // GET /api/user  (current authenticated user)
@@ -110,7 +111,10 @@ func UserSetUsingOrg(c *middleware.Context) Response {
 }
 
 func ChangeUserPassword(c *middleware.Context, cmd m.ChangeUserPasswordCommand) Response {
-	userQuery := m.GetUserByIdQuery{Id: c.UserId}
+
+  log.Info("%v", cmd)
+	
+  userQuery := m.GetUserByIdQuery{Id: c.UserId}
 
 	if err := bus.Dispatch(&userQuery); err != nil {
 		return ApiError(500, "Could not read user from database", err)
@@ -144,3 +148,18 @@ func SearchUsers(c *middleware.Context) Response {
 
 	return Json(200, query.Result)
 }
+
+func SaveUserPreferences(c *middleware.Context, cmd m.SavePreferenceCommand) Response {
+
+  log.Info("%v", cmd.PrefData)
+
+  cmd.PrefId = c.UserId
+  cmd.PrefType = `user`
+
+  if err := bus.Dispatch(&cmd); err != nil {
+    return ApiError(500, "Failed to saved user preferences", err)
+  }
+
+  return ApiSuccess("User preferences saved")
+
+}
diff --git a/pkg/models/preferences.go b/pkg/models/preferences.go
new file mode 100644
index 00000000000..cdaab0f6af8
--- /dev/null
+++ b/pkg/models/preferences.go
@@ -0,0 +1,28 @@
+package models
+
+import (
+	"errors"
+)
+
+// Typed errors
+var (
+	ErrPreferenceNotFound = errors.New("Preference not found")
+)
+
+type Preference struct {
+	Id            int64
+	PrefId        int64
+	PrefType      string
+	PrefData      map[string]interface{}
+}
+
+// ---------------------
+// COMMANDS
+
+type SavePreferenceCommand struct {
+	
+  PrefData map[string]interface{} `json:"prefData"`
+  PrefId   int64                  `json:"-"`
+	PrefType string                 `json:"-"`
+
+}
diff --git a/pkg/services/sqlstore/user.go b/pkg/services/sqlstore/user.go
index 96b8c24b8fc..a5c992d10b6 100644
--- a/pkg/services/sqlstore/user.go
+++ b/pkg/services/sqlstore/user.go
@@ -11,6 +11,7 @@ import (
 	m "github.com/grafana/grafana/pkg/models"
 	"github.com/grafana/grafana/pkg/setting"
 	"github.com/grafana/grafana/pkg/util"
+  "github.com/grafana/grafana/pkg/log"
 )
 
 func init() {
@@ -27,6 +28,7 @@ func init() {
 	bus.AddHandler("sql", DeleteUser)
 	bus.AddHandler("sql", SetUsingOrg)
 	bus.AddHandler("sql", UpdateUserPermissions)
+  bus.AddHandler("sql", SaveUserPreferences)
 }
 
 func getOrgIdForNewUser(cmd *m.CreateUserCommand, sess *session) (int64, error) {
@@ -346,3 +348,26 @@ func UpdateUserPermissions(cmd *m.UpdateUserPermissionsCommand) error {
 		return err
 	})
 }
+
+func SaveUserPreferences(cmd *m.SavePreferenceCommand) error {
+  return inTransaction2(func(sess *session) error {
+
+    log.Info("%v", cmd)
+
+    pref := m.Preference{
+      PrefId: cmd.PrefId,
+      PrefType: cmd.PrefType,
+      PrefData: cmd.PrefData,
+    }
+
+    sess.Table("preferences").Where("pref_id", pref.PrefId).And("pref_type", pref.PrefType)
+
+    if _, err := sess.Update(&pref); err != nil {
+      return err
+    }
+
+    log.Info("%v", pref)
+
+    return nil
+  })  
+}