From 1d27bd099b05d720531e0882d52c67ad198b6807 Mon Sep 17 00:00:00 2001
From: paulmasson <paulmasson@analyticphysics.com>
Date: Fri, 31 Mar 2017 15:38:15 -0700
Subject: [PATCH] Add methods for offline scripts

---
 src/sage/plot/plot3d/base.pyx                | 28 ++++---------
 src/sage/repl/rich_output/backend_ipython.py | 41 +++++++++++++++++++-
 2 files changed, 47 insertions(+), 22 deletions(-)

diff --git a/src/sage/plot/plot3d/base.pyx b/src/sage/plot/plot3d/base.pyx
index 688dfbd255f..36ec4d36247 100644
--- a/src/sage/plot/plot3d/base.pyx
+++ b/src/sage/plot/plot3d/base.pyx
@@ -377,28 +377,14 @@ cdef class Graphics3d(SageObject):
             options['online'] = True
 
         if options['online']:
-            scripts = ( """
-<script src="https://cdn.rawgit.com/mrdoob/three.js/r80/build/three.min.js"></script>
-<script src="https://cdn.rawgit.com/mrdoob/three.js/r80/examples/js/controls/OrbitControls.js"></script>
-            """ )
+            from sage.misc.package import installed_packages
+            version = installed_packages()['threejs']
+            scripts = """
+<script src="https://cdn.rawgit.com/mrdoob/three.js/{0}/build/three.min.js"></script>
+<script src="https://cdn.rawgit.com/mrdoob/three.js/{0}/examples/js/controls/OrbitControls.js"></script>
+            """.format(version)
         else:
-            from sage.repl.rich_output.backend_ipython import BackendIPythonNotebook
-            if isinstance(backend, BackendIPythonNotebook):
-                scripts = ( """
-<script src="/nbextensions/threejs/three.min.js"></script>
-<script src="/nbextensions/threejs/OrbitControls.js"></script>
-<script>
-  if ( !window.THREE ) document.write('\
-<script src="https://cdn.rawgit.com/mrdoob/three.js/r80/build/three.min.js"><\/script>\
-<script src="https://cdn.rawgit.com/mrdoob/three.js/r80/examples/js/controls/OrbitControls.js"><\/script>');
-</script>
-                """ )
-            else:
-                from sage.env import SAGE_SHARE
-                scripts = ( """
-<script src="{0}/threejs/three.min.js"></script>
-<script src="{0}/threejs/OrbitControls.js"></script>
-                """.format( SAGE_SHARE ) )
+            scripts = backend.threejs_offline_scripts()
 
         b = self.bounding_box()
         bounds = '[{{"x":{}, "y":{}, "z":{}}}, {{"x":{}, "y":{}, "z":{}}}]'.format(
diff --git a/src/sage/repl/rich_output/backend_ipython.py b/src/sage/repl/rich_output/backend_ipython.py
index e5e85105df8..fa98be8bb0b 100644
--- a/src/sage/repl/rich_output/backend_ipython.py
+++ b/src/sage/repl/rich_output/backend_ipython.py
@@ -396,6 +396,24 @@ def is_in_terminal(self):
         """
         return True
 
+    def threejs_offline_scripts(self):
+        """
+        Three.js offline scripts for the IPython command line
+
+        EXAMPLES::
+
+            sage: from sage.repl.rich_output.backend_ipython import BackendIPythonCommandline
+            sage: backend = BackendIPythonCommandline()
+            sage: backend.threejs_offline_scripts()
+            '...<script ...</script>...'
+        """
+        from sage.env import SAGE_SHARE
+        return """
+<script src="{0}/threejs/three.min.js"></script>
+<script src="{0}/threejs/OrbitControls.js"></script>
+        """.format(SAGE_SHARE)
+
+
 
 IFRAME_TEMPLATE = \
 """
@@ -554,4 +572,25 @@ def displayhook(self, plain_text, rich_output):
         else:
             raise TypeError('rich_output type not supported')
 
-        
+    def threejs_offline_scripts(self):
+        """
+        Three.js offline scripts for the IPython notebook
+
+        EXAMPLES::
+
+            sage: from sage.repl.rich_output.backend_ipython import BackendIPythonNotebook
+            sage: backend = BackendIPythonNotebook()
+            sage: backend.threejs_offline_scripts()
+            '...<script ...</script>...'
+        """
+        from sage.misc.package import installed_packages
+        version = installed_packages()['threejs']
+        return """
+<script src="/nbextensions/threejs/three.min.js"></script>
+<script src="/nbextensions/threejs/OrbitControls.js"></script>
+<script>
+  if ( !window.THREE ) document.write('\
+<script src="https://cdn.rawgit.com/mrdoob/three.js/{0}/build/three.min.js"><\/script>\
+<script src="https://cdn.rawgit.com/mrdoob/three.js/{0}/examples/js/controls/OrbitControls.js"><\/script>');
+</script>
+        """.format(version)