# (c) J.Y.Amihud 2023
# GPL-3 or any later version
# This is a http server that will serve files over network, for multiusering
# across the internet.
# Server side
from http.server import BaseHTTPRequestHandler, HTTPServer
from subprocess import *
import json
import os
import zlib
import sys
import time
import random
import mimetypes
import datetime
import hashlib
import urllib.request
import urllib.parse
import subprocess
# Colors are used to make the
clr = {
"norm":"\033[00m", # Reset to normal
"bold":"\033[01m", # Bold Text
"ital":"\033[03m", # Italic Text
"undr":"\033[04m", # Underlined
"blnk":"\033[05m", # Blinking
# Text
"tdbl":"\033[30m", # Dark Black
"tdrd":"\033[31m", # Dark Red
"tdgr":"\033[32m", # Dark Green
"tdyl":"\033[33m", # Dark Yellow
"tdbu":"\033[34m", # Dark Blue
"tdma":"\033[35m", # Dark Magenta
"tdcy":"\033[36m", # Dark Cyan
"tdwh":"\033[37m", # Dark White
"tbbl":"\033[90m", # Bright Black
"tbrd":"\033[91m", # Bright Red
"tbgr":"\033[92m", # Bright Green
"tbyl":"\033[93m", # Bright Yellow
"tbbu":"\033[94m", # Bright Blue
"tbma":"\033[95m", # Bright Magenta
"tbcy":"\033[96m", # Bright Cyan
"tbwh":"\033[97m", # Bright White
# Background
"bdbl":"\033[40m", # Dark Black
"bdrd":"\033[41m", # Dark Red
"bdgr":"\033[42m", # Dark Green
"bdyl":"\033[43m", # Dark Yellow
"bdbu":"\033[44m", # Dark Blue
"bdma":"\033[45m", # Dark Magenta
"bdcy":"\033[46m", # Dark Cyan
"bdwh":"\033[47m", # Dark White
"bbbl":"\033[100m", # Bright Black
"bbrd":"\033[101m", # Bright Red
"bbgr":"\033[102m", # Bright Green
"bbyl":"\033[103m", # Bright Yellow
"bbbu":"\033[104m", # Bright Blue
"bbma":"\033[105m", # Bright Magenta
"bbcy":"\033[106m", # Bright Cyan
"bbwh":"\033[108m" # Bright White
}
# TODO: Make this not stupidly hardcoded.
blender = "/home/vcs/Software/blender-3.3.0-linux-x64/blender"
try:
PROJECT = sys.argv[2]
except:
print("Please specify a port number and then project folder.")
exit()
try:
ORIGINAL_PROJECT = sys.argv[3]
except:
print("Orignal Project Name Was Not Specified.")
ORIGINAL_PROJECT = " not a folder that can exist "
def fileinfo(filename):
fd = {}
fd["md5"] = hashlib.md5(open(filename,'rb').read()).hexdigest()
fd["filesize"] = os.path.getsize(filename)
return fd
class handler(BaseHTTPRequestHandler):
def start_page(self, code):
self.send_response(code)
def do_GET(self):
# Basic security measure
self.path = self.path.replace("/..", "/")
self.path = self.path.replace("//", "/")
if self.path == "/":
self.start_page(200)
self.wfile.write(b"Welcome to Blender-Pipeline Http Server.\nUse /list/[Folder Name] to list contents of a specific folder.\nUse /download/[File Name] to download the file.\nUse /blend/[Blender File Name] to see Blend File Dependencies.")
elif self.path.startswith("/list/"):
foldername = self.path[5:]
folder = PROJECT+foldername
walk = list(os.walk(folder))[0]
# folders
data = {"folders":walk[1],
"files":{}}
for i in walk[2]:
fd = fileinfo(folder+"/"+i)
data["files"][i] = fd
senddata = json.dumps(data).encode("utf-8")
self.start_page(200)
self.send_header('Content-type', 'application/json')
self.end_headers()
self.wfile.write(senddata)
# THIS WAS A GOOD IDEA ON PAPER, BUT WITH A 100 GB PROJECT
# IT TOOK FOREVER TO DO THIS FOR ONE PART OF THIS PROJECT.
# elif self.path.startswith("/tree/"):
# foldername = self.path[5:]
# folder = PROJECT+foldername
# walk = list(os.walk(folder))
# data = {"folders":[],
# "files":{}}
# for i in walk:
# fol = i[0].replace(PROJECT, "")[1:]
# data["folders"].append(fol)
# for fil in i[2]:
# fd = fileinfo(PROJECT+"/"+fol+"/"+fil)
# data["files"][fol+"/"+fil] = fd
# senddata = json.dumps(data).encode("utf-8")
# self.start_page(200)
# self.send_header('Content-type', 'application/json')
# self.end_headers()
# self.wfile.write(senddata)
elif self.path.startswith("/blend/"):
filename = self.path[6:]
fullfilename = PROJECT+filename
if os.path.exists(fullfilename):
try:
linked = subprocess.check_output([blender, "-b", fullfilename, "-P", os.getcwd()+"/network/get_linked_files.py"]).decode("utf-8")
except:
linked = ""
linked = linked[linked.find("!START_DATA!")+13:linked.rfind("!END_DATA!")]
linked = linked.split("\n")
while "" in linked:
linked.remove("")
parsed = []
for i in linked:
# This is a little weird, sometimes the files have the full URL
# to the data.
if i.startswith(PROJECT): # This is good when the server is on the same machine as the orignal project
parsed.append(i[len(PROJECT):])
elif i.startswith(ORIGINAL_PROJECT): # This is when the server is a mirror
parsed.append(i[len(ORIGINAL_PROJECT):])
else:
backward = i.count("/..")
foldername = filename[:filename.rfind("/")]
f = foldername
for g in range(backward):
f = f[:f.rfind("/")]
cleanname = f+i.replace("//..", "").replace("/..", "").replace( "//", "/")
parsed.append(cleanname)
data = {"files":{}}
for f in parsed:
try:
fd = fileinfo(PROJECT+"/"+f)
data["files"][f] = fd
except Exception as e:
print(e)
linked = json.dumps(data)
self.start_page(200)
self.send_header('Content-type', 'application/json')
self.end_headers()
self.wfile.write(linked.encode("utf-8"))
else:
self.start_page(404)
self.wfile.write(b"File: "+filename.encode("utf-8")+b" doesn't exist on server.")
elif self.path.startswith("/download/"):
filename = self.path[9:]
fullfilename = PROJECT+filename
if os.path.exists(fullfilename):
self.start_page(200)
self.send_header('Content-type', mimetypes.guess_type(filename))
self.end_headers()
f = open(fullfilename, "rb")
f = f.read()
self.wfile.write(f)
else:
self.start_page(404)
self.wfile.write(b"File: "+filename.encode("utf-8")+b" doesn't exist")
elif self.path.startswith("/download_z/"):
filename = self.path[11:]
fullfilename = PROJECT+filename
if os.path.exists(fullfilename):
self.start_page(200)
self.send_header('Content-type', mimetypes.guess_type(filename))
f = open(fullfilename, "rb")
f = f.read()
z = zlib.compress(f)
self.send_header('Content-length', len(z))
self.end_headers()
self.wfile.write(z)
else:
self.start_page(404)
self.wfile.write(b"File: "+filename.encode("utf-8")+b" doesn't exist")
try:
PORT = int(sys.argv[1])
except:
print("Please specify a port number and then project folder.")
exit()
serve = HTTPServer(("", PORT), handler)
serve.serve_forever()