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() {
w.wsclientsMu.Lock()
if w.closing {
w.wsclientsMu.Unlock()
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) {
if page == nil {
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)
@ -130,7 +130,7 @@ func ExportPNG(ms *xmain.State, page playwright.Page, svg []byte) (outputImage [
pngPrefix := "data:image/png;base64,"
if !strings.HasPrefix(pngString, pngPrefix) {
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):]
return base64.StdEncoding.DecodeString(splicedPNGString)