182 lines
6.9 KiB
Go
182 lines
6.9 KiB
Go
package d2svg
|
|
|
|
import (
|
|
"fmt"
|
|
"html"
|
|
"io"
|
|
|
|
"oss.terrastruct.com/d2/d2target"
|
|
"oss.terrastruct.com/d2/d2themes"
|
|
"oss.terrastruct.com/d2/lib/geo"
|
|
"oss.terrastruct.com/d2/lib/label"
|
|
"oss.terrastruct.com/d2/lib/svg"
|
|
"oss.terrastruct.com/util-go/go2"
|
|
)
|
|
|
|
// this func helps define a clipPath for shape class and sql_table to draw border-radius
|
|
func clipPathForBorderRadius(diagramHash string, shape d2target.Shape) string {
|
|
box := geo.NewBox(
|
|
geo.NewPoint(float64(shape.Pos.X), float64(shape.Pos.Y)),
|
|
float64(shape.Width),
|
|
float64(shape.Height),
|
|
)
|
|
topX, topY := box.TopLeft.X+box.Width, box.TopLeft.Y
|
|
|
|
out := fmt.Sprintf(`<clipPath id="%v-%v">`, diagramHash, shape.ID)
|
|
out += fmt.Sprintf(`<path d="M %f %f L %f %f S %f %f %f %f `, box.TopLeft.X, box.TopLeft.Y+float64(shape.BorderRadius), box.TopLeft.X, box.TopLeft.Y+float64(shape.BorderRadius), box.TopLeft.X, box.TopLeft.Y, box.TopLeft.X+float64(shape.BorderRadius), box.TopLeft.Y)
|
|
out += fmt.Sprintf(`L %f %f L %f %f `, box.TopLeft.X+box.Width-float64(shape.BorderRadius), box.TopLeft.Y, topX-float64(shape.BorderRadius), topY)
|
|
|
|
out += fmt.Sprintf(`S %f %f %f %f `, topX, topY, topX, topY+float64(shape.BorderRadius))
|
|
out += fmt.Sprintf(`L %f %f `, topX, topY+box.Height-float64(shape.BorderRadius))
|
|
|
|
if len(shape.Columns) != 0 {
|
|
out += fmt.Sprintf(`L %f %f L %f %f`, topX, topY+box.Height, box.TopLeft.X, box.TopLeft.Y+box.Height)
|
|
} else {
|
|
out += fmt.Sprintf(`S %f % f %f %f `, topX, topY+box.Height, topX-float64(shape.BorderRadius), topY+box.Height)
|
|
out += fmt.Sprintf(`L %f %f `, box.TopLeft.X+float64(shape.BorderRadius), box.TopLeft.Y+box.Height)
|
|
out += fmt.Sprintf(`S %f %f %f %f`, box.TopLeft.X, box.TopLeft.Y+box.Height, box.TopLeft.X, box.TopLeft.Y+box.Height-float64(shape.BorderRadius))
|
|
out += fmt.Sprintf(`L %f %f`, box.TopLeft.X, box.TopLeft.Y+float64(shape.BorderRadius))
|
|
}
|
|
out += fmt.Sprintf(`Z %f %f" `, box.TopLeft.X, box.TopLeft.Y)
|
|
return out + `fill="none" /> </clipPath>`
|
|
}
|
|
|
|
func tableHeader(diagramHash string, shape d2target.Shape, box *geo.Box, text string, textWidth, textHeight, fontSize float64, inlineTheme *d2themes.Theme) string {
|
|
rectEl := d2themes.NewThemableElement("rect", inlineTheme)
|
|
rectEl.X, rectEl.Y = box.TopLeft.X, box.TopLeft.Y
|
|
rectEl.Width, rectEl.Height = box.Width, box.Height
|
|
rectEl.Fill = shape.Fill
|
|
rectEl.FillPattern = shape.FillPattern
|
|
rectEl.ClassName = "class_header"
|
|
if shape.BorderRadius != 0 {
|
|
rectEl.ClipPath = fmt.Sprintf("%v-%v", diagramHash, shape.ID)
|
|
}
|
|
str := rectEl.Render()
|
|
|
|
if text != "" {
|
|
tl := label.InsideMiddleLeft.GetPointOnBox(
|
|
box,
|
|
float64(d2target.HeaderPadding),
|
|
float64(shape.Width),
|
|
textHeight,
|
|
)
|
|
|
|
textEl := d2themes.NewThemableElement("text", inlineTheme)
|
|
textEl.X = tl.X
|
|
textEl.Y = tl.Y + textHeight*3/4
|
|
textEl.Fill = shape.GetFontColor()
|
|
textEl.ClassName = "text"
|
|
textEl.Style = fmt.Sprintf("text-anchor:%s;font-size:%vpx",
|
|
"start", 4+fontSize,
|
|
)
|
|
textEl.Content = svg.EscapeText(text)
|
|
str += textEl.Render()
|
|
}
|
|
return str
|
|
}
|
|
|
|
func tableRow(shape d2target.Shape, box *geo.Box, nameText, typeText, constraintText string, fontSize, longestNameWidth, longestTypeWidth float64, inlineTheme *d2themes.Theme) string {
|
|
// Row is made up of name, type, and constraint
|
|
// e.g. | diagram int FK |
|
|
nameTL := label.InsideMiddleLeft.GetPointOnBox(
|
|
box,
|
|
d2target.NamePadding,
|
|
0,
|
|
fontSize,
|
|
)
|
|
|
|
textEl := d2themes.NewThemableElement("text", inlineTheme)
|
|
textEl.X = nameTL.X
|
|
textEl.Y = nameTL.Y + fontSize*3/4
|
|
textEl.Fill = shape.PrimaryAccentColor
|
|
textEl.ClassName = "text"
|
|
textEl.Style = fmt.Sprintf("text-anchor:%s;font-size:%vpx", "start", fontSize)
|
|
textEl.Content = svg.EscapeText(nameText)
|
|
out := textEl.Render()
|
|
|
|
textEl.X += longestNameWidth + d2target.TypePadding
|
|
textEl.Fill = shape.NeutralAccentColor
|
|
textEl.Content = svg.EscapeText(typeText)
|
|
out += textEl.Render()
|
|
|
|
textEl.X = box.TopLeft.X + (box.Width - d2target.NamePadding)
|
|
textEl.Fill = shape.SecondaryAccentColor
|
|
textEl.Style = fmt.Sprintf("text-anchor:%s;font-size:%vpx", "end", fontSize)
|
|
textEl.Content = constraintText
|
|
out += textEl.Render()
|
|
|
|
return out
|
|
}
|
|
|
|
func drawTable(writer io.Writer, diagramHash string, targetShape d2target.Shape, inlineTheme *d2themes.Theme) {
|
|
rectEl := d2themes.NewThemableElement("rect", inlineTheme)
|
|
rectEl.X = float64(targetShape.Pos.X)
|
|
rectEl.Y = float64(targetShape.Pos.Y)
|
|
rectEl.Width = float64(targetShape.Width)
|
|
rectEl.Height = float64(targetShape.Height)
|
|
rectEl.Fill, rectEl.Stroke = d2themes.ShapeTheme(targetShape)
|
|
rectEl.FillPattern = targetShape.FillPattern
|
|
rectEl.ClassName = "shape"
|
|
rectEl.Style = targetShape.CSSStyle()
|
|
if targetShape.BorderRadius != 0 {
|
|
rectEl.Rx = float64(targetShape.BorderRadius)
|
|
rectEl.Ry = float64(targetShape.BorderRadius)
|
|
}
|
|
fmt.Fprint(writer, rectEl.Render())
|
|
|
|
box := geo.NewBox(
|
|
geo.NewPoint(float64(targetShape.Pos.X), float64(targetShape.Pos.Y)),
|
|
float64(targetShape.Width),
|
|
float64(targetShape.Height),
|
|
)
|
|
rowHeight := box.Height / float64(1+len(targetShape.SQLTable.Columns))
|
|
headerBox := geo.NewBox(box.TopLeft, box.Width, rowHeight)
|
|
|
|
fmt.Fprint(writer,
|
|
tableHeader(diagramHash, targetShape, headerBox, targetShape.Label,
|
|
float64(targetShape.LabelWidth), float64(targetShape.LabelHeight), float64(targetShape.FontSize), inlineTheme),
|
|
)
|
|
|
|
var longestNameWidth int
|
|
var longestTypeWidth int
|
|
for _, f := range targetShape.Columns {
|
|
longestNameWidth = go2.Max(longestNameWidth, f.Name.LabelWidth)
|
|
longestTypeWidth = go2.Max(longestTypeWidth, f.Type.LabelWidth)
|
|
}
|
|
|
|
rowBox := geo.NewBox(box.TopLeft.Copy(), box.Width, rowHeight)
|
|
rowBox.TopLeft.Y += headerBox.Height
|
|
for idx, f := range targetShape.Columns {
|
|
fmt.Fprint(writer,
|
|
tableRow(targetShape, rowBox, f.Name.Label, f.Type.Label, f.ConstraintAbbr(), float64(targetShape.FontSize), float64(longestNameWidth), float64(longestTypeWidth), inlineTheme),
|
|
)
|
|
rowBox.TopLeft.Y += rowHeight
|
|
|
|
lineEl := d2themes.NewThemableElement("line", inlineTheme)
|
|
if idx == len(targetShape.Columns)-1 && targetShape.BorderRadius != 0 {
|
|
lineEl.X1, lineEl.Y1 = rowBox.TopLeft.X+float64(targetShape.BorderRadius), rowBox.TopLeft.Y
|
|
lineEl.X2, lineEl.Y2 = rowBox.TopLeft.X+rowBox.Width-float64(targetShape.BorderRadius), rowBox.TopLeft.Y
|
|
} else {
|
|
lineEl.X1, lineEl.Y1 = rowBox.TopLeft.X, rowBox.TopLeft.Y
|
|
lineEl.X2, lineEl.Y2 = rowBox.TopLeft.X+rowBox.Width, rowBox.TopLeft.Y
|
|
}
|
|
lineEl.Stroke = targetShape.Fill
|
|
lineEl.Style = "stroke-width:2"
|
|
fmt.Fprint(writer, lineEl.Render())
|
|
}
|
|
|
|
if targetShape.Icon != nil && targetShape.Type != d2target.ShapeImage {
|
|
iconPosition := label.FromString(targetShape.IconPosition)
|
|
iconSize := d2target.GetIconSize(box, targetShape.IconPosition)
|
|
|
|
tl := iconPosition.GetPointOnBox(box, label.PADDING, float64(iconSize), float64(iconSize))
|
|
|
|
fmt.Fprintf(writer, `<image href="%s" x="%f" y="%f" width="%d" height="%d" />`,
|
|
html.EscapeString(targetShape.Icon.String()),
|
|
tl.X,
|
|
tl.Y,
|
|
iconSize,
|
|
iconSize,
|
|
)
|
|
}
|
|
}
|