From 85903fa9b5dcd961a1dd6c292f2a7f5afa7c389d Mon Sep 17 00:00:00 2001
From: Jaeger <JaegerCode@gmail.com>
Date: Fri, 3 Apr 2020 20:16:00 +0800
Subject: [PATCH] feat: rules add attrs

---
 src/Dom/Query.php | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/Dom/Query.php b/src/Dom/Query.php
index 1d346b8..9e75bea 100644
--- a/src/Dom/Query.php
+++ b/src/Dom/Query.php
@@ -209,7 +209,13 @@ class Query
                 })->all();
                 break;
             default:
-                $content = $pqObj->attr($rule['attr']);
+                if(preg_match('/attr\((.+)\)/', $rule['attr'], $arr)) {
+                    $content = $pqObj->attr($arr[1]);
+                } elseif (preg_match('/attrs\((.+)\)/', $rule['attr'], $arr)) {
+                    $content = (new Elements($pqObj))->attrs($arr[1])->all();
+                } else {
+                    $content = $pqObj->attr($rule['attr']);
+                }
                 break;
         }