PlantUML integration

domain/section/plantuml/plantuml.go

@@ -0,0 +1,116 @@
+// 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 plantuml
+
+import (
+	"bytes"
+	"crypto/tls"
+	"encoding/base64"
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+
+	"github.com/documize/community/core/env"
+	"github.com/documize/community/domain"
+	"github.com/documize/community/domain/section/provider"
+)
+
+// Provider represents Mermaid Diagram
+type Provider struct {
+	Runtime *env.Runtime
+	Store   *domain.Store
+}
+
+// Meta describes us
+func (*Provider) Meta() provider.TypeMeta {
+	section := provider.TypeMeta{}
+
+	section.ID = "f1067a60-45e5-40b5-89f6-aa3b03dd7f35"
+	section.Title = "PlantUML Diagram"
+	section.Description = "Diagrams generated from text"
+	section.ContentType = "plantuml"
+	section.PageType = "tab"
+	section.Order = 9990
+
+	return section
+}
+
+// Command stub.
+func (p *Provider) Command(ctx *provider.Context, w http.ResponseWriter, r *http.Request) {
+	query := r.URL.Query()
+	method := query.Get("method")
+
+	if len(method) == 0 {
+		provider.WriteMessage(w, "plantuml", "missing method name")
+		return
+	}
+
+	switch method {
+	case "preview":
+		var payload struct {
+			Data string `json:"data"`
+		}
+
+		defer r.Body.Close()
+		body, err := ioutil.ReadAll(r.Body)
+		if err != nil {
+			provider.WriteMessage(w, "plantuml", "Bad payload")
+			return
+		}
+
+		err = json.Unmarshal(body, &payload)
+		if err != nil {
+			provider.WriteMessage(w, "plantuml", "Cannot unmarshal")
+			return
+		}
+
+		diagram := p.generateDiagram(ctx, payload.Data)
+		payload.Data = diagram
+		provider.WriteJSON(w, payload)
+		return
+	}
+
+	provider.WriteEmpty(w)
+}
+
+// Render returns data as-is (HTML).
+func (p *Provider) Render(ctx *provider.Context, config, data string) string {
+	return p.generateDiagram(ctx, data)
+}
+
+// Refresh just sends back data as-is.
+func (*Provider) Refresh(ctx *provider.Context, config, data string) string {
+	return data
+}
+
+func (p *Provider) generateDiagram(ctx *provider.Context, data string) string {
+	org, _ := p.Store.Organization.GetOrganization(ctx.Request, ctx.OrgID)
+
+	var transport = &http.Transport{
+		TLSClientConfig: &tls.Config{
+			InsecureSkipVerify: true, // TODO should be glick.InsecureSkipVerifyTLS (from -insecure flag) but get error: x509: certificate signed by unknown authority
+		}}
+	client := &http.Client{Transport: transport}
+
+	resp, _ := client.Post(org.ConversionEndpoint+"/api/plantuml", "application/text", bytes.NewReader([]byte(data)))
+	defer func() {
+		if e := resp.Body.Close(); e != nil {
+			fmt.Println("resp.Body.Close error: " + e.Error())
+		}
+	}()
+
+	png, _ := ioutil.ReadAll(resp.Body)
+	pngEncoded := base64.StdEncoding.EncodeToString(png)
+
+	return string(fmt.Sprintf("data:image/png;base64, %s", pngEncoded))
+}