Add docker and requirements.txt

.gitignore

@@ -0,0 +1 @@
+clones/

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.10.1
+WORKDIR /matterport-dl
+
+COPY requirements.txt requirements.txt
+RUN pip3 install -r requirements.txt
+
+COPY docker-entrypoint.sh docker-entrypoint.sh
+COPY graph_posts graph_posts
+COPY matterport-dl.py matterport-dl.py
+
+CMD [ "/bin/sh", "docker-entrypoint.sh" ]

README.md

@@ -14,6 +14,33 @@ A tool to download/archive [Matterport](https://matterport.com) virtual tours.
 -   Add `--proxy 127.0.0.1:1234` to a download run to use a proxy for requests
 -   Add `--advanced-download` to a download run to try and download the needed textures and files for supporting dollhouse/floorplan views.  NOTE: Must use built in webserver to host content for this to work.
 
+# Docker
+## Docker params
+```
+docker build -t matterport-dl .
+docker run -v $(pwd)/clones:/matterport-dl/clones -e M_ID=[url_or_page_id] matterport-dl
+docker run -p 8080:8080 -v $(pwd)/clones:/matterport-dl/clones -e M_ID=[url_or_page_id] -e BIND_IP=0.0.0.0 -e BIND_PORT=8080 -e ADV_DL=true -e PROXY=127.0.0.1:1234 -e BASE_FOLDER="./clones" matterport-dl
+```
+
+* M_ID Matterport ID or URL
+* BIND_PORT Defaults to 8080 if not set
+* BIND_IP IP address to bind to. Use `0.0.0.0` unless setting docker network to host.
+* ADV_DL is for the --advanced-download flag, and is off by default. Setting this to anything will activate it.
+* PROXY is for the --proxy flag, and is off by default.
+* BASE_FOLDER is where the downloads go. Defaults to "./clones"
+
+## Docker example
+```
+docker build -t matterport-dl .
+docker run -v $(pwd)/clones:/matterport-dl/clones -e M_ID="https://my.matterport.com/show/?m=roWLLMMmPL8" -e ADV_DL=true matterport-dl
+docker run -p 8080:8080 -v $(pwd)/clones:/matterport-dl/clones -e M_ID=roWLLMMmPL8 -e BIND_IP=0.0.0.0 -d matterport-dl
+```
+
+## Docker debugging
+```
+docker build --no-cache -t matterport-dl .
+docker run -t -i -p 8080:8080 -v $(pwd)/clones:/matterport-dl/clones -e M_ID=roWLLMMmPL8 -e BIND_IP=0.0.0.0 matterport-dl /bin/bash
+```
 
 # Additional Notes
 * It is possible to host these Matterport archives using standard web servers however: 1) Certain features beyond the tour itself may not work.  2)  #1 may be fixable by specific rewrite rules for apache/nginx.  These are not currently provided but if you look at `OurSimpleHTTPRequestHandler` class near the bottom of the source file you can likely figure out what redirects we do.

docker-entrypoint.sh

@@ -0,0 +1,24 @@
+#!/bin/bash
+set -e
+
+if [[ -z "${BIND_PORT}" ]] && [[ -n "$BIND_IP" ]]; then
+  echo "Defaulting to port 8080"
+  BIND_PORT=8080
+fi
+
+if [[ -n "${PROXY}" ]]; then
+  echo "Using proxy: '$PROXY'"
+  PROXYARG="--proxy $PROXY"
+fi
+
+if [[ -n "${BASE_FOLDER}" ]]; then
+  echo "Using base folder: '$BASE_FOLDER'"
+  BASEFOLDERARG="--base-folder $BASE_FOLDER"
+fi
+
+if [[ -n "${ADV_DL}" ]]; then
+  ADVDLARG="--advanced-download"
+fi
+
+cd /matterport-dl
+/usr/local/bin/python3 matterport-dl.py $M_ID $BIND_IP $BIND_PORT $ADVDLARG $PROXYARG $BASEFOLDERARG

matterport-dl.py

@@ -6,6 +6,7 @@
 '''
 
 import requests
+import functools
 import json
 import threading
 import concurrent.futures
@@ -22,7 +23,9 @@
 from http.server import HTTPServer, SimpleHTTPRequestHandler
 import decimal
 
+print = functools.partial(print, flush=True)
 
+baseCloneDir = "clones/"
 
 # Weird hack
 accessurls = []
@@ -306,6 +309,7 @@ def downloadPage(pageid):
         accessurl = f'{match.group(1)}~/{{filename}}{match.group(3)}'
         print(accessurl)
     else:
+        print('Requesting', pageid)
         raise Exception("Can't find urls")
 
 
@@ -390,6 +394,10 @@ def getPageId(url):
     return url.split("m=")[-1].split("&")[0]
 
 class OurSimpleHTTPRequestHandler(SimpleHTTPRequestHandler):
+    def __init__(self, *args, **kwargs):
+        SERVEDIR=pageId
+        super().__init__(*args, directory=SERVEDIR, **kwargs)
+
     def send_error(self, code, message=None):
         if code == 404:
             logging.warning(f'404 error: {self.path} may not be downloading everything right')
@@ -404,7 +412,7 @@ def do_GET(self):
             self.path = f"/js/{SHOWCASE_INTERNAL_NAME}"
 
         if self.path.startswith("/locale/messages/strings_") and not os.path.exists(f".{self.path}"):
-            redirect_msg = "original request was for a locale we do not have downloaded"
+            redirect_msg = "original request was for a locale that is not downloaded"
             self.path = "/locale/strings.json"
         raw_path, _, query = self.path.partition('?')
         if "crop=" in query and raw_path.endswith(".jpg"):
@@ -500,24 +508,40 @@ def getCommandLineArg(name, has_value):
 
 if __name__ == "__main__":
     ADVANCED_DOWNLOAD_ALL = getCommandLineArg("--advanced-download", False)
+    customBaseCloneDir = getCommandLineArg("--base-folder", True)
+    if customBaseCloneDir:
+      baseCloneDir = customBaseCloneDir
     PROXY = getCommandLineArg("--proxy", True)
     OUR_OPENER = getUrlOpener(PROXY)
     urllib.request.install_opener(OUR_OPENER)
-    pageId = ""
-    if len(sys.argv) > 1:
-        pageId = getPageId(sys.argv[1])
-    openDirReadGraphReqs("graph_posts",pageId)
-    if len(sys.argv) == 2:
+    pageId = getPageId(sys.argv[1])
+    if len(sys.argv) == 2: # Download mode
+        os.chdir(baseCloneDir)
+        openDirReadGraphReqs("graph_posts",pageId)
         initiateDownload(pageId)
-    elif len(sys.argv) == 4:
-        os.chdir(getPageId(pageId))
+    elif len(sys.argv) == 4: # Serve mode
+        if os.path.isdir(pageId) and not os.path.isdir(os.path.join(baseCloneDir, pageId)):
+          print("WARNING:")
+          print("The folder '" + pageId + "' does not exist in the '" + baseCloneDir + "' directory, but does exist in the root directory.")
+          print("If you recently updated matterport-dl, then please move the '" + pageId + "' folder into the '" + baseCloneDir + "' directory")
+          print("For the time being, this will continue to work, but eventually all downloads need to be moved")
+          print(" ")
+          time.sleep(2)
+        else:
+          os.chdir(baseCloneDir)
+        if not os.path.isdir(pageId):
+          print("The folder '" + pageId + "' does not exist. Please download it using the download command. See the readme for more details")
+          sys.exit()
         try:
             logging.basicConfig(filename='server.log', encoding='utf-8', level=logging.DEBUG,  format='%(asctime)s %(levelname)-8s %(message)s',datefmt='%Y-%m-%d %H:%M:%S')
         except ValueError:
             logging.basicConfig(filename='server.log', level=logging.DEBUG,  format='%(asctime)s %(levelname)-8s %(message)s',datefmt='%Y-%m-%d %H:%M:%S')
-        logging.info("Server started up")
-        print ("View in browser: http://" + sys.argv[2] + ":" + sys.argv[3])
         httpd = HTTPServer((sys.argv[2], int(sys.argv[3])), OurSimpleHTTPRequestHandler)
+        logging.info("Server started up")
+        if sys.argv[2] == '0.0.0.0':
+          print ("View in browser: http://localhost:" + sys.argv[3])
+        else:
+          print ("View in browser: http://" + sys.argv[2] + ":" + sys.argv[3])
         httpd.serve_forever()
     else:
         print (f"Usage:\n\tFirst Download: matterport-dl.py [url_or_page_id]\n\tThen launch the server 'matterport-dl.py [url_or_page_id] 127.0.0.1 8080' and open http://127.0.0.1:8080 in a browser\n\t--proxy 127.0.0.1:1234 -- to have it use this web proxy\n\t--advanced-download -- Use this option to try and download the cropped files for dollhouse/floorplan support")

requirements.txt

@@ -0,0 +1,2 @@
+tqdm>=4.62.3
+requests>=2.26.0