Add url shortener

Author: oreilles

.env.ci

@@ -11,6 +11,8 @@ QSL_DATA_ROOT=/io/data
 
 GEORAMA_DATA_INTEGRATION_ROOT=/io/data
 
+GEORAMA_WEBGISURL=https://localhost:9309
+
 GEORAMA_DB_PW=test
 GEORAMA_DB_USER=postgres
 GEORAMA_DB_PORT=54321

geogirafe/config.json

@@ -38,8 +38,8 @@
     "url": "https://empty"
   },
   "share": {
-    "service": "",
-    "createUrl": "https://empty"
+    "service": "gmf",
+    "createUrl": "http://localhost:4242/webgis/short/create"
   },
   "projections": {
     "EPSG:3857": "W-M",

src/georama/core/settings.py

@@ -37,7 +37,7 @@ class Base(Configuration):
     # SECURITY WARNING: don't run with debug turned on in production!
     DEBUG = values.BooleanValue(False, environ_prefix="GEORAMA")
 
-    WEBGISURL = values.Value(environ_prefix="GEORAMA")
+    WEBGISURL = values.Value("https://localhost:9309", environ_prefix="GEORAMA")
     SITE_TITLE = values.Value("demo.georama.io", environ_prefix="GEORAMA")
 
     LOGGING = {

src/georama/data_integration/models.py

@@ -6,9 +6,15 @@
 from django.db import models
 from django.utils.translation import gettext as _
 from qgis_server_light.interface.common import BBox
-from qgis_server_light.interface.exporter.extract import Crs, Custom, DataSource
+from qgis_server_light.interface.exporter.extract import (
+    Crs,
+    Custom,
+    DataSource,
+    Raster,
+    Style,
+    Vector,
+)
 from qgis_server_light.interface.exporter.extract import Field as QslField
-from qgis_server_light.interface.exporter.extract import Raster, Style, Vector
 from qgis_server_light.interface.job.common.input import QslJobLayer
 from xsdata.formats.dataclass.parsers import DictDecoder
 from xsdata.formats.dataclass.parsers.config import ParserConfig

src/georama/webgis/migrations/0024_urlshortener.py

@@ -0,0 +1,20 @@
+# Generated by Django 5.2.10 on 2026-03-16 15:00
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("webgis", "0023_publishedastheme_location_publishedastheme_zoom"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="UrlShortener",
+            fields=[
+                ("id", models.CharField(primary_key=True, serialize=False)),
+                ("url", models.CharField()),
+            ],
+        ),
+    ]

src/georama/webgis/models/__init__.py

@@ -1,3 +1,4 @@
+from georama.webgis.models.short import UrlShortener
 from georama.webgis.models.themes import (
     Interface,
     Layer,
@@ -16,4 +17,5 @@
     "PublishedAsLayerWms",
     "PublishedAsLayerWmts",
     "PublishedAsTheme",
+    "UrlShortener",
 ]

src/georama/webgis/models/short.py

@@ -0,0 +1,6 @@
+from django.db import models
+
+
+class UrlShortener(models.Model):
+    id = models.CharField(primary_key=True)
+    url = models.CharField()

src/georama/webgis/urls.py

@@ -79,4 +79,8 @@
         views.translation_json,
         name="translation_json",
     ),
+    path(
+        "/short/get/<str:id>", views.UrlShortenerRetrieve.as_view(), name="get_shortened_url"
+    ),
+    path("/short/create", views.UrlShortenerCreate.as_view(), name="shorten_url"),
 ]

src/georama/webgis/views.py

@@ -4,9 +4,10 @@
 from django.conf import settings
 from django.contrib.auth.mixins import PermissionRequiredMixin
 from django.contrib.auth.models import User
-from django.db import transaction
-from django.http import HttpRequest, HttpResponse, HttpResponseNotFound
+from django.db import IntegrityError, transaction
+from django.http import HttpRequest, HttpResponse, HttpResponseNotFound, JsonResponse
 from django.shortcuts import redirect, render
+from django.utils.crypto import get_random_string
 from django.utils.http import url_has_allowed_host_and_scheme
 from django.views import View
 from qgis_server_light.interface.common import BBox
@@ -46,7 +47,12 @@
     Theme,
     ThemesJson,
 )
-from georama.webgis.models import LayerGroupMp, PublishedAsLayerWms, PublishedAsTheme
+from georama.webgis.models import (
+    LayerGroupMp,
+    PublishedAsLayerWms,
+    PublishedAsTheme,
+    UrlShortener,
+)
 from georama.webgis.models import OgcServer as WebGisOgcServer
 
 
@@ -616,3 +622,33 @@ def translation_json(request: HttpRequest):
     return HttpResponse(
         json.dumps(translation, indent=2), status=200, content_type="application/json"
     )
+
+
+class UrlShortenerCreate(View):
+    def post(self, request: HttpRequest):
+        url = request.POST.get("url", "").strip()
+        if not url:
+            return JsonResponse({"error": "Missing URL."}, status=400)
+
+        if not url.startswith(settings.WEBGISURL):
+            return JsonResponse({"error": "Invalid URL."}, status=400)
+
+        while True:
+            try:
+                short = UrlShortener.objects.create(id=get_random_string(length=6), url=url)
+                break
+            except IntegrityError:
+                continue
+
+        short_url = request.build_absolute_uri(f"/webgis/short/get/{short.id}")
+        return JsonResponse({"short_url": short_url}, status=201)
+
+
+class UrlShortenerRetrieve(View):
+    def get(self, _request: HttpRequest, id: str):
+        try:
+            short = UrlShortener.objects.get(id=id)
+        except UrlShortener.DoesNotExist:
+            return JsonResponse({"error": "Not found."}, status=404)
+
+        return JsonResponse({"long_url": short.url}, status=200)

tests/unit/webgis/test_views.py

@@ -3,6 +3,7 @@
 import pytest
 
 from georama.webgis.models import PublishedAsLayerWms
+from django.conf import settings
 
 pytestmark = pytest.mark.django_db
 
@@ -37,3 +38,33 @@ def test_publish_as_wms_view(
             assert response.status_code == 200
             assert b"TestPointLayer" in response.content
             assert PublishedAsLayerWms.objects.filter(title="TestPointLayer").exists()
+
+
+class TestShortUrlViews:
+    def test_shorten_url_create_and_get(self, client):
+        url = f"{settings.WEBGISURL}/1234"
+        response_create = client.post("/webgis/short/create", data={"url": url})
+
+        assert response_create.status_code == 201, response_create.json()
+        payload_create = response_create.json()
+        assert "short_url" in payload_create
+
+        short_id = payload_create["short_url"].rstrip("/").split("/")[-1]
+        response_get = client.get(f"/webgis/short/get/{short_id}")
+
+        assert response_get.status_code == 200, response_create.json()
+
+        payload_get = response_get.json()
+        assert payload_get["long_url"] == url
+
+    def test_shorten_url_requires_webgis_url(self, client):
+        response = client.post("/webgis/short/create", data={"url": "https://example.org/foobar"})
+
+        assert response.status_code == 400
+        assert response.json()["error"] == "Invalid URL."
+
+    def test_get_long_url_not_found(self, client):
+        response = client.get("/webgis/short/get/does-not-exist")
+
+        assert response.status_code == 404
+        assert response.json()["error"] == "Not found."