Merge pull request #18 from Carifio24/serve-ngrok

Allow using ngrok for QR code sharing

glue_ar/qr_dialog.py

@@ -16,4 +16,5 @@ def __init__(self, parent=None, img=None):
         self.pix = QPixmap.fromImage(self.img)
         self.ui = load_ui("qr_dialog.ui", self, directory=dirname(__file__))
         self.ui.label_image.setPixmap(self.pix)
-        self.setFixedSize(300, 325)
+        width, height = img.size
+        self.setFixedSize(width, height + 25)

glue_ar/tools.py

@@ -6,6 +6,7 @@
 from glue_vispy_viewers.scatter.scatter_viewer import Vispy3DScatterViewerState
 from glue_vispy_viewers.volume.layer_state import VolumeLayerState
 
+import ngrok
 import pyvista as pv
 
 from qtpy import compat
@@ -119,19 +120,28 @@ def activate(self):
             _, gltf_base = split(gltf_tmp.name)
             export_modelviewer(html_tmp.name, gltf_base, self.viewer.state.title)
 
+
             port = 4000
             directory, filename = split(html_tmp.name)
             server = run_ar_server(port, directory)
+            use_ngrok = os.getenv("NGROK_AUTHTOKEN", None) is not None
 
             try:
                 thread = Thread(target=server.serve_forever)
                 thread.start()
 
-                ip = get_local_ip()
-                url = f"http://{ip}:{port}/{filename}"
+                if use_ngrok:
+                    listener = ngrok.forward(port, authtoken_from_env=True)
+                    url = f"{listener.url()}/{filename}"
+                else:
+                    ip = get_local_ip()
+                    url = f"http://{ip}:{port}/{filename}"
                 img = create_qr(url)
                 dialog = QRDialog(parent=self.viewer, img=img)
                 dialog.exec_()
 
             finally:
                 server.shutdown()
+                if use_ngrok:
+                    listener.close()
+

requirements.txt

@@ -2,6 +2,7 @@ gltflib
 glue-core
 glue-qt
 glue-vispy-viewers
+ngrok
 pillow
 pyvista
 segno

setup.cfg

@@ -11,6 +11,7 @@ install_requires =
     glue-core
     glue-qt
     glue-vispy-viewers
+    ngrok
     pillow
     pyvista
     segno