NAS-137693 / 25.10-RC.1 / Moving tn_connect urls from staging to production (by sonicaj)

src/middlewared/middlewared/alembic/versions/25.10/2025-09-24_21-11_tnc_prod_endpoints.py

@@ -0,0 +1,59 @@
+"""
+Moving tn_connect url from staging to production
+
+Revision ID: 53193fbee3ee
+Revises: 5ea9f662ced4
+Create Date: 2025-09-24 16:44:06.551471+00:00
+"""
+from alembic import op
+
+
+# revision identifiers, used by Alembic.
+revision = '53193fbee3ee'
+down_revision = '5ea9f662ced4'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    conn = op.get_bind()
+
+    production_urls = {
+        'account_service_base_url': 'https://account-service.tys1.truenasconnect.net/',
+        'leca_service_base_url': 'https://dns-service.tys1.truenasconnect.net/',
+        'tnc_base_url': 'https://web.truenasconnect.net/',
+        'heartbeat_url': 'https://heartbeat-service.tys1.truenasconnect.net/'
+    }
+
+    result = conn.execute(
+        'SELECT id, account_service_base_url, leca_service_base_url, tnc_base_url, heartbeat_url FROM truenas_connect'
+    ).fetchall()
+
+    if not result:
+        return
+
+    row = result[0]
+    row_id = row[0]
+    current_urls = {
+        'account_service_base_url': row[1],
+        'leca_service_base_url': row[2],
+        'tnc_base_url': row[3],
+        'heartbeat_url': row[4]
+    }
+
+    all_non_prod = all(
+        current_urls.get(column) and ('.dev.' in current_urls[column] or '.staging.' in current_urls[column])
+        for column in production_urls.keys()
+    )
+
+    if all_non_prod:
+        set_clauses = []
+        for column, new_url in production_urls.items():
+            set_clauses.append(f"{column} = {new_url!r}")
+
+        update_sql = f"UPDATE truenas_connect SET {', '.join(set_clauses)} WHERE id = {row_id}"
+        conn.execute(update_sql)
+
+
+def downgrade():
+    pass

src/middlewared/middlewared/plugins/truenas_connect/update.py

@@ -35,16 +35,16 @@ class TrueNASConnectModel(sa.Model):
     status = sa.Column(sa.String(255), default=Status.DISABLED.name, nullable=False)
     certificate_id = sa.Column(sa.ForeignKey('system_certificate.id'), index=True, nullable=True)
     account_service_base_url = sa.Column(
-        sa.String(255), nullable=False, default='https://account-service.staging.truenasconnect.net/'
+        sa.String(255), nullable=False, default='https://account-service.tys1.truenasconnect.net/'
     )
     leca_service_base_url = sa.Column(
-        sa.String(255), nullable=False, default='https://dns-service.staging.truenasconnect.net/'
+        sa.String(255), nullable=False, default='https://dns-service.tys1.truenasconnect.net/'
     )
     tnc_base_url = sa.Column(
-        sa.String(255), nullable=False, default='https://web.staging.truenasconnect.net/'
+        sa.String(255), nullable=False, default='https://web.truenasconnect.net/'
     )
     heartbeat_url = sa.Column(
-        sa.String(255), nullable=False, default='https://heartbeat-service.staging.truenasconnect.net/'
+        sa.String(255), nullable=False, default='https://heartbeat-service.tys1.truenasconnect.net/'
     )
     last_heartbeat_failure_datetime = sa.Column(sa.String(255), nullable=True, default=None)