Call source builder API when source code is provided

Author: willdurand

src/olympia/lib/settings_base.py

@@ -833,6 +833,7 @@ def get_language_url_map():
     'olympia.scanners.tasks.run_customs': {'queue': 'devhub'},
     'olympia.scanners.tasks.run_narc_on_version': {'queue': 'devhub'},
     'olympia.scanners.tasks.run_yara': {'queue': 'devhub'},
+    'olympia.versions.tasks.call_source_builder': {'queue': 'devhub'},
     'olympia.versions.tasks.soft_block_versions': {'queue': 'devhub'},
     # Crons.
     'olympia.addons.tasks.update_addon_average_daily_users': {'queue': 'cron'},
@@ -1557,3 +1558,7 @@ def read_only_mode(env):
 SWAGGER_SCHEMA_FILE = path('schema.yml')
 
 SWAGGER_UI_ENABLED = env('SWAGGER_UI_ENABLED', default=False) or TARGET != 'production'
+
+# Source builder settings.
+SOURCE_BUILDER_API_URL = env('SOURCE_BUILDER_API_URL', default=None)
+SOURCE_BUILDER_API_TIMEOUT = 5  # seconds

src/olympia/versions/migrations/0051_create_enable_source_builder_switch.py

@@ -0,0 +1,13 @@
+# Generated by Django 4.2.26 on 2025-11-13 08:55
+
+from django.db import migrations
+from olympia.core.db.migrations import CreateWaffleSwitch
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('versions', '0050_remove_versionreviewerflags_needs_human_review_by_mad'),
+    ]
+
+    operations = [CreateWaffleSwitch('enable-source-builder')]

src/olympia/versions/models.py

@@ -826,6 +826,8 @@ def flag_if_sources_were_provided(self, user):
         from olympia.activity.utils import log_and_notify
         from olympia.reviewers.models import NeedsHumanReview
 
+        from .tasks import call_source_builder
+
         if self.source:
             # Add Activity Log, notifying staff, relevant reviewers and
             # other authors of the add-on.
@@ -835,6 +837,9 @@ def flag_if_sources_were_provided(self, user):
                 reason = NeedsHumanReview.REASONS.PENDING_REJECTION_SOURCES_PROVIDED
                 NeedsHumanReview.objects.create(version=self, reason=reason)
 
+            if waffle.switch_is_active('enable-source-builder'):
+                call_source_builder.delay(version_pk=self.pk)
+
     @classmethod
     def transformer(cls, versions):
         """Attach all the compatible apps and the file to the versions."""

src/olympia/versions/tasks.py

@@ -8,6 +8,7 @@
 from django.db import transaction
 from django.template import loader
 
+import requests
 from django_statsd.clients import statsd
 from PIL import Image
 
@@ -24,6 +25,7 @@
 from olympia.files.utils import get_background_images
 from olympia.lib.crypto.tasks import duplicate_addon_version
 from olympia.reviewers.models import NeedsHumanReview
+from olympia.scanners.tasks import make_adapter_with_retry
 from olympia.users.models import UserProfile
 from olympia.users.utils import get_task_user
 from olympia.versions.compare import VersionString
@@ -416,3 +418,32 @@ def soft_block_versions(version_ids, reason=REASON_VERSION_DELETED, **kw):
         ),
         overwrite_block_metadata=False,
     )
+
+
+@task
+def call_source_builder(version_pk):
+    log.info('Calling source builder API for Version %s', version_pk)
+
+    try:
+        version = Version.objects.get(pk=version_pk)
+
+        with requests.Session() as http:
+            adapter = make_adapter_with_retry()
+            http.mount('http://', adapter)
+            http.mount('https://', adapter)
+
+            json_payload = {
+                'addon_guid': version.addon.guid,
+                'version': version.version,
+                'download_source_url': '',
+                'license_slug': version.license.slug,
+            }
+            http.post(
+                url=settings.SOURCE_BUILDER_API_URL,
+                json=json_payload,
+                timeout=settings.SOURCE_BUILDER_API_TIMEOUT,
+            )
+    except Exception:
+        log.exception(
+            'Error while calling source builder API for Version %s.', version_pk
+        )