From c70ae88a2a89404f008436e9bc17551459b5f6dc Mon Sep 17 00:00:00 2001
From: Anonymous Contributor <anonymous@invalid>
Date: Sat, 2 Sep 2023 15:33:21 +0200
Subject: [PATCH] Add dirFiles Twig function

---
 four-creative.php         | 11 ++++++++++-
 php/DirFilesExtension.php | 35 +++++++++++++++++++++++++++++++++++
 2 files changed, 45 insertions(+), 1 deletion(-)
 create mode 100644 php/DirFilesExtension.php

diff --git a/four-creative.php b/four-creative.php
index 472f033..f651fbd 100644
--- a/four-creative.php
+++ b/four-creative.php
@@ -5,5 +5,14 @@ use Grav\Common\Theme;
 
 class FourCreative extends Theme
 {
-    // Access plugin events in this class
+    public static function getSubscribedEvents()
+    {
+        return ['onTwigExtensions' => ['onTwigExtensions', 0]];
+    }
+
+    public function onTwigExtensions()
+    {
+        require_once __DIR__.DIRECTORY_SEPARATOR.'php'.DIRECTORY_SEPARATOR.'DirFilesExtension.php';
+        $this->grav['twig']->twig->addExtension(new DirFilesExtension());
+    }
 }
diff --git a/php/DirFilesExtension.php b/php/DirFilesExtension.php
new file mode 100644
index 0000000..432cfe8
--- /dev/null
+++ b/php/DirFilesExtension.php
@@ -0,0 +1,35 @@
+<?php
+declare(strict_types=1);
+
+namespace Grav\Theme;
+
+use Grav\Common\Twig\Extension\GravExtension;
+
+class DirFilesExtension extends GravExtension
+{
+    public function getName(): string
+    {
+        return 'DirFilesExtension';
+    }
+
+    public function getFunctions(): array
+    {
+        return [new \Twig_SimpleFunction('dirFiles', [$this, 'getDirFiles'])];
+    }
+
+    public function getDirFiles(string $dir, string $exts): array
+    {
+        $res = array();
+        $dir = '..'.DIRECTORY_SEPARATOR.(str_starts_with($dir, '/') ? substr($dir, 1) : $dir);
+        if (file_exists($dir) && $dirHandle = opendir($dir))
+        {
+            while ($entry = readdir($dirHandle)) {
+                if (preg_match('/^[a-zA-Z0-9-_][a-zA-Z0-9-_.]*\.('.$exts.')$/', $entry)) {
+                    $res[] = $dir.DIRECTORY_SEPARATOR.$entry;
+                }
+            }
+        }
+
+        return $res;
+    }
+}