add: auto link feature

api/markdown.py

@@ -1,6 +1,5 @@
 from flask import Blueprint, request, jsonify
 from sqlalchemy import or_
-
 import api
 from api import require_auth, etag_response
 from contexts.RequestContext import RequestContext
@@ -29,6 +28,7 @@ def get_home():
         if markdown is None:
             return jsonify({}), 204
         return jsonify(markdown.to_dict()), 200
+
 @markdown_bp.route('/by_path/<int:path_id>', methods=['GET'])
 @limiter.limit(api.get_rate_limit)
 @etag_response
@@ -36,6 +36,7 @@ def get_markdowns_by_path(path_id):
     with get_db() as session:
         markdowns = session.query(Markdown).filter(Markdown.path_id == path_id).all()
         return jsonify([md.to_dict() for md in markdowns]), 200
+
 @markdown_bp.route('/get_index/<int:path_id>', methods=['GET'])
 @limiter.limit(api.get_rate_limit)
 @etag_response
@@ -66,11 +67,16 @@ def create_markdown():
     title = data.get('title')
     content = data.get('content')
     path_id = data.get('path_id')
+    shortcut = data.get('shortcut', "")
     if not title or not content:
         return jsonify({"error": "missing required fields"}), 400
-    new_markdown = Markdown(title=title, content=content, path_id=path_id)
+    new_markdown = Markdown(title=title, content=content, path_id=path_id, shortcut=shortcut)
     with get_db() as session:
         try:
+            if shortcut != "":
+                r = session.query(Markdown).filter(Markdown.shortcut == shortcut).all()
+                if len(r) > 0:
+                    return jsonify({"error": "duplicate shortcut"}), 400
             session.add(new_markdown)
             session.commit()
             return jsonify(new_markdown.to_dict()), 201
@@ -89,10 +95,19 @@ def update_markdown(markdown_id):
         if markdown is None:
             return jsonify({"error": "file not found"}), 404
         data = request.json
+        if data.get('shortcut', "") != "":
+            r = session.query(Markdown).filter(
+                Markdown.shortcut == data.get('shortcut')
+            ).filter(
+                Markdown.id != markdown_id
+            ).all()
+            if len(r) > 0:
+                return jsonify({"error": "duplicate shortcut"}), 400
         if request.method == "PUT":
             markdown.title = data.get('title')
             markdown.content = data.get('content')
             markdown.path_id = data.get('path_id')
+            markdown.shortcut = data.get('shortcut', '')
         elif request.method == "PATCH":
             if 'title' in data:
                 markdown.title = data.get('title')
@@ -100,6 +115,8 @@ def update_markdown(markdown_id):
                 markdown.content = data.get('content')
             if 'path_id' in data:
                 markdown.path_id = data.get('path_id')
+            if 'shortcut' in data:
+                markdown.shortcut = data.get('shortcut')
         session.commit()
         return jsonify(markdown.to_dict()), 200
 
@@ -157,8 +174,16 @@ def move_backward(markdown_id):
 @markdown_bp.route('/search/<string:keyword>', methods=['GET'])
 @limiter.limit(api.get_rate_limit)
 def search_markdowns(keyword):
-    with (get_db() as session):
+    with get_db() as session:
         res = session.query(Markdown).filter(
             or_(Markdown.title.like(keyword), Markdown.content.like(keyword))
         ).all()
         return jsonify([md.to_dict() for md in res]), 200
+
+@markdown_bp.route('/links', methods=['GET'])
+@limiter.limit(api.get_rate_limit)
+def get_links():
+    with get_db() as session:
+        mds = [md.to_dict() for md in session.query(Markdown).filter(Markdown.shortcut != "").all()]
+        links = [f"[{md['shortcut']}]: {md['id']}" for md in mds]
+        return jsonify(links), 200