Flywheel extension and custom json formatter

1 files changed, 30 insertions, 0 deletions

diff --git a/app/class/wflywheel/repository.php b/app/class/wflywheel/repository.php
new file mode 100644
index 0000000..baad42e
--- /dev/null
+++ b/app/class/wflywheel/repository.php
@@ -0,0 +1,30 @@
+<?php
+namespace WFlywheel;
+
+class Repository extends \JamesMoss\Flywheel\Repository
+{
+    /**
+     * Get an array containing the path of all files in this repository
+     *
+     * @return array An array, item is a file
+     */
+    public function getAllFiles()
+    {
+        $ext       = $this->formatter->getFileExtension();
+        $files     = glob($this->path . DIRECTORY_SEPARATOR . '*.' . $ext);
+        return $files;
+    }
+
+    /**
+     * Get an array containing the id of all files in this repository
+     *
+     * @return array An array, item is a id
+     */
+    public function getAllIds()
+    {
+        $ext = $this->formatter->getFileExtension();
+        return array_map(function($path) use ($ext) {
+            return $this->getIdFromPath($path, $ext);
+        }, $this->getAllFiles());
+    }
+}