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)
+}

domain/organization/store.go

@@ -52,7 +52,7 @@ func (s Store) GetOrganization(ctx domain.RequestContext, id string) (org org.Or
         c_anonaccess AS allowanonymousaccess, c_authprovider AS authprovider,
         coalesce(c_authconfig,`+s.EmptyJSON()+`) AS authconfig,
 	    coalesce(c_sub,`+s.EmptyJSON()+`) AS subscription,
-        c_maxtags AS maxtags, c_created AS created, c_revised AS revised, c_theme AS theme
+        c_maxtags AS maxtags, c_theme AS theme, c_created AS created, c_revised AS revised
         FROM dmz_org
         WHERE c_refid=?`),
 		id)
@@ -176,3 +176,31 @@ func (s Store) CheckDomain(ctx domain.RequestContext, domain string) string {
 
 	return ""
 }
+
+// Logo fetchs stored image from store or NULL.
+func (s Store) Logo(ctx domain.RequestContext, domain string) (l []byte, err error) {
+	row := s.Runtime.Db.QueryRow(s.Bind("SELECT c_logo FROM dmz_org WHERE c_domain=? AND c_active=true"), domain)
+
+	err = row.Scan(&l)
+	if err == sql.ErrNoRows {
+		err = nil
+		return nil, nil
+	}
+	if err != nil {
+		return nil, err
+	}
+
+	return l, nil
+}
+
+// UploadLogo saves custom logo to the organization record.
+func (s Store) UploadLogo(ctx domain.RequestContext, logo []byte) (err error) {
+	_, err = ctx.Transaction.Exec(s.Bind("UPDATE dmz_org SET c_logo=?, c_revised=? WHERE c_refid=?"),
+		logo, time.Now().UTC(), ctx.OrgID)
+
+	if err != nil {
+		err = errors.Wrap(err, fmt.Sprintf("unable to save custom logo for org %s", ctx.OrgID))
+	}
+
+	return
+}