From 9d585b8ff7b650aa8d256277ed28a8b3116a8bda Mon Sep 17 00:00:00 2001
From: Riddhesh Sanghvi <riddheshsanghvi96@gmail.com>
Date: Wed, 10 Oct 2018 22:40:14 +0530
Subject: [PATCH 1/2] Add cron cleanup hook

Signed-off-by: Riddhesh Sanghvi <riddheshsanghvi96@gmail.com>
---
 src/helper/hooks.php | 62 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 src/helper/hooks.php

diff --git a/src/helper/hooks.php b/src/helper/hooks.php
new file mode 100644
index 0000000..98db627
--- /dev/null
+++ b/src/helper/hooks.php
@@ -0,0 +1,62 @@
+<?php
+
+if ( ! class_exists( 'EE' ) ) {
+	return;
+}
+
+use EE\Model\Cron;
+use EE\Model\Site;
+
+/**
+ * Hook to cleanup cron entries if any.
+ *
+ * @param string $site_url The site to be cleaned up.
+ */
+function cleanup_cron_entries( $site_url ) {
+
+	if ( ! Site::find( $site_url ) ) {
+		return;
+	}
+
+	$cron_jobs = Cron::where( [ 'site_url' => $site_url ] );
+
+	if ( ! empty( $cron_jobs ) ) {
+		foreach ( $cron_jobs as $cron_job ) {
+			$cron_job->delete();
+		}
+		regenerate_cron_config();
+	}
+}
+
+/**
+ * Regenerates cron config from DB.
+ */
+function regenerate_cron_config() {
+
+	$config_template = file_get_contents( __DIR__ . '/../../templates/config.ini.mustache' );
+	$crons           = Cron::all();
+
+	if ( empty( $cron ) ) {
+		EE::exec( 'docker rm -f ' . EE_CRON_SCHEDULER );
+	}
+
+	foreach ( $crons as &$cron ) {
+		$job_type       = 'host' === $cron->site_url ? 'job-local' : 'job-exec';
+		$id             = $cron->site_url . '-' . preg_replace( '/[^a-zA-Z0-9\@]/', '-', $cron->command ) . '-' . EE\Utils\random_password( 5 );
+		$id             = preg_replace( '/--+/', '-', $id );
+		$cron->job_type = $job_type;
+		$cron->id       = $id;
+
+		if ( 'host' !== $cron->site_url ) {
+			$cron->container = str_replace( '.', '', $cron->site_url ) . '_php_1';
+		}
+	}
+
+	$me     = new Mustache_Engine();
+	$config = $me->render( $config_template, $crons );
+
+	file_put_contents( EE_ROOT_DIR . '/services/cron/config.ini', $config );
+	EE_DOCKER::restart_container( EE_CRON_SCHEDULER );
+}
+
+EE::add_hook( 'site_cleanup', 'cleanup_cron_entries' );

From 525aded4f7ba9b59a2383f5ce612fcb50053146e Mon Sep 17 00:00:00 2001
From: Riddhesh Sanghvi <riddheshsanghvi96@gmail.com>
Date: Wed, 10 Oct 2018 22:40:22 +0530
Subject: [PATCH 2/2] Add loading of hook file

Signed-off-by: Riddhesh Sanghvi <riddheshsanghvi96@gmail.com>
---
 composer.json | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/composer.json b/composer.json
index d0a1b3c..dc62481 100644
--- a/composer.json
+++ b/composer.json
@@ -11,7 +11,10 @@
         "psr-4": {
             "": "src/"
         },
-        "files": [ "cron-command.php" ]
+        "files": [
+            "cron-command.php",
+            "src/helper/hooks.php"
+        ]
     },
     "extra": {
         "branch-alias": {