feat: add support for importing dns records

carlalexander · carlalexander · commit 1922720bb8fc · 2020-11-19T01:53:01.000-05:00
diff --git a/phpmd.xml b/phpmd.xml
@@ -12,6 +12,7 @@
         <exclude name="StaticAccess" />
     </rule>
     <rule ref="rulesets/codesize.xml">
+        <exclude name="ExcessiveClassComplexity" />
         <exclude name="TooManyPublicMethods"/>
     </rule>
     <rule ref="rulesets/controversial.xml" />
diff --git a/src/ApiClient.php b/src/ApiClient.php
@@ -678,6 +678,18 @@ public function getUser(): Collection
         return $this->request('get', '/user');
     }
 
+    /**
+     * Change the value of the DNS record in the given DNS zone ID or name.
+     */
+    public function importDnsRecord($zoneIdOrName, array $subdomains = [])
+    {
+        $zone = $this->getDnsZone($zoneIdOrName);
+
+        $this->request('post', "/zones/{$zone['id']}/import-records", [
+            'subdomains' => array_filter($subdomains),
+        ]);
+    }
+
     /**
      * Invalidate the content delivery network cache for the given project environment.
      */
diff --git a/src/Command/AbstractCommand.php b/src/Command/AbstractCommand.php
@@ -99,6 +99,22 @@ protected function execute(InputInterface $input, OutputInterface $output)
         return $this->perform($input, new ConsoleOutput($input, $output)) ?? 0;
     }
 
+    /**
+     * Get the value of an argument that should be an array.
+     */
+    protected function getArrayArgument(InputInterface $input, string $argument, bool $requiredNonInteractive = true): array
+    {
+        $value = $input->getArgument($argument);
+
+        if (null === $value && $requiredNonInteractive && !$input->isInteractive()) {
+            throw new InvalidArgumentException(sprintf('You must pass a "%s" argument when running in non-interactive mode', $argument));
+        } elseif (null !== $value && !is_array($value)) {
+            throw new InvalidArgumentException(sprintf('The "%s" argument must be an array value', $argument));
+        }
+
+        return (array) $value;
+    }
+
     /**
      * Get the value of a option that should be an array.
      */
diff --git a/src/Command/Dns/CreateDnsZoneCommand.php b/src/Command/Dns/CreateDnsZoneCommand.php
@@ -71,5 +71,15 @@ protected function perform(InputInterface $input, ConsoleOutput $output)
         }
 
         $output->info('DNS zone created');
+
+        if ($output->confirm('Do you want to import the root DNS records for this domain', false)) {
+            $this->apiClient->importDnsRecord($zone['id']);
+        }
+
+        if ($output->confirm('Do you want to import DNS records for subdomains of this domain', false)) {
+            $this->invoke($output, ImportDnsRecordsCommand::NAME, [
+                'zone' => $zone['domain_name'],
+            ]);
+        }
     }
 }
diff --git a/src/Command/Dns/ImportDnsRecordsCommand.php b/src/Command/Dns/ImportDnsRecordsCommand.php
@@ -0,0 +1,56 @@
+<?php
+
+declare(strict_types=1);
+
+/*
+ * This file is part of Ymir command-line tool.
+ *
+ * (c) Carl Alexander <support@ymirapp.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Ymir\Cli\Command\Dns;
+
+use Symfony\Component\Console\Input\InputArgument;
+use Symfony\Component\Console\Input\InputInterface;
+use Ymir\Cli\Console\ConsoleOutput;
+
+class ImportDnsRecordsCommand extends AbstractDnsCommand
+{
+    /**
+     * The name of the command.
+     *
+     * @var string
+     */
+    public const NAME = 'dns:zone:import-records';
+
+    /**
+     * {@inheritdoc}
+     */
+    protected function configure()
+    {
+        $this
+            ->setName(self::NAME)
+            ->setDescription('Import')
+            ->addArgument('zone', InputArgument::REQUIRED, 'The name of the DNS zone that the DNS record belongs to')
+            ->addArgument('subdomain', InputArgument::IS_ARRAY | InputArgument::OPTIONAL, 'The subdomains that we want to import');
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    protected function perform(InputInterface $input, ConsoleOutput $output)
+    {
+        $subdomains = $this->getArrayArgument($input, 'subdomain', false);
+
+        if (empty($subdomains)) {
+            $subdomains = explode(',', (string) $output->ask('Please enter a comma-separated list of subdomains to import DNS records from (leave blank to import the root DNS records)'));
+        }
+
+        $this->apiClient->importDnsRecord($this->getStringArgument($input, 'zone'), $subdomains);
+
+        $output->info('DNS records imported');
+    }
+}

Original file line number	Diff line number	Diff line change
`@@ -71,5 +71,15 @@ protected function perform(InputInterface $input, ConsoleOutput $output)`
`71`	`71`	`}`
`72`	`72`
`73`	`73`	`$output->info('DNS zone created');`
	`74`	`+`
	`75`	`+ if ($output->confirm('Do you want to import the root DNS records for this domain', false)) {`
	`76`	`+ $this->apiClient->importDnsRecord($zone['id']);`
	`77`	`+ }`
	`78`	`+`
	`79`	`+ if ($output->confirm('Do you want to import DNS records for subdomains of this domain', false)) {`
	`80`	`+ $this->invoke($output, ImportDnsRecordsCommand::NAME, [`
	`81`	`+ 'zone' => $zone['domain_name'],`
	`82`	`+ ]);`
	`83`	`+ }`
`74`	`84`	`}`
`75`	`85`	`}`