Userscript iFrame

couchpotato/core/plugins/userscript/__init__.py

@@ -1,6 +1,6 @@
-from .main import Extension
+from .main import Userscript
 
 def start():
-    return Extension()
+    return Userscript()
 
 config = []

couchpotato/core/plugins/userscript/main.py

@@ -1,3 +1,4 @@
+from couchpotato import index
 from couchpotato.api import addApiView
 from couchpotato.core.event import fireEvent
 from couchpotato.core.helpers.request import getParam, jsonified
@@ -10,12 +11,12 @@ from flask.helpers import url_for
 log = CPLog(__name__)
 
 
-class Extension(Plugin):
+class Userscript(Plugin):
 
     def __init__(self):
-        addApiView('userscript', self.getExtension)
-
-        addApiView('userscript.add_via_url', self.addViaUrl)
+        addApiView('userscript.get', self.getExtension)
+        addApiView('userscript', self.iFrame)
+        addApiView('userscript.add_via_url', self.getViaUrl)
 
     def getExtension(self):
 
@@ -38,9 +39,8 @@ class Extension(Plugin):
 
         return version
 
-    def addViaUrl(self):
-        url = getParam('url')
-        return self.renderTemplate(__file__, 'iframe.html', url = url)
+    def iFrame(self):
+        return index()
 
     def getViaUrl(self):
 

couchpotato/core/plugins/userscript/static/userscript.css

@@ -0,0 +1,14 @@
+.page.userscript {
+	position: absolute;
+	width: 100%;
+	top: 0;
+	bottom: 0;
+	left: 0;
+	right: 0;
+}
+
+	.page.userscript .frame.loading {
+		text-align: center;
+		font-size: 20px;
+		padding: 20px;
+	}

couchpotato/core/plugins/userscript/static/userscript.js

@@ -0,0 +1,46 @@
+Page.Userscript = new Class({
+
+	Extends: PageBase,
+
+	name: 'userscript',
+	has_tab: false,
+
+	options: {
+		'onOpened': function(){
+			App.stopLoadTimer();
+			App.getBlock('header').hide();
+		}
+	},
+
+	indexAction: function(param){
+		var self = this;
+
+		self.el.adopt(
+			self.frame = new Element('div.frame.loading', {
+				'text': 'Loading...'
+			})
+		);
+
+		var url = window.location.href.split('url=')[1];
+
+		Api.request('userscript.add_via_url', {
+			'data': {
+				'url': url
+			},
+			'onComplete': function(json){
+				self.frame.empty();
+				self.frame.removeClass('loading');
+
+				if(json.error)
+					self.frame.set('html', json.error);
+				else {
+					var item = new Block.Search.Item(json.movie);
+					self.frame.adopt(item);
+					item.showOptions();
+				}
+			}
+		});
+
+	}
+
+});