Code Coverage |
||||||||||
Lines |
Functions and Methods |
Classes and Traits |
||||||||
| Total | |
0.00% |
0 / 69 |
|
0.00% |
0 / 2 |
CRAP | |
0.00% |
0 / 1 |
| CargoFileData | |
0.00% |
0 / 69 |
|
0.00% |
0 / 2 |
812 | |
0.00% |
0 / 1 |
| getTableSchema | |
0.00% |
0 / 25 |
|
0.00% |
0 / 1 |
132 | |||
| storeValuesForFile | |
0.00% |
0 / 44 |
|
0.00% |
0 / 1 |
306 | |||
| 1 | <?php |
| 2 | |
| 3 | /** |
| 4 | * Static functions for dealing with the "_fileData" table. |
| 5 | * |
| 6 | * @author Yaron Koren |
| 7 | */ |
| 8 | class CargoFileData { |
| 9 | |
| 10 | /** |
| 11 | * Set the schema based on what has been entered in LocalSettings.php. |
| 12 | * |
| 13 | * @return CargoTableSchema |
| 14 | */ |
| 15 | public static function getTableSchema() { |
| 16 | global $wgCargoFileDataColumns; |
| 17 | |
| 18 | $fieldTypes = []; |
| 19 | |
| 20 | if ( in_array( 'mediaType', $wgCargoFileDataColumns ) ) { |
| 21 | $fieldTypes['_mediaType'] = [ 'type' => 'String' ]; |
| 22 | } |
| 23 | if ( in_array( 'path', $wgCargoFileDataColumns ) ) { |
| 24 | $fieldTypes['_path'] = [ 'type' => 'String', 'hidden' => true ]; |
| 25 | } |
| 26 | if ( in_array( 'lastUploadDate', $wgCargoFileDataColumns ) ) { |
| 27 | $fieldTypes['_lastUploadDate'] = [ 'type' => 'Datetime' ]; |
| 28 | } |
| 29 | if ( in_array( 'fullText', $wgCargoFileDataColumns ) ) { |
| 30 | $fieldTypes['_fullText'] = [ 'type' => 'Searchtext' ]; |
| 31 | } |
| 32 | if ( in_array( 'numPages', $wgCargoFileDataColumns ) ) { |
| 33 | $fieldTypes['_numPages'] = [ 'type' => 'Integer' ]; |
| 34 | } |
| 35 | |
| 36 | $tableSchema = new CargoTableSchema(); |
| 37 | foreach ( $fieldTypes as $field => $fieldVals ) { |
| 38 | $fieldDesc = new CargoFieldDescription(); |
| 39 | foreach ( $fieldVals as $fieldKey => $fieldVal ) { |
| 40 | if ( $fieldKey == 'type' ) { |
| 41 | $fieldDesc->mType = $fieldVal; |
| 42 | } elseif ( $fieldKey == 'list' ) { |
| 43 | // Not currently used. |
| 44 | $fieldDesc->mIsList = true; |
| 45 | $fieldDesc->setDelimiter( '|' ); |
| 46 | } elseif ( $fieldKey == 'hidden' ) { |
| 47 | $fieldDesc->mIsHidden = true; |
| 48 | } |
| 49 | } |
| 50 | $tableSchema->mFieldDescriptions[$field] = $fieldDesc; |
| 51 | } |
| 52 | |
| 53 | return $tableSchema; |
| 54 | } |
| 55 | |
| 56 | /** |
| 57 | * @param Title|null $title |
| 58 | * @param bool $createReplacement |
| 59 | */ |
| 60 | public static function storeValuesForFile( $title, $createReplacement ) { |
| 61 | global $wgCargoFileDataColumns, $wgLocalFileRepo; |
| 62 | |
| 63 | if ( $title == null ) { |
| 64 | return; |
| 65 | } |
| 66 | |
| 67 | // Exit if we're not in the File namespace. |
| 68 | if ( $title->getNamespace() != NS_FILE ) { |
| 69 | return; |
| 70 | } |
| 71 | |
| 72 | $fileDataTable = $createReplacement ? '_fileData__NEXT' : '_fileData'; |
| 73 | |
| 74 | // If there is no _fileData table, getTableSchemas() will |
| 75 | // throw an error. |
| 76 | try { |
| 77 | $tableSchemas = CargoUtils::getTableSchemas( [ $fileDataTable ] ); |
| 78 | } catch ( MWException $e ) { |
| 79 | return; |
| 80 | } |
| 81 | |
| 82 | $repo = new LocalRepo( $wgLocalFileRepo ); |
| 83 | $file = LocalFile::newFromTitle( $title, $repo ); |
| 84 | |
| 85 | $fileDataValues = []; |
| 86 | |
| 87 | if ( in_array( 'mediaType', $wgCargoFileDataColumns ) ) { |
| 88 | $fileDataValues['_mediaType'] = $file->getMimeType(); |
| 89 | } |
| 90 | |
| 91 | if ( in_array( 'path', $wgCargoFileDataColumns ) ) { |
| 92 | $fileDataValues['_path'] = $file->getLocalRefPath(); |
| 93 | } |
| 94 | |
| 95 | if ( in_array( 'lastUploadDate', $wgCargoFileDataColumns ) ) { |
| 96 | $fileDataValues['_lastUploadDate'] = $file->getTimestamp(); |
| 97 | } |
| 98 | |
| 99 | if ( in_array( 'fullText', $wgCargoFileDataColumns ) ) { |
| 100 | global $wgCargoPDFToText; |
| 101 | |
| 102 | if ( $wgCargoPDFToText == '' ) { |
| 103 | // Display an error message? |
| 104 | } elseif ( $file->getMimeType() != 'application/pdf' ) { |
| 105 | // We only handle PDF files. |
| 106 | } else { |
| 107 | // Copied in part from the PdfHandler extension. |
| 108 | $filePath = $file->getLocalRefPath(); |
| 109 | $cmd = wfEscapeShellArg( $wgCargoPDFToText ) . ' ' . wfEscapeShellArg( $filePath ) . ' - '; |
| 110 | $retval = ''; |
| 111 | $txt = wfShellExec( $cmd, $retval ); |
| 112 | if ( $retval == 0 ) { |
| 113 | $txt = str_replace( "\r\n", "\n", $txt ); |
| 114 | $txt = str_replace( "\f", "\n\n", $txt ); |
| 115 | $fileDataValues['_fullText'] = $txt; |
| 116 | } |
| 117 | } |
| 118 | } |
| 119 | |
| 120 | if ( in_array( 'numPages', $wgCargoFileDataColumns ) ) { |
| 121 | global $wgCargoPDFInfo; |
| 122 | if ( $wgCargoPDFInfo == '' ) { |
| 123 | // Display an error message? |
| 124 | } elseif ( $file->getMimeType() != 'application/pdf' ) { |
| 125 | // We only handle PDF files. |
| 126 | } else { |
| 127 | $filePath = $file->getLocalRefPath(); |
| 128 | $cmd = wfEscapeShellArg( $wgCargoPDFInfo ) . ' ' . wfEscapeShellArg( $filePath ); |
| 129 | $retval = ''; |
| 130 | $txt = wfShellExec( $cmd, $retval ); |
| 131 | if ( $retval == 0 ) { |
| 132 | $lines = explode( PHP_EOL, $txt ); |
| 133 | $matched = preg_grep( '/^Pages\:/', $lines ); |
| 134 | foreach ( $matched as $line ) { |
| 135 | $fileDataValues['_numPages'] = intval( trim( substr( $line, 7 ) ) ); |
| 136 | } |
| 137 | } |
| 138 | } |
| 139 | } |
| 140 | |
| 141 | CargoStore::storeAllData( $title, $fileDataTable, $fileDataValues, $tableSchemas[$fileDataTable] ); |
| 142 | } |
| 143 | |
| 144 | } |