1 files changed, 30 insertions, 0 deletions
diff --git a/app/class/Flywheel/Repository.php b/app/class/Flywheel/Repository.php
new file mode 100644
index 0000000..d0b68eb
--- /dev/null
+++ b/app/class/Flywheel/Repository.php
@@ -0,0 +1,30 @@
+<?php
+namespace Wcms\Flywheel;
+
+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());
+    }
+}