Merge pull request filbertkm#21 from lucaswerkmeister/stdin

Add option to read entity IDs from standard input

Author: filbertkm

README.md

@@ -63,3 +63,9 @@ Import a range of entities:
 ```
 php maintenance/importEntities.php --range Q1:Q20
 ```
+
+Import a list of entities printed by another program:
+
+```
+printf 'Q%s\n' {1..20} {100..120} | php maintenance/importEntities.php --stdin
+```

maintenance/importEntities.php

@@ -39,6 +39,7 @@ private function addOptions() {
 		$this->addOption( 'entity', 'ID of entity to import', false, true );
 		$this->addOption( 'query', 'Import items with property and entity id value', false, true );
 		$this->addOption( 'range', 'Range of ids to import', false, true );
+		$this->addOption( 'stdin', 'Read entity IDs to import from standard input', false, false );
 		$this->addOption( 'all-properties', 'Import all properties', false, false );
 	}
 
@@ -98,7 +99,7 @@ private function extractOptions() {
 	}
 
 	private function getValidOptions() {
-		return [ 'entity', 'file', 'all-properties', 'query', 'range' ];
+		return [ 'entity', 'file', 'all-properties', 'query', 'range', 'stdin' ];
 	}
 
 	private function newEntityIdListBuilderFactory() {

src/EntityId/EntityIdListBuilderFactory.php

@@ -65,6 +65,8 @@ public function newEntityIdListBuilder( $mode ) {
 				return $this->newRangeEntityIdListBuilder();
 			case 'query':
 				return $this->newQueryEntityIdListBuilder();
+			case 'stdin':
+				return $this->newStdinEntityIdListBuilder();
 			default:
 				throw new InvalidArgumentException( 'Unknown import mode: ' . $mode );
 		}
@@ -96,4 +98,8 @@ private function newRangeEntityIdListBuilder() {
 		return new RangeEntityIdListBuilder( $this->idParser );
 	}
 
+	private function newStdinEntityIdListBuilder() {
+		return new StdinEntityIdListBuilder();
+	}
+
 }

src/EntityId/StdinEntityIdListBuilder.php

@@ -0,0 +1,27 @@
+<?php
+
+namespace Wikibase\Import\EntityId;
+
+use RuntimeException;
+
+/**
+ * @licence GNU GPL v2+
+ * @author Lucas Werkmeister < lucas.werkmeister@wikimedia.de >
+ */
+class StdinEntityIdListBuilder implements EntityIdListBuilder {
+
+	/**
+	 * @param string $input (ignored)
+	 *
+	 * @throws RuntimeException
+	 * @return string[]
+	 */
+	public function getEntityIds( $input ) {
+		$entityIds = [];
+		while ( $line = fgets( STDIN ) ) {
+			$entityIds[] = trim( $line );
+		}
+		return $entityIds;
+	}
+
+}