Handle key generation when key is not present in .env

Author: paulbalandan

system/Commands/Encryption/GenerateKey.php

@@ -67,13 +67,15 @@ class GenerateKey extends BaseCommand
     public function run(array $params)
     {
         $prefix = $params['prefix'] ?? CLI::getOption('prefix');
+
         if (in_array($prefix, [null, true], true)) {
             $prefix = 'hex2bin';
         } elseif (! in_array($prefix, ['hex2bin', 'base64'], true)) {
             $prefix = CLI::prompt('Please provide a valid prefix to use.', ['hex2bin', 'base64'], 'required'); // @codeCoverageIgnore
         }
 
         $length = $params['length'] ?? CLI::getOption('length');
+
         if (in_array($length, [null, true], true)) {
             $length = 32;
         }
@@ -127,9 +129,7 @@ protected function setNewEncryptionKey(string $key, array $params): bool
 
         if ($currentKey !== '' && ! $this->confirmOverwrite($params)) {
             // Not yet testable since it requires keyboard input
-            // @codeCoverageIgnoreStart
-            return false;
-            // @codeCoverageIgnoreEnd
+            return false; // @codeCoverageIgnore
         }
 
         return $this->writeNewEncryptionKeyToFile($currentKey, $key);
@@ -163,13 +163,18 @@ protected function writeNewEncryptionKeyToFile(string $oldKey, string $newKey):
             copy($baseEnv, $envFile);
         }
 
-        $ret = file_put_contents($envFile, preg_replace(
-            $this->keyPattern($oldKey),
-            "\nencryption.key = {$newKey}",
-            file_get_contents($envFile)
-        ));
+        $envFileContents = (string) file_get_contents($envFile);
+        $replacementKey  = "\nencryption.key = {$newKey}";
 
-        return $ret !== false;
+        if (strpos($envFileContents, 'encryption.key') === false) {
+            return file_put_contents($envFile, $replacementKey, FILE_APPEND) !== false;
+        }
+
+        return file_put_contents($envFile, preg_replace(
+            $this->keyPattern($oldKey),
+            $replacementKey,
+            $envFileContents
+        )) !== false;
     }
 
     /**

tests/system/Commands/GenerateKeyTest.php

@@ -74,13 +74,13 @@ protected function resetEnvironment()
 
     public function testGenerateKeyShowsEncodedKey()
     {
-        command('key:generate -show');
+        command('key:generate --show');
         $this->assertStringContainsString('hex2bin:', $this->getBuffer());
 
-        command('key:generate -prefix base64 -show');
+        command('key:generate --prefix base64 --show');
         $this->assertStringContainsString('base64:', $this->getBuffer());
 
-        command('key:generate -prefix hex2bin -show');
+        command('key:generate --prefix hex2bin --show');
         $this->assertStringContainsString('hex2bin:', $this->getBuffer());
     }
 
@@ -95,12 +95,12 @@ public function testGenerateKeyCreatesNewKey()
         $this->assertStringContainsString(env('encryption.key'), file_get_contents($this->envPath));
         $this->assertStringContainsString('hex2bin:', file_get_contents($this->envPath));
 
-        command('key:generate -prefix base64 -force');
+        command('key:generate --prefix base64 --force');
         $this->assertStringContainsString('successfully set.', $this->getBuffer());
         $this->assertStringContainsString(env('encryption.key'), file_get_contents($this->envPath));
         $this->assertStringContainsString('base64:', file_get_contents($this->envPath));
 
-        command('key:generate -prefix hex2bin -force');
+        command('key:generate --prefix hex2bin --force');
         $this->assertStringContainsString('successfully set.', $this->getBuffer());
         $this->assertStringContainsString(env('encryption.key'), file_get_contents($this->envPath));
         $this->assertStringContainsString('hex2bin:', file_get_contents($this->envPath));
@@ -115,4 +115,17 @@ public function testDefaultShippedEnvIsMissing()
         $this->assertStringContainsString('Both default shipped', $this->getBuffer());
         $this->assertStringContainsString('Error in setting', $this->getBuffer());
     }
+
+    /**
+     * @see https://github.com/codeigniter4/CodeIgniter4/issues/6838
+     */
+    public function testKeyGenerateWhenKeyIsMissingInDotEnvFile()
+    {
+        file_put_contents($this->envPath, '');
+
+        command('key:generate');
+
+        $this->assertStringContainsString('Application\'s new encryption key was successfully set.', $this->getBuffer());
+        $this->assertSame("\nencryption.key = " . env('encryption.key'), file_get_contents($this->envPath));
+    }
 }