UriBuilder: Add getQRCodeDataUri method

Vectorface · Jan 31, 2024 · c98adc4 · c98adc4
1 parent 35d58bb
commit c98adc4
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/OtpAuth/UriBuilder.php b/src/OtpAuth/UriBuilder.php
@@ -2,6 +2,8 @@
 
 namespace Vectorface\OtpAuth;
 
+use Endroid\QrCode\Builder\Builder;
+use Endroid\QrCode\Writer\PngWriter;
 use Vectorface\OtpAuth\Paramters\Algorithm;
 use Vectorface\OtpAuth\Paramters\Type;
 
@@ -136,4 +138,15 @@ public function __toString(): string
     {
         return $this->getUri();
     }
+
+    public function getQRCodeDataUri(): string
+    {
+        return Builder::create()
+            ->data($this->getUri())
+            ->writer(new PngWriter)
+            ->size(260)
+            ->margin(10)
+            ->build()
+            ->getDataUri();
+    }
 }
diff --git a/tests/OtpAuth/UriBuilderTest.php b/tests/OtpAuth/UriBuilderTest.php
@@ -19,8 +19,12 @@ public function synopsis()
             ->account("MyAcct")
             ->secret("FOO");
 
+        /* The builder can generate otpauth URLs */
         $this->assertEquals("otpauth://totp/MyAcct?secret=FOO", "$uriBuilder");
 
+        /* ... or QR codes as data URIs */
+        $this->assertStringStartsWith("data:image/png;base64,", $uriBuilder->getQRCodeDataUri());
+
         /* It is also possible to construct complex OTP URIs, including HOTP */
         $uriBuilder = (new UriBuilder())
             ->type(Type::HOTP)