feat: rules add attrs

jae-jae · Apr 3, 2020 · 85903fa · 85903fa
1 parent e527c63
commit 85903fa
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/Dom/Query.php b/src/Dom/Query.php
@@ -209,7 +209,13 @@ protected function extractContent(phpQueryObject $pqObj, $ruleName, $rule)
                 })->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;
         }