mattermost/api/preference.go

// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved.
// See License.txt for license information.

package api

import (
	l4g "code.google.com/p/log4go"
	"encoding/json"
	"github.com/gorilla/mux"
	"github.com/mattermost/platform/model"
	"net/http"
)

func InitPreference(r *mux.Router) {
	l4g.Debug("Initializing preference api routes")

	sr := r.PathPrefix("/preferences").Subrouter()
	sr.Handle("/set", ApiAppHandler(setPreferences)).Methods("POST")
	sr.Handle("/{category:[A-Za-z0-9_]+}/{name:[A-Za-z0-9_]+}", ApiAppHandler(getPreferencesByName)).Methods("GET")
}

func setPreferences(c *Context, w http.ResponseWriter, r *http.Request) {
	var preferences []*model.Preference

	decoder := json.NewDecoder(r.Body)
	if err := decoder.Decode(&preferences); err != nil {
		c.Err = model.NewAppError("setPreferences", "Unable to decode preferences from request", err.Error())
		c.Err.StatusCode = http.StatusBadRequest
		return
	}

	for _, preference := range preferences {
		if c.Session.UserId != preference.UserId {
			c.Err = model.NewAppError("setPreferences", "Unable to set preferences for other user", "session.user_id="+c.Session.UserId+", preference.user_id="+preference.UserId)
			c.Err.StatusCode = http.StatusUnauthorized
			return
		}
	}

	if result := <-Srv.Store.Preference().SaveOrUpdate(preferences...); result.Err != nil {
		c.Err = result.Err
		return
	}

	w.Write([]byte("true"))
}

func getPreferencesByName(c *Context, w http.ResponseWriter, r *http.Request) {
	params := mux.Vars(r)
	category := params["category"]
	name := params["name"]

	if result := <-Srv.Store.Preference().GetByName(c.Session.UserId, category, name); result.Err != nil {
		c.Err = result.Err
		return
	} else {
		data := result.Data.([]*model.Preference)

		if len(data) == 0 {
			if category == model.PREFERENCE_CATEGORY_DIRECT_CHANNELS && name == model.PREFERENCE_NAME_SHOW {
				// add direct channels for a user that existed before preferences were added
				data = addDirectChannels(c.Session.UserId, c.Session.TeamId)
			}
		}

		w.Write([]byte(model.PreferenceListToJson(data)))
	}
}

func addDirectChannels(userId, teamId string) []*model.Preference {
	var profiles map[string]*model.User
	if result := <-Srv.Store.User().GetProfiles(teamId); result.Err != nil {
		l4g.Error("Failed to add direct channel preferences for user user_id=%s, team_id=%s, err=%v", userId, teamId, result.Err.Error())
		return []*model.Preference{}
	} else {
		profiles = result.Data.(map[string]*model.User)
	}

	var preferences []*model.Preference

	for id := range profiles {
		if id == userId {
			continue
		}

		profile := profiles[id]

		preference := &model.Preference{
			UserId:   userId,
			Category: model.PREFERENCE_CATEGORY_DIRECT_CHANNELS,
			Name:     model.PREFERENCE_NAME_SHOW,
			AltId:    profile.Id,
			Value:    "true",
		}

		if result := <-Srv.Store.Preference().Save(preference); result.Err != nil {
			l4g.Error("Failed to add direct channel preferences for user user_id=%s, alt_id=%s, team_id=%s, err=%v", userId, profile.Id, teamId, result.Err.Error())
			continue
		}

		preferences = append(preferences, preference)

		if len(preferences) >= 10 {
			break
		}
	}

	return preferences
}