Added origin in document path to avoid conflicts for documents with the same name from different origins (such as forked documents)

src/api/api_document.py

@@ -9,10 +9,11 @@ bp = Blueprint('api_document', __name__, url_prefix='/api/doc')
 
 @bp.route('/build')
 def build():
-	doc_name = get_arg('doc')
+	origin = get_arg('origin')
+	doc_name = get_arg('doc_name')
 	branch = get_arg('branch', 'master')
 	apikey = get_arg('apikey')
-	doc = Document(doc_name, branch)
+	doc = Document(origin, doc_name, branch)
 	
 	if doc.get_api_key() != apikey:
 		raise BusinessException("Invalid API key")
@@ -25,7 +26,7 @@ def build():
 @bp.route('/clone')
 def clone():
 	repo = get_arg('repo')
-	doc_name = get_arg('doc', os.path.splitext(os.path.basename(repo))[0])
+	doc_name = get_arg('doc_name', os.path.splitext(os.path.basename(repo))[0])
 	branch = get_arg('branch', 'master')
 	source_dir = get_arg('source', 'source')
 	

src/data/document.py

@@ -5,6 +5,8 @@ from web_utils.task import ProcessTask
 from web_utils.business_exception import BusinessException
 import shutil
 from unicodedata import normalize
+from urllib.parse import quote as url_encode
+from urllib.parse import unquote as url_decode
 import string
 
 def os_path_separators():
@@ -14,32 +16,39 @@ def os_path_separators():
 			seps.append(sep)
 	return seps
 
-def sanitize_name(initial_name):
+def sanitize_name(initial_name, slashReplacement = '_', removeUnknownCharacters = True):
 	# Sort out unicode characters
 	name = normalize('NFKD', initial_name).encode('ascii', 'ignore').decode('ascii')
-	# Replace path separators with underscores
-	name = name.replace('/', '_slash_')
+	
+	# Replace path separators
 	for sep in os_path_separators():
-		name = name.replace(sep, '_')
+		name = name.replace(sep, slashReplacement)
+	
 	# Ensure only valid characters
-	valid_chars = "-_.(){0}{1}".format(string.ascii_letters, string.digits)
-	name = "".join(ch for ch in name if ch in valid_chars)
+	if removeUnknownCharacters:
+		valid_chars = "-_.{0}{1}{2}".format(string.ascii_letters, string.digits, slashReplacement)
+		name = "".join(ch for ch in name if ch in valid_chars)
+	
 	if len(name) == 0 or '..' in name:
 		raise BusinessException("Invalid name: " + initial_name)
+	
 	return name
 
 class Document:
-	def __init__(self, doc_name, branch = 'master', allow_invalid = False):
+	def __init__(self, origin, doc_name, branch = 'master', allow_invalid = False):
+		self.origin = Document.decode_origin(origin) if '!' in origin else origin
+		self.encoded_origin = Document.encode_origin(self.origin)
 		self.doc_name = doc_name
 		self.branch = branch
 		
-		doc_path = Document.make_doc_path(doc_name, branch)
+		doc_path = Document.make_doc_path(self.origin, doc_name, branch)
+		print(doc_path)
 		if not os.path.isdir(doc_path + "/repo/.git"):
 			if allow_invalid:
 				self.valid = False
 				return
 			else:
-				raise BusinessException("This document does not exist: "+doc_name+"@"+branch)
+				raise BusinessException("This document does not exist: "+self.origin+'/'+doc_name+"@"+branch)
 		
 		self.doc_path = doc_path
 		self.valid = True
@@ -57,32 +66,56 @@ class Document:
 		return task
 	
 	def delete(self):
-		shutil.rmtree(self.doc_path)
+		if not self.valid:
+			raise Exception("Internal error")
+		self.delete_folder()
 		
 	def delete_folder(self):
-		doc_path = Document.make_doc_path(self.doc_name, self.branch)
+		doc_path = Document.make_doc_path(self.origin, self.doc_name, self.branch)
 		shutil.rmtree(doc_path)
+		doc_root = os.path.dirname(doc_path)
+		if len(os.listdir(doc_root)) == 0:
+			os.rmdir(doc_root)
+			origin_root = os.path.dirname(doc_root)
+			if len(os.listdir(origin_root)) == 0:
+				os.rmdir(origin_root)
 	
 	def get_url(self):
-		return "/doc/" + sanitize_name(self.doc_name)+'/'+sanitize_name(self.branch) + "/index.html"
+		return "/doc/" + self.encoded_origin + "/" + sanitize_name(self.doc_name)+'/'+sanitize_name(self.branch) + "/index.html"
 	
 	def get_api_key(self):
 		with open(self.doc_path + "/apikey") as f:
 			return f.read().replace('\n', '')
 		
 	@staticmethod
-	def make_doc_path(doc_name, branch):
-		doc_path = os.path.realpath(get_document_root()+'/'+sanitize_name(doc_name)+'/'+sanitize_name(branch))
+	def encode_origin(origin):
+		return url_encode(origin, safe='').replace('%', '!')
+	
+	@staticmethod
+	def decode_origin(origin):
+		return url_decode(origin.replace('!', '%'))
+	
+	@staticmethod
+	def make_doc_path(origin, doc_name, branch):
+		doc_path = os.path.realpath(get_document_root()+'/'+Document.encode_origin(origin)+'/'+sanitize_name(doc_name)+'/'+sanitize_name(branch))
 		if not doc_path.startswith(get_document_root()):
-			raise BusinessException("Invalid document path for "+doc_name+"@"+branch)
+			raise BusinessException("Invalid document path for "+origin+"/"+doc_name+"@"+branch)
 		return doc_path
 	
+	@staticmethod
+	def get_origin(repo):
+		result = sanitize_name(os.path.dirname(repo).replace('https://', ''), '/', False)
+		if '!' in result:
+			raise BusinessException("Invalid character: !")
+		return result
+	
 	@staticmethod
 	def clone(repo, branch, doc_name, source_dir):
 		# check the document does not already exist
-		doc_path = Document.make_doc_path(doc_name, branch)
+		origin = Document.get_origin(repo)
+		doc_path = Document.make_doc_path(origin, doc_name, branch)
 		if os.path.isdir(doc_path):
-			raise BusinessException("This document already exists: "+doc_name+"@"+branch)
+			raise BusinessException("This document already exists: "+origin+"/"+doc_name+"@"+branch)
 			
 		if source_dir != sanitize_name(source_dir):
 			raise BusinessException("Invalid source directory name: " + source_dir)
@@ -93,11 +126,12 @@ class Document:
 			
 		# Generate an API key
 		apikey = str(uuid.uuid4())
+		print("generated API key: " + apikey)
 		
 		target_dir = doc_path + "/repo"
 		os.makedirs(target_dir, exist_ok = True)
-		with open(doc_path + "/apikey", "wb") as apikey_file:
-			apikey_file:write(apikey)
+		with open(doc_path + "/apikey", "w") as apikey_file:
+			apikey_file.write(apikey)
 		
 		cmd = []
 		cmd.append(['git', 'init', '--initial-branch=' + branch])
@@ -116,10 +150,11 @@ class Document:
 	@staticmethod
 	def list():
 		result = []
-		for doc_name in os.listdir(get_document_root()):
-			for branch in os.listdir(get_document_root() + "/" + doc_name):
-				doc = Document(doc_name, branch, allow_invalid = True)
-				result.append(doc)
+		for origin in os.listdir(get_document_root()):
+			for doc_name in os.listdir(get_document_root() + "/" + origin):
+				for branch in os.listdir(get_document_root() + "/" + origin + "/" + doc_name):
+					doc = Document(origin, doc_name, branch, allow_invalid = True)
+					result.append(doc)
 		return result
 
 def get_document_root():

src/templates/admin/document/manage.html

@@ -3,10 +3,10 @@
 {% block title %}{{ doc.doc_name }} / {{ doc.branch }} (admin){% endblock %}
 
 {% block content %}
-	<h1>Gestion de {{ doc.doc_name }} / {{ doc.branch }}</h1>
-	<p>URL permettant de déclencher la compilation : {{ url_for('api_document.build', doc = doc.doc_name, branch = doc.branch, apikey = doc.get_api_key(), _external = True) }}</p>
+	<h1>Gestion de {{ doc.origin }} / {{ doc.doc_name }} / {{ doc.branch }}</h1>
+	<p>URL permettant de déclencher la compilation : {{ url_for('api_document.build', origin = doc.encoded_origin, doc_name = doc.doc_name, branch = doc.branch, apikey = doc.get_api_key(), _external = True) }}</p>
 	<a href="{{doc.get_url()}}" class="button">Consulter</a><br/>
-	<a href="{{ url_for('admin_document.build', doc_name = doc.doc_name, branch = doc.branch) }}" class="button">Compiler</a><br/>
+	<a href="{{ url_for('admin_document.build', origin = doc.encoded_origin, doc_name = doc.doc_name, branch = doc.branch) }}" class="button">Compiler</a><br/>
 	<br/>
-	<a href="{{ url_for('admin_document.delete', doc_name = doc.doc_name, branch = doc.branch) }}" class="button danger" data-confirm="Êtes-vous sûr de vouloir supprimer le document {{ doc.doc_name }} / {{ doc.branch }} ?">Supprimer</a>
+	<a href="{{ url_for('admin_document.delete', origin = doc.encoded_origin, doc_name = doc.doc_name, branch = doc.branch) }}" class="button danger" data-confirm="Êtes-vous sûr de vouloir supprimer le document {{ doc.doc_name }} / {{ doc.branch }} ?">Supprimer</a>
 {% endblock %}

src/templates/admin/index.html

@@ -10,9 +10,9 @@
 		<ul>
 			{% for doc in documents %}
 				{% if doc.valid %}
-					<li>{{ doc.doc_name }} / {{ doc.branch }} <a href="{{ doc.get_url() }}" class="button">Consulter</a> <a href="{{ url_for('admin_document.manage', doc_name = doc.doc_name, branch = doc.branch) }}" class="button">Gérer</a></li>
+					<li>{{ doc.origin }} / {{ doc.doc_name }} / {{ doc.branch }} <a href="{{ doc.get_url() }}" class="button">Consulter</a> <a href="{{ url_for('admin_document.manage', origin = doc.encoded_origin, doc_name = doc.doc_name, branch = doc.branch) }}" class="button">Gérer</a></li>
 				{% else %}
-					<li>{{ doc.doc_name }} / {{ doc.branch }} (document invalide) <a href="{{ url_for('admin_document.delete_invalid', doc_name = doc.doc_name, branch = doc.branch) }}" class="confirm danger">Supprimer le dossier</a></li>
+					<li>{{ doc.origin }} / {{ doc.doc_name }} / {{ doc.branch }} (document invalide) <a href="{{ url_for('admin_document.delete_invalid', origin = doc.encoded_origin, doc_name = doc.doc_name, branch = doc.branch) }}" class="confirm danger">Supprimer le dossier</a></li>
 				{% endif %}
 			{% endfor %}
 		</ul>

src/web/admin/admin_document.py

@@ -20,29 +20,29 @@ def new():
 		
 		clone_task = Document.clone(repo, branch, doc_name, source_dir)
 		
-		return render_template("admin/command_output.html", task = clone_task, next = url_for('admin_document.manage', doc_name = doc_name, branch = branch))
+		return render_template("admin/command_output.html", task = clone_task, next = url_for('admin_document.manage', origin = Document.encode_origin(Document.get_origin(repo)), doc_name = doc_name, branch = branch))
 	else:
 		return render_template("admin/document/new.html")
 
-@bp.route('/manage/<doc_name>/<branch>')
-def manage(doc_name, branch):
-	return render_template("admin/document/manage.html", doc=Document(doc_name, branch))
+@bp.route('/manage/<origin>/<doc_name>/<branch>')
+def manage(origin, doc_name, branch):
+	return render_template("admin/document/manage.html", doc=Document(origin, doc_name, branch))
 
-@bp.route('/build/<doc_name>/<branch>')
-def build(doc_name, branch):
-	doc = Document(doc_name, branch)
+@bp.route('/build/<origin>/<doc_name>/<branch>')
+def build(origin, doc_name, branch):
+	doc = Document(origin, doc_name, branch)
 	build_task = doc.build()
 	
-	return render_template("admin/command_output.html", task = build_task, next = url_for('admin_document.manage', doc_name = doc_name, branch = branch))
+	return render_template("admin/command_output.html", task = build_task, next = url_for('admin_document.manage', origin = doc.encoded_origin, doc_name = doc.doc_name, branch = doc.branch))
 
-@bp.route('/delete/<doc_name>/<branch>')
-def delete(doc_name, branch):
-	doc = Document(doc_name, branch)
+@bp.route('/delete/<origin>/<doc_name>/<branch>')
+def delete(origin, doc_name, branch):
+	doc = Document(origin, doc_name, branch)
 	doc.delete()
 	return redirect(url_for('admin.index'), code=302)
 
-@bp.route('/delete_invalid/<doc_name>/<branch>')
-def delete_invalid(doc_name, branch):
-	doc = Document(doc_name, branch, allow_invalid = True)
+@bp.route('/delete_invalid/<origin>/<doc_name>/<branch>')
+def delete_invalid(origin, doc_name, branch):
+	doc = Document(origin, doc_name, branch, allow_invalid = True)
 	doc.delete_folder()
 	return redirect(url_for('admin.index'), code=302)

src/web/web_document.py

@@ -6,7 +6,7 @@ from data.document import Document
 
 bp = Blueprint('web_document', __name__, url_prefix='/doc')
 
-@bp.route('/<doc_name>/<branch>/<path:path>')
-def index(doc_name, branch, path):
-	doc = Document(doc_name, branch)
+@bp.route('/<origin>/<doc_name>/<branch>/<path:path>')
+def index(origin, doc_name, branch, path):
+	doc = Document(origin, doc_name, branch)
 	return send_from_directory(doc.doc_path + "/build/html", path)