enhance(directives): Add a @postmeta directive (Fixes #79) (#90)

Log1x · Jul 29, 2023 · 4e97354 · 4e97354
2 parents 0c9e9fc + 3682429
commit 4e97354
Show file tree

Hide file tree

Showing 3 changed files with 93 additions and 0 deletions.
diff --git a/docs/usage/wordpress.md b/docs/usage/wordpress.md
@@ -103,6 +103,21 @@ If `@query` is not used and an argument is not passed to `@posts`, it will use t
 @endnoposts
 ```
 
+## @postmeta
+
+`@postmeta` allows you to get the post meta for a post using [`get_post_meta`](https://developer.wordpress.org/reference/functions/get_post_meta/).
+
+`@postmeta` will always expect a `key` as the first argument with an optional `post_id` for the second.
+
+The third argument is optional and allows you to control the `$single` bool to only return a single value.
+
+```php
+@postmeta('key')
+@postmeta('key', 1)
+@postmeta(1)
+@postmeta('key', 1, true)
+```
+
 ## @title
 
 `@title` echo's the current posts title using [`get_the_title()`](https://developer.wordpress.org/reference/functions/get_the_title/).

diff --git a/src/Directives/WordPress.php b/src/Directives/WordPress.php
@@ -139,6 +139,50 @@
         return '<?php wp_reset_postdata(); endif; ?>';
     },
 
+    /*
+    |---------------------------------------------------------------------
+    | @postmeta
+    |---------------------------------------------------------------------
+    */
+
+    'postmeta' => function ($expression) {
+        if (! empty($expression)) {
+            $expression = Util::parse($expression);
+
+            if (Util::isIdentifier($expression->get(0))) {
+                if (empty($expression->get(1))) {
+                    return "<?php echo get_post_meta({$expression->get(0)}); ?>";
+                }
+
+                if (empty($expression->get(2))) {
+                    $expression->put(2, 'false');
+                }
+
+                return "<?php echo get_post_meta({$expression->get(0)}, {$expression->get(1)}, {$expression->get(2)}); ?>";
+            }
+
+            if (Util::isIdentifier($expression->get(1))) {
+                if (empty($expression->get(2))) {
+                    $expression->put(2, 'false');
+                }
+
+                return "<?php echo get_post_meta({$expression->get(1)}, {$expression->get(0)}, {$expression->get(2)}); ?>";
+            }
+
+            if (empty($expression->get(1))) {
+                $expression->put(1, 'false');
+            }
+
+            if (! Util::isIdentifier($expression->get(0))) {
+                return "<?php echo get_post_meta(get_the_ID(), {$expression->get(0)}, {$expression->get(1)}); ?>";
+            }
+
+            return "<?php echo get_post_meta(get_the_ID(), {$expression->get(0)}, {$expression->get(1)}); ?>";
+        }
+
+        return '<?php echo get_post_meta(get_the_ID()); ?>';
+    },
+
     /*
     |---------------------------------------------------------------------
     | @title / @content / @excerpt / @permalink / @thumbnail

diff --git a/tests/Unit/WordPressTest.php b/tests/Unit/WordPressTest.php
@@ -94,6 +94,40 @@
     });
 });
 
+describe('@postmeta', function () {
+    it('compiles correctly', function () {
+        $directive = "@postmeta('foo')";
+
+        $compiled = $this->compile($directive);
+
+        expect($compiled)->toBe("<?php echo get_post_meta(get_the_ID(), 'foo', false); ?>");
+    });
+
+    it('compiles correctly with object', function () {
+        $directive = "@postmeta('foo', \$post->ID)";
+
+        $compiled = $this->compile($directive);
+
+        expect($compiled)->toBe("<?php echo get_post_meta(\$post->ID, 'foo', false); ?>");
+    });
+
+    it('compiles correctly with post', function () {
+        $directive = "@postmeta('foo', 1)";
+
+        $compiled = $this->compile($directive);
+
+        expect($compiled)->toBe("<?php echo get_post_meta(1, 'foo', false); ?>");
+    });
+
+    it('compiles correctly with post and single', function () {
+        $directive = "@postmeta('foo', 1, true)";
+
+        $compiled = $this->compile($directive);
+
+        expect($compiled)->toBe("<?php echo get_post_meta(1, 'foo', true); ?>");
+    });
+});
+
 describe('@title', function () {
     it('compiles correctly', function () {
         $directive = '@title';