Implement 'equals starts with' ('^=') and 'equals or starts with hyph…

…enated' ('|=') selectors (#221) * Improve detection of element in query * style: reformat * style: reformat * test: multiple named selects * Implement 'equals starts with' ('^=') and 'equals or starts with hyphenated' ('|=') selectors --------- Co-authored-by: Tomáš Fedor <[email protected]> Co-authored-by: Greg Bowler <[email protected]>
PhpGt · Jan 3, 2024 · c0a0667 · c0a0667
1 parent aa941bb
commit c0a0667
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 5 deletions.
diff --git a/src/Translator.php b/src/Translator.php
@@ -12,15 +12,15 @@ class Translator {
 		. '|(#(?P<id>[\w-]*))'
 		. '|(\.(?P<class>[\w-]*))'
 		. '|(?P<sibling>\s*\+\s*)'
-		. "|(\[(?P<attribute>[\w-]*)((?P<attribute_equals>[=~$*]+)(?P<attribute_value>(.+\[\]'?)|[^\]]+))*\])+"
+		. "|(\[(?P<attribute>[\w-]*)((?P<attribute_equals>[=~$|^*]+)(?P<attribute_value>(.+\[\]'?)|[^\]]+))*\])+"
 		. '|(?P<descendant>\s+)'
 		. '/';
 
 	const EQUALS_EXACT = "=";
 	const EQUALS_CONTAINS_WORD = "~=";
 	const EQUALS_ENDS_WITH = "$=";
 	const EQUALS_CONTAINS = "*=";
-	const EQUALS_STARTS_WITH_OR_STARTS_WITH_HYPHENATED = "|=";
+	const EQUALS_OR_STARTS_WITH_HYPHENATED = "|=";
 	const EQUALS_STARTS_WITH = "^=";
 
 	public function __construct(
@@ -245,11 +245,24 @@ protected function convertSingleSelector(string $css):string {
 					);
 					break;
 
-				case self::EQUALS_STARTS_WITH_OR_STARTS_WITH_HYPHENATED:
-					throw new NotYetImplementedException();
+				case self::EQUALS_OR_STARTS_WITH_HYPHENATED:
+					array_push(
+						$xpath,
+						"["
+						. "@{$currentThreadItem['content']}=\"{$valueString}\" or "
+						. "starts-with(@{$currentThreadItem['content']}, \"{$valueString}-\")"
+						. "]"
+					);
+					break;
 
 				case self::EQUALS_STARTS_WITH:
-					throw new NotYetImplementedException();
+					array_push(
+						$xpath,
+						"[starts-with("
+						. "@{$currentThreadItem['content']}, \"{$valueString}\""
+						. ")]"
+					);
+					break;
 
 				case self::EQUALS_ENDS_WITH:
 					array_push(

diff --git a/test/phpunit/TranslatorTest.php b/test/phpunit/TranslatorTest.php
@@ -320,6 +320,30 @@ public function testAttributeDollarSelector() {
 		);
 	}
 
+	public function testAttributeEqualsOrStartsWithHypehnatedSelector() {
+		$document = new DOMDocument("1.0", "UTF-8");
+		$document->loadHTML("<div class='en'></div><div class='en-'></div><div class='en-uk'></div><div class='es'></div>");
+		$xpath = new DOMXPath($document);
+
+		$selector = new Translator("[class|=en]");
+		self::assertEquals(
+			3,
+			$xpath->query($selector)->length
+		);
+	}
+
+	public function testAttributeStartsWithSelector() {
+		$document = new DOMDocument("1.0", "UTF-8");
+		$document->loadHTML("<div class='class1'></div><div class='foo class1'></div><div class='class1 class2'></div><div class='class2'></div>");
+		$xpath = new DOMXPath($document);
+
+		$selector = new Translator("[class^=class1]");
+		self::assertEquals(
+			2,
+			$xpath->query($selector)->length
+		);
+	}
+
 	public function testClassSelector() {
 		$document = new DOMDocument("1.0", "UTF-8");
 		$document->loadHTML(Helper::HTML_COMPLEX);