115 lines
3.5 KiB
PHP
115 lines
3.5 KiB
PHP
#!/usr/bin/env php
|
|
<?php
|
|
|
|
$ename = 'DB';
|
|
$eport = 5432;
|
|
$confpath = '/var/www/config.php';
|
|
|
|
// check DB_NAME, which will be set automatically for a linked "db" container
|
|
if (!env($ename . '_PORT', '')) {
|
|
error('The env ' . $ename .'_PORT does not exist. Make sure to run with "--link mypostgresinstance:' . $ename . '"');
|
|
}
|
|
|
|
$config = array();
|
|
$config['DB_TYPE'] = 'pgsql';
|
|
$config['DB_HOST'] = env($ename . '_PORT_' . $eport . '_TCP_ADDR');
|
|
$config['DB_PORT'] = env($ename . '_PORT_' . $eport . '_TCP_PORT');
|
|
|
|
// database credentials for this instance
|
|
// database name (DB_NAME) can be supplied or detaults to "ttrss"
|
|
// database user (DB_USER) can be supplied or defaults to database name
|
|
// database pass (DB_PASS) can be supplied or defaults to database user
|
|
$config['DB_NAME'] = env($ename . '_NAME', 'ttrss');
|
|
$config['DB_USER'] = env($ename . '_USER', $config['DB_NAME']);
|
|
$config['DB_PASS'] = env($ename . '_PASS', $config['DB_USER']);
|
|
|
|
if (!dbcheck($config)) {
|
|
echo 'Database login failed, trying to create...' . PHP_EOL;
|
|
// superuser account to create new database and corresponding user account
|
|
// username (SU_USER) can be supplied or defaults to "docker"
|
|
// password (SU_PASS) can be supplied or defaults to username
|
|
|
|
$super = $config;
|
|
|
|
$super['DB_NAME'] = null;
|
|
$super['DB_USER'] = env($ename . '_ENV_USER', 'docker');
|
|
$super['DB_PASS'] = env($ename . '_ENV_PASS', $super['DB_USER']);
|
|
|
|
$pdo = dbconnect($super);
|
|
$pdo->exec('CREATE ROLE ' . ($config['DB_USER']) . ' WITH LOGIN PASSWORD ' . $pdo->quote($config['DB_PASS']));
|
|
$pdo->exec('CREATE DATABASE ' . ($config['DB_NAME']) . ' WITH OWNER ' . ($config['DB_USER']));
|
|
unset($pdo);
|
|
|
|
if (dbcheck($config)) {
|
|
echo 'Database login created and confirmed' . PHP_EOL;
|
|
} else {
|
|
error('Database login failed, trying to create login failed as well');
|
|
}
|
|
}
|
|
|
|
$pdo = dbconnect($config);
|
|
try {
|
|
$pdo->query('SELECT 1 FROM ttrss_feeds');
|
|
// reached this point => table found, assume db is complete
|
|
}
|
|
catch (PDOException $e) {
|
|
echo 'Database table not found, applying schema... ' . PHP_EOL;
|
|
$schema = file_get_contents('schema/ttrss_schema_' . $config['DB_TYPE'] . '.sql');
|
|
$schema = preg_replace('/--(.*?);/', '', $schema);
|
|
$schema = preg_replace('/[\r\n]/', ' ', $schema);
|
|
$schema = trim($schema, ' ;');
|
|
foreach (explode(';', $schema) as $stm) {
|
|
$pdo->exec($stm);
|
|
}
|
|
unset($pdo);
|
|
}
|
|
|
|
$contents = file_get_contents($confpath);
|
|
foreach ($config as $name => $value) {
|
|
$contents = preg_replace('/(define\s*\(\'' . $name . '\',\s*)(.*)(\);)/', '$1"' . $value . '"$3', $contents);
|
|
}
|
|
file_put_contents($confpath, $contents);
|
|
|
|
function env($name, $default = null)
|
|
{
|
|
$v = getenv($name) ?: $default;
|
|
|
|
if ($v === null) {
|
|
error('The env ' . $name . ' does not exist');
|
|
}
|
|
|
|
return $v;
|
|
}
|
|
|
|
function error($text)
|
|
{
|
|
echo 'Error: ' . $text . PHP_EOL;
|
|
exit(1);
|
|
}
|
|
|
|
function dbconnect($config)
|
|
{
|
|
$map = array('host' => 'HOST', 'port' => 'PORT', 'dbname' => 'NAME', 'user' => 'USER', 'password' => 'PASS');
|
|
$dsn = $config['DB_TYPE'] . ':';
|
|
foreach ($map as $d => $h) {
|
|
if (isset($config['DB_' . $h])) {
|
|
$dsn .= $d . '=' . $config['DB_' . $h] . ';';
|
|
}
|
|
}
|
|
$pdo = new \PDO($dsn);
|
|
$pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
|
|
return $pdo;
|
|
}
|
|
|
|
function dbcheck($config)
|
|
{
|
|
try {
|
|
dbconnect($config);
|
|
return true;
|
|
}
|
|
catch (PDOException $e) {
|
|
return false;
|
|
}
|
|
}
|
|
|