From 39798d1fb0a64829c6b8c475100d9c2be72d6bcc Mon Sep 17 00:00:00 2001
From: vincent-gao <vincent.gao@dpc.vic.gov.au>
Date: Wed, 27 Mar 2024 13:32:00 +1100
Subject: [PATCH] fix: [SDPAP-8510] custom url issue (#147)

---
 .../tide_site_simple_sitemap.module           | 32 +++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/modules/tide_site_simple_sitemap/tide_site_simple_sitemap.module b/modules/tide_site_simple_sitemap/tide_site_simple_sitemap.module
index 6e5594c..ea6a5dd 100644
--- a/modules/tide_site_simple_sitemap/tide_site_simple_sitemap.module
+++ b/modules/tide_site_simple_sitemap/tide_site_simple_sitemap.module
@@ -7,6 +7,7 @@
 
 use Drupal\simple_sitemap\Entity\SimpleSitemap;
 use Drupal\simple_sitemap\Entity\SimpleSitemapInterface;
+use Drupal\taxonomy\Entity\Term;
 
 /**
  * Implements hook_ENTITY_TYPE_delete().
@@ -29,3 +30,34 @@ function tide_site_simple_sitemap_taxonomy_term_delete($term) {
     watchdog_exception('tide_site_simple_sitemap', $exception);
   }
 }
+
+/**
+ * Implements hook_ENTITY_TYPE_insert().
+ */
+function tide_site_simple_sitemap_simple_sitemap_insert(SimpleSitemap $entity) {
+  $sitemap_id =$entity->id();
+  if ($term_id = _tide_site_simple_sitemap_extract_site_number($sitemap_id)){
+    if(Term::load($term_id) && $entity->getType()->getOriginalId() === 'tide_default_sitemap_type'){
+      /** @var Drupal\simple_sitemap\Manager\Generator $custom_generator */
+      $custom_generator = \Drupal::service('simple_sitemap.generator');
+      $custom_generator->setSitemaps($entity->id());
+      $site_based_custom_link = $custom_generator->customLinkManager()->get('/');
+      if (isset($site_based_custom_link[$sitemap_id]) && empty($site_based_custom_link[$sitemap_id])){
+        $custom_generator->customLinkManager()->add(
+          '/',
+          ["path" => "/", "priority" => "1.0", "changefreq" => "daily"]
+        );
+      }
+    }
+  }
+}
+
+/**
+ * Extract site number from the sitemap id.
+ */
+function _tide_site_simple_sitemap_extract_site_number($string) {
+  if (preg_match('/^site-(\d+)$/', $string, $matches)) {
+    return $matches[1];
+  }
+  return null;
+}