includes/installer/SqliteDatabaseCreator.php

   1 <?php
   2
   3 namespace MediaWiki\Installer;
   4
   5 use MediaWiki\Installer\Task\ITaskContext;
   6 use MediaWiki\MainConfigNames;
   7 use MediaWiki\Status\Status;
   8 use Wikimedia\Rdbms\DatabaseSqlite;
   9 use Wikimedia\Rdbms\ILoadBalancer;
  10
  11 class SqliteDatabaseCreator extends DatabaseCreator {
  12         /** @var string */
  13         private $dataDir;
  14
  15         protected function __construct( ITaskContext $context ) {
  16                 parent::__construct( $context );
  17                 $this->dataDir = $context->getConfigVar( MainConfigNames::SQLiteDataDir );
  18         }
  19
  20         public function existsLocally( $database ) {
  21                 $file = DatabaseSqlite::generateFileName( $this->dataDir, $database );
  22                 return file_exists( $file );
  23         }
  24
  25         public function existsInLoadBalancer( ILoadBalancer $loadBalancer, $database ) {
  26                 return $this->existsLocally( $database );
  27         }
  28
  29         public function createLocally( $database ): Status {
  30                 return $this->makeStubDBFile( $database );
  31         }
  32
  33         public function createInLoadBalancer( ILoadBalancer $loadBalancer, $database ): Status {
  34                 return $this->createLocally( $database );
  35         }
  36
  37         private function makeStubDBFile( $db ) {
  38                 $file = DatabaseSqlite::generateFileName( $this->dataDir, $db );
  39
  40                 if ( file_exists( $file ) ) {
  41                         if ( !is_writable( $file ) ) {
  42                                 return Status::newFatal( 'config-sqlite-readonly', $file );
  43                         }
  44                         return Status::newGood();
  45                 }
  46
  47                 $oldMask = umask( 0177 );
  48                 if ( file_put_contents( $file, '' ) === false ) {
  49                         umask( $oldMask );
  50                         return Status::newFatal( 'config-sqlite-cant-create-db', $file );
  51                 }
  52                 umask( $oldMask );
  53
  54                 return Status::newGood();
  55         }
  56
  57 }