LinkTool

Author: 2ik

django_editorjs_fields/config.py

@@ -65,7 +65,13 @@
         'Embed': {'class': 'Embed'},
         'Delimiter': {'class': 'Delimiter'},
         'Warning': {'class': 'Warning', 'inlineToolbar': True},
-        'LinkTool': {'class': 'LinkTool'},
+        'LinkTool': {
+            'class': 'LinkTool',
+            'config': {
+                # Backend endpoint for url data fetching
+                'endpoint': reverse_lazy('editorjs_linktool'),
+            }
+        },
         'Marker': {'class': 'Marker', 'inlineToolbar': True},
         'Table': {'class': 'Table', 'inlineToolbar': True},
     }

django_editorjs_fields/templatetags/editorjs.py

@@ -99,6 +99,34 @@ def generate_embed(data):
     return f'<div class="embed {service}">{iframe}{caption}</div>'
 
 
+def generate_link(data):
+
+    link, meta = data.get('link'), data.get('meta')
+
+    if not link or not meta:
+        return ''
+
+    title = meta.get('title')
+    description = meta.get('description')
+    image = meta.get('image')
+
+    wrapper = f'<div class="link-block"><a href="{ link }" target="_blank" rel="nofollow noopener noreferrer">'
+
+    if image.get('url'):
+        image_url = image.get('url')
+        wrapper += f'<div class="link-block__image" style="background-image: url(\'{image_url}\');"></div>'
+
+    if title:
+        wrapper += f'<p class="link-block__title">{title}</p>'
+
+    if description:
+        wrapper += f'<p class="link-block__description">{description}</p>'
+
+    wrapper += f'<p class="link-block__link">{link}</p>'
+    wrapper += '</a></div>'
+    return wrapper
+
+
 @register.filter(is_safe=True)
 def editorjs(value):
     if not value or value == 'null':
@@ -139,5 +167,7 @@ def editorjs(value):
             html_list.append(generate_embed(data))
         elif type == 'Quote':
             html_list.append(generate_quote(data))
+        elif type == 'LinkTool':
+            html_list.append(generate_link(data))
 
     return mark_safe(''.join(html_list))

django_editorjs_fields/urls.py

@@ -1,12 +1,17 @@
 from django.contrib.admin.views.decorators import staff_member_required
 from django.urls import path
 
-from .views import ImageUploadView
+from .views import ImageUploadView, LinkToolView
 
 urlpatterns = [
     path(
         'image_upload/',
         staff_member_required(ImageUploadView.as_view()),
         name='editorjs_image_upload',
     ),
+    path(
+        'linktool/',
+        staff_member_required(LinkToolView.as_view()),
+        name='editorjs_linktool',
+    ),
 ]

django_editorjs_fields/views.py

@@ -1,6 +1,14 @@
+import json
+import logging
 import os
 from datetime import datetime
+from urllib.error import HTTPError, URLError
+from urllib.parse import urlencode
+from urllib.request import Request, urlopen
 
+# from django.conf import settings
+from django.core.exceptions import ValidationError
+from django.core.validators import URLValidator
 from django.http import JsonResponse
 from django.utils.decorators import method_decorator
 from django.views import View
@@ -10,9 +18,12 @@
                      IMAGE_UPLOAD_PATH_DATE)
 from .utils import storage
 
+LOGGER = logging.getLogger('django_editorjs_fields')
+
 
 class ImageUploadView(View):
     http_method_names = ["post"]
+    # http_method_names = ["post", "delete"]
 
     @method_decorator(csrf_exempt)
     def dispatch(self, request, *args, **kwargs):
@@ -35,9 +46,6 @@ def post(self, request):
                     {'success': 0, 'message': 'You can only upload images.'}
                 )
 
-            # filesize = len(the_file['content'])
-            # filetype = the_file['content-type']
-
             filename, extension = os.path.splitext(the_file.name)
 
             if IMAGE_NAME_ORIGINAL is False:
@@ -57,3 +65,82 @@ def post(self, request):
 
             return JsonResponse({'success': 1, 'file': {"url": link}})
         return JsonResponse({'success': 0})
+
+    # def delete(self, request):
+    #     path_file = request.GET.get('pathFile')
+
+    #     if not path_file:
+    #         return JsonResponse({'success': 0, 'message': 'Parameter "pathFile" Not Found'})
+
+    #     base_dir = getattr(settings, "BASE_DIR", '')
+    #     path_file = f'{base_dir}{path_file}'
+
+    #     if not os.path.isfile(path_file):
+    #         return JsonResponse({'success': 0, 'message': 'File Not Found'})
+
+    #     os.remove(path_file)
+
+    #     return JsonResponse({'success': 1})
+
+
+class LinkToolView(View):
+    http_method_names = ["get"]
+
+    @method_decorator(csrf_exempt)
+    def dispatch(self, request, *args, **kwargs):
+        return super().dispatch(request, *args, **kwargs)
+
+    def get(self, request):
+
+        url = request.GET.get('url', '')
+
+        LOGGER.debug('Starting to get meta for: %s', url)
+
+        if not any([url.startswith(s) for s in ('http://', 'https://')]):
+            LOGGER.debug('Adding the http protocol to the link: %s', url)
+            url = 'http://' + url
+
+        validate = URLValidator(schemes=['http', 'https'])
+
+        try:
+            validate(url)
+        except ValidationError as e:
+            LOGGER.error(e)
+        else:
+            try:
+                LOGGER.debug('Let\'s try to get meta from: %s', url)
+
+                full_url = 'https://api.microlink.io/?' + \
+                    urlencode({'url': url})
+
+                req = Request(full_url, headers={
+                    'User-Agent': request.META.get('HTTP_USER_AGENT', 'Mozilla/5.0 (Windows NT 6.1; Win64; x64)')
+                })
+                res = urlopen(req)
+            except HTTPError as e:
+                LOGGER.error('The server couldn\'t fulfill the request.')
+                LOGGER.error('Error code: %s %s', e.code, e.msg)
+            except URLError as e:
+                LOGGER.error('We failed to reach a server. url: %s', url)
+                LOGGER.error('Reason: %s', e.reason)
+            else:
+                res_body = res.read()
+                res_json = json.loads(res_body.decode("utf-8"))
+
+                if 'success' in res_json.get('status'):
+                    data = res_json.get('data')
+
+                    if data:
+                        LOGGER.debug('Response meta: %s', data)
+                        meta = {}
+                        meta['title'] = data.get('title')
+                        meta['description'] = data.get('description')
+                        meta['image'] = data.get('image')
+
+                        return JsonResponse({
+                            'success': 1,
+                            'link': data.get('url', url),
+                            'meta': meta
+                        })
+
+        return JsonResponse({'success': 0})