wails: add execute js on document load

ciderapp · Mar 5, 2023 · a6d39ee · a6d39ee
1 parent 9cd2422
commit a6d39ee
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 0 deletions.
diff --git a/v2/internal/frontend/desktop/windows/frontend.go b/v2/internal/frontend/desktop/windows/frontend.go
@@ -665,6 +665,12 @@ func (f *Frontend) ExecJS(js string) {
 	})
 }
 
+func (f *Frontend) ExecJSOnDocumentLoad(js string) {
+	f.mainWindow.Invoke(func() {
+		f.chromium.EvalOnDocumentCreate(js)
+	})
+}
+
 func (f *Frontend) navigationCompleted(sender *edge.ICoreWebView2, args *edge.ICoreWebView2NavigationCompletedEventArgs) {
 	if f.frontendOptions.OnDomReady != nil {
 		go f.frontendOptions.OnDomReady(f.ctx)

diff --git a/v2/internal/frontend/desktop/windows/go-webview2/pkg/edge/chromium.go b/v2/internal/frontend/desktop/windows/go-webview2/pkg/edge/chromium.go
@@ -184,6 +184,21 @@ func (e *Chromium) Eval(script string) {
 	return
 }
 
+func (e *Chromium) EvalOnDocumentCreate(script string) {
+
+	_script, err := windows.UTF16PtrFromString(script)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	_, _, _ = e.webview.vtbl.AddScriptToExecuteOnDocumentCreated.Call(
+		uintptr(unsafe.Pointer(e.webview)),
+		uintptr(unsafe.Pointer(_script)),
+		0,
+	)
+	return
+}
+
 func (e *Chromium) Show() error {
 	return e.controller.PutIsVisible(true)
 }

diff --git a/v2/internal/frontend/frontend.go b/v2/internal/frontend/frontend.go
@@ -67,6 +67,7 @@ type Frontend interface {
 	Run(context.Context) error
 	RunMainLoop()
 	ExecJS(js string)
+	ExecJSOnDocumentLoad(js string)
 	Hide()
 	Show()
 	Quit()

diff --git a/v2/pkg/runtime/window.go b/v2/pkg/runtime/window.go
@@ -169,6 +169,12 @@ func WindowExecJS(ctx context.Context, js string) {
 	appFrontend.ExecJS(js)
 }
 
+// WindowExecJSOnDocumentLoad executes the given Js in the window when it is loaded
+func WindowExecJSOnDocumentLoad(ctx context.Context, js string) {
+	appFrontend := getFrontend(ctx)
+	appFrontend.ExecJSOnDocumentLoad(js)
+}
+
 func WindowSetBackgroundColour(ctx context.Context, R, G, B, A uint8) {
 	appFrontend := getFrontend(ctx)
 	col := &options.RGBA{