fix: cleaning prefixes removes all instances instead of retaining wan…

…ted ones
wbstack · Jan 3, 2024 · 4b030b3 · 4b030b3
1 parent ca78c10
commit 4b030b3
Showing 1 changed file with 16 additions and 23 deletions.
diff --git a/wikibase/queryService/ui/queryHelper/QueryHelper.js b/wikibase/queryService/ui/queryHelper/QueryHelper.js
@@ -86,7 +86,6 @@ wikibase.queryService.ui.queryHelper.QueryHelper = ( function( $, wikibase, _ )
 	SELF.prototype.setQuery = function( query ) {
 		var prefixes = wikibase.queryService.RdfNamespaces.ALL_PREFIXES;
 		this._query.parse( query, prefixes );
-
 		this._selectorBox.setQuery( this._query );
 	};
 
@@ -111,28 +110,22 @@ wikibase.queryService.ui.queryHelper.QueryHelper = ( function( $, wikibase, _ )
 	 * @private
 	 */
 	SELF.prototype._cleanQueryPrefixes = function( query ) {
-		var prefixRegex = /PREFIX ([a-z]+): <(.*)>/gi,
-			m,
-			prefixes = {},
-			cleanQuery = query.replace( prefixRegex, '' ).replace( /\n+/g, '\n' );
-
-		while ( ( m = prefixRegex.exec( query ) ) ) {
-			var prefix = m[1];
-			var uri = m[2].replace( /[\-\[\]\/\{\}\(\)\*\+\?\.\\\^\$\|]/g, '\\$&' );
-
-			var newQuery = cleanQuery.replace( new RegExp( '<' + uri + '([^/>#]+?)>', 'gi' ),
-					prefix + ':$1' );
-
-			if ( cleanQuery !== newQuery ) {
-				cleanQuery = newQuery;
-				if ( !wikibase.queryService.RdfNamespaces.STANDARD_PREFIXES[prefix] ) {
-					prefixes[m[0]] = true;
-				}
-			}
-		}
-
-		cleanQuery = Object.keys( prefixes ).join( '\n' ) + '\n\n' + cleanQuery.trim();
-		return cleanQuery;
+        var prefixRe = /^\s*PREFIX\s+([a-z]+):\s+<.*>\s*$/;
+        return = query.split( '\n' )
+            .map(function ( line ) {
+                if ( !prefixRe.test( line ) ) {
+                    return line;
+                }
+                line = line.trim();
+                var match = line.match( prefixRe );
+                var prefix = match[1];
+                if ( prefix in wikibase.queryService.RdfNamespaces.STANDARD_PREFIXES ) {
+                    return null;
+                }
+                return line;
+            })
+            .filter( Boolean )
+            .join( '\n' );
 	};
 
 	/**