// Copyright 2016 Documize Inc. <legal@documize.com>. All rights reserved.
//
// This software (Documize Community Edition) is licensed under
// GNU AGPL v3 http://www.gnu.org/licenses/agpl-3.0.en.html
//
// You can operate outside the AGPL restrictions by purchasing
// Documize Enterprise Edition and obtaining a commercial license
// by contacting <sales@documize.com>.
//
// https://documize.com

package organization

import (
	"database/sql"
	"encoding/json"
	"io/ioutil"
	"net/http"

	"github.com/documize/community/core/env"
	"github.com/documize/community/core/request"
	"github.com/documize/community/core/response"
	"github.com/documize/community/core/streamutil"
	"github.com/documize/community/domain"
	"github.com/documize/community/model/org"
)

// Handler contains the runtime information such as logging and database.
type Handler struct {
	Runtime *env.Runtime
	Store   *domain.Store
}

// Get returns the requested organization.
func (h *Handler) Get(w http.ResponseWriter, r *http.Request) {
	method := "org.Get"
	ctx := domain.GetRequestContext(r)

	orgID := request.Param(r, "orgID")

	if orgID != ctx.OrgID {
		response.WriteForbiddenError(w)
		return
	}

	org, err := h.Store.Organization.GetOrganization(ctx, ctx.OrgID)
	if err != nil && err != sql.ErrNoRows {
		response.WriteServerError(w, method, err)
		h.Runtime.Log.Error(method, err)
		return
	}

	response.WriteJSON(w, org)
}

// Update saves organization amends.
func (h *Handler) Update(w http.ResponseWriter, r *http.Request) {
	method := "org.Update"
	ctx := domain.GetRequestContext(r)

	if !ctx.Administrator {
		response.WriteForbiddenError(w)
		return
	}

	defer streamutil.Close(r.Body)
	body, err := ioutil.ReadAll(r.Body)

	if err != nil {
		response.WriteServerError(w, method, err)
		h.Runtime.Log.Error(method, err)
		return
	}

	var org = org.Organization{}
	err = json.Unmarshal(body, &org)

	org.RefID = ctx.OrgID

	ctx.Transaction, err = h.Runtime.Db.Beginx()
	if err != nil {
		response.WriteServerError(w, method, err)
		h.Runtime.Log.Error(method, err)
		return
	}

	err = h.Store.Organization.UpdateOrganization(ctx, org)
	if err != nil {
		ctx.Transaction.Rollback()
		response.WriteServerError(w, method, err)
		h.Runtime.Log.Error(method, err)
		return
	}

	ctx.Transaction.Commit()

	response.WriteJSON(w, org)
}

// GetInstanceSetting returns the requested organization level setting.
func (h *Handler) GetInstanceSetting(w http.ResponseWriter, r *http.Request) {
	ctx := domain.GetRequestContext(r)

	orgID := request.Param(r, "orgID")
	if orgID != ctx.OrgID || !ctx.Administrator {
		response.WriteForbiddenError(w)
		return
	}

	key := request.Query(r, "key")
	setting, _ := h.Store.Setting.GetUser(orgID, "", key, "")
	if len(setting) == 0 {
		setting = "{}"
	}

	response.WriteJSON(w, setting)
}

// SaveInstanceSetting saves org level setting.
func (h *Handler) SaveInstanceSetting(w http.ResponseWriter, r *http.Request) {
	method := "org.SaveInstanceSetting"
	ctx := domain.GetRequestContext(r)

	orgID := request.Param(r, "orgID")
	if orgID != ctx.OrgID || !ctx.Administrator {
		response.WriteForbiddenError(w)
		return
	}

	key := request.Query(r, "key")

	defer streamutil.Close(r.Body)
	body, err := ioutil.ReadAll(r.Body)
	if err != nil {
		response.WriteServerError(w, method, err)
		h.Runtime.Log.Error(method, err)
		return
	}

	config := string(body)
	h.Store.Setting.SetUser(orgID, "", key, config)

	response.WriteEmpty(w)
}

// GetGlobalSetting returns the requested organization level setting.
func (h *Handler) GetGlobalSetting(w http.ResponseWriter, r *http.Request) {
	ctx := domain.GetRequestContext(r)

	if !ctx.GlobalAdmin {
		response.WriteForbiddenError(w)
		return
	}

	key := request.Query(r, "key")
	setting, _ := h.Store.Setting.Get(key, "")

	response.WriteJSON(w, setting)
}

// SaveGlobalSetting saves org level setting.
func (h *Handler) SaveGlobalSetting(w http.ResponseWriter, r *http.Request) {
	method := "org.SaveGlobalSetting"
	ctx := domain.GetRequestContext(r)

	if !ctx.GlobalAdmin {
		response.WriteForbiddenError(w)
		return
	}

	key := request.Query(r, "key")

	defer streamutil.Close(r.Body)
	body, err := ioutil.ReadAll(r.Body)
	if err != nil {
		response.WriteServerError(w, method, err)
		h.Runtime.Log.Error(method, err)
		return
	}

	config := string(body)
	h.Store.Setting.Set(key, config)

	response.WriteEmpty(w)
}