phpvms/app/Support/Database.php

149 lines
3.7 KiB
PHP
Raw Normal View History

<?php
namespace App\Support;
use Carbon\Carbon;
use Illuminate\Database\QueryException;
use Illuminate\Support\Facades\DB;
use Illuminate\Support\Facades\Log;
2018-08-27 00:40:04 +08:00
use Symfony\Component\Yaml\Yaml;
class Database
{
/**
* @return string
*/
protected static function time(): string
{
return Carbon::now('UTC');
}
/**
* @param $yaml_file
* @param bool $ignore_errors
2018-08-27 00:40:04 +08:00
*
* @throws \Exception
2018-08-27 00:40:04 +08:00
*
* @return array
*/
public static function seed_from_yaml_file($yaml_file, $ignore_errors = false): array
{
$yml = file_get_contents($yaml_file);
return static::seed_from_yaml($yml, $ignore_errors);
}
/**
* @param $yml
* @param bool $ignore_errors
2018-08-27 00:40:04 +08:00
*
* @throws \Exception
2018-08-27 00:40:04 +08:00
*
* @return array
*/
public static function seed_from_yaml($yml, $ignore_errors = false): array
{
$imported = [];
$yml = Yaml::parse($yml);
if (empty($yml)) {
return $imported;
}
foreach ($yml as $table => $data) {
$imported[$table] = 0;
$id_column = 'id';
if (array_key_exists('id_column', $data)) {
$id_column = $data['id_column'];
}
$ignore_on_update = [];
if (array_key_exists('ignore_on_update', $data)) {
$ignore_on_update = $data['ignore_on_update'];
}
if (array_key_exists('data', $data)) {
$rows = $data['data'];
} else {
$rows = $data;
}
foreach ($rows as $row) {
try {
static::insert_row($table, $row, $id_column, $ignore_on_update);
} catch (QueryException $e) {
if ($ignore_errors) {
continue;
}
throw $e;
}
2018-08-27 00:40:04 +08:00
$imported[$table]++;
}
}
return $imported;
}
/**
* @param string $table
* @param array $row
* @param string $id_col The ID column to use for update/insert
* @param array $ignore_on_updates
* @param bool $ignore_errors
2018-08-27 00:40:04 +08:00
*
* @return mixed
*/
public static function insert_row(
$table,
$row,
$id_col = 'id',
$ignore_on_updates = [],
$ignore_errors = true
) {
2018-08-27 00:40:04 +08:00
// encrypt any password fields
if (array_key_exists('password', $row)) {
$row['password'] = bcrypt($row['password']);
}
if (empty($row)) {
return $row;
}
2018-08-27 00:40:04 +08:00
// if any time fields are == to "now", then insert the right time
foreach ($row as $column => $value) {
if (strtolower($value) === 'now') {
$row[$column] = static::time();
}
}
$count = 0;
if (array_key_exists($id_col, $row)) {
$count = DB::table($table)->where($id_col, $row[$id_col])->count($id_col);
}
try {
if ($count > 0) {
foreach ($ignore_on_updates as $ignore_column) {
if (array_key_exists($ignore_column, $row)) {
unset($row[$ignore_column]);
}
}
DB::table($table)
->where($id_col, $row[$id_col])
->update($row);
} else {
DB::table($table)->insert($row);
}
} catch (QueryException $e) {
2018-07-16 04:06:25 +08:00
Log::error($e);
if (!$ignore_errors) {
throw $e;
}
}
return $row;
}
}