Finished querying extension version. See readme.

Designburo · Designburo · commit f679cae9f06e · 2024-01-31T22:15:44.000+01:00
diff --git a/README.md b/README.md
@@ -75,9 +75,17 @@ This return
 
 It takes two arguments : 
 
-**Extension name** tba
+**Extension name** ( Name of the extension to query )
 
-**type** tba
+**type** ( Query type, currenly only **version** is supported )
+
+Example :
+```wikitext
+{{#w8y:action=extension
+|Extension name=ParserFunctions
+|type=version
+}}
+```
 
 ## Lua
 Call are the same as for the parser functions.
diff --git a/includes/Scribunto/ScribuntoLuaLibrary.php b/includes/Scribunto/ScribuntoLuaLibrary.php
@@ -42,11 +42,12 @@ public function w8y( ?array $arguments ): array {
 		switch ( $action ) {
 			case "extension":
 				$eName = Utils::getOptionSetting( 'Extension name', true, $arguments );
-				if ( $eName === null ) {
+				$eVersion = Utils::getOptionSetting( 'version', true, $arguments );
+				if ( $eName === null || $eVersion === null ) {
 					return [];
 				}
 				$query = new Extensions();
-				$result = $query->doQuery( $eName, 'lua' );
+				$result = $query->doQuery( $eName, $eVersion, 'lua' );
 				return [ $this->convertToLuaTable( $result ) ];
 			case "wiki":
 				$id = Utils::getOptionSetting( 'id', true, $arguments );
diff --git a/includes/TagHooks.php b/includes/TagHooks.php
@@ -50,11 +50,12 @@ public function w8y( Parser &$parser ): mixed {
 		switch ( $action ) {
 			case "extension":
 				$eName = Utils::getOptionSetting( 'Extension name' );
-				if ( $eName === null ) {
+				$eType = Utils::getOptionSetting( 'type' );
+				if ( $eName === null || $eType === null ) {
 					break;
 				}
 				$extension = new Extensions();
-				$result = $extension->doQuery( $eName );
+				$result = $extension->doQuery( $eName, $eType );
 				break;
 			case "query":
 				$query = new Query();
diff --git a/includes/data/query/Extensions.php b/includes/data/query/Extensions.php
@@ -67,16 +67,23 @@ private function getExtensionVersions( string $extensionName, DBConnRef $dbr ) {
 
 	/**
 	 * @param string $extensionName
+	 * @param string $queryType
 	 * @param string $export
 	 *
 	 * @return mixed
 	 */
-	public function doQuery( string $extensionName, string $export = "table" ): mixed {
+	public function doQuery( string $extensionName, string $queryType, string $export = "table" ): mixed {
 		$lb = MediaWikiServices::getInstance()->getDBLoadBalancer();
 		$dbr = $lb->getConnectionRef( DB_REPLICA );
 		$result = [];
 
-		$result = $this->getExtensionVersions( $extensionName, $dbr );
+		switch ( $queryType ) {
+			case "version":
+				$result = $this->getExtensionVersions( $extensionName, $dbr );
+				break;
+			default:
+				$result = [];
+		}
 
 		switch ( $export ) {
 			case "table":
