From c438b9f8a5015085ccb0c5e87d119afa3732fc9e Mon Sep 17 00:00:00 2001
From: shortcutme <tamas@zeronet.io>
Date: Wed, 21 Feb 2018 03:02:00 +0100
Subject: [PATCH] Create a restricted ZeroFrame wrapper

---
 src/Ui/media/WrapperZeroFrame.coffee  | 22 ++++++++++++++++++++++
 src/Ui/media/lib/ZeroWebsocket.coffee |  2 +-
 2 files changed, 23 insertions(+), 1 deletion(-)
 create mode 100644 src/Ui/media/WrapperZeroFrame.coffee

diff --git a/src/Ui/media/WrapperZeroFrame.coffee b/src/Ui/media/WrapperZeroFrame.coffee
new file mode 100644
index 00000000..14313dd3
--- /dev/null
+++ b/src/Ui/media/WrapperZeroFrame.coffee
@@ -0,0 +1,22 @@
+class WrapperZeroFrame
+	constructor: (wrapper) ->
+		@wrapperCmd = wrapper.cmd
+		@wrapperResponse = wrapper.ws.response
+		console.log "WrapperZeroFrame", wrapper
+
+	cmd: (cmd, params={}, cb=null) =>
+		@wrapperCmd(cmd, params, cb)
+
+	response: (to, result) =>
+		@wrapperResponse(to, result)
+
+	isProxyRequest: ->
+		return window.location.pathname == "/"
+
+	certSelectGotoSite: (elem) =>
+		href = $(elem).attr("href")
+		if @isProxyRequest() # Fix for proxy request
+			$(elem).attr("href", "http://zero#{href}")
+
+
+window.zeroframe = new WrapperZeroFrame(window.wrapper)
diff --git a/src/Ui/media/lib/ZeroWebsocket.coffee b/src/Ui/media/lib/ZeroWebsocket.coffee
index 01228226..aa795677 100644
--- a/src/Ui/media/lib/ZeroWebsocket.coffee
+++ b/src/Ui/media/lib/ZeroWebsocket.coffee
@@ -37,7 +37,7 @@ class ZeroWebsocket
 		@log "Unknown command", message
 
 
-	response: (to, result) ->
+	response: (to, result) =>
 		@send {"cmd": "response", "to": to, "result": result}