Enable custom logo upload and rendering

domain/organization/endpoint.go

@@ -12,8 +12,11 @@
 package organization
 
 import (
+	"bytes"
 	"database/sql"
 	"encoding/json"
+	"github.com/documize/community/model/audit"
+	"io"
 	"io/ioutil"
 	"net/http"
 
@@ -99,3 +102,50 @@ func (h *Handler) Update(w http.ResponseWriter, r *http.Request) {
 
 	response.WriteJSON(w, org)
 }
+
+// UploadLogo stores log for organization.
+func (h *Handler) UploadLogo(w http.ResponseWriter, r *http.Request) {
+	method := "organization.UploadLogo"
+	ctx := domain.GetRequestContext(r)
+
+	if !ctx.Administrator {
+		response.WriteForbiddenError(w)
+		return
+	}
+
+	// We use default logo if body is empty.
+	logo := []byte{}
+
+	filedata, _, err := r.FormFile("attachment")
+	if err == nil {
+		b := new(bytes.Buffer)
+		_, err = io.Copy(b, filedata)
+		if err != nil {
+			response.WriteServerError(w, method, err)
+			h.Runtime.Log.Error(method, err)
+		} else {
+			logo = b.Bytes()
+		}
+	}
+
+	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.UploadLogo(ctx, logo)
+	if err != nil {
+		ctx.Transaction.Rollback()
+		response.WriteServerError(w, method, err)
+		h.Runtime.Log.Error(method, err)
+		return
+	}
+
+	ctx.Transaction.Commit()
+
+	h.Store.Audit.Record(ctx, audit.EventTypeOrganizationLogo)
+
+	response.WriteEmpty(w)
+}