luciano/d2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Small tweaks

Browse source
This commit is contained in:
Bernard Xie 2022-11-17 11:34:14 -08:00
parent 6ef4246565
commit 0e70c0baa5
No known key found for this signature in database
GPG key ID: 3C3E0036CE0F892C
2 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
cmd/d2/watch.go
View file

@ -147,7 +147,6 @@ func (w *watcher) run() error {
func (w *watcher) close() { func (w *watcher) close() {
w.wsclientsMu.Lock() w.wsclientsMu.Lock()
if w.closing { if w.closing {
w.wsclientsMu.Unlock() w.wsclientsMu.Unlock()
return return

4
lib/png/png.go
View file

@ -117,7 +117,7 @@ var genPNGScript string
func ExportPNG(ms *xmain.State, page playwright.Page, svg []byte) (outputImage []byte, err error) { func ExportPNG(ms *xmain.State, page playwright.Page, svg []byte) (outputImage []byte, err error) {
if page == nil { if page == nil {
ms.Log.Error.Printf("Playwright was not initialized properly for PNG export") ms.Log.Error.Printf("Playwright was not initialized properly for PNG export")
return nil, fmt.Errorf("Playwright page is not initialized for png export") return nil, fmt.Errorf("Playwright page is nil")
} }
encodedSVG := base64.StdEncoding.EncodeToString(svg) encodedSVG := base64.StdEncoding.EncodeToString(svg)
@ -130,7 +130,7 @@ func ExportPNG(ms *xmain.State, page playwright.Page, svg []byte) (outputImage [
pngPrefix := "data:image/png;base64," pngPrefix := "data:image/png;base64,"
if !strings.HasPrefix(pngString, pngPrefix) { if !strings.HasPrefix(pngString, pngPrefix) {
ms.Log.Error.Printf("failed to convert D2 file to PNG") ms.Log.Error.Printf("failed to convert D2 file to PNG")
return nil, fmt.Errorf("playwright export generated invalid png") return nil, fmt.Errorf("Playwright export generated invalid png")
} }
splicedPNGString := pngString[len(pngPrefix):] splicedPNGString := pngString[len(pngPrefix):]
return base64.StdEncoding.DecodeString(splicedPNGString) return base64.StdEncoding.DecodeString(splicedPNGString)