yii2-bootstrap/framework/yii/db/oci/Schema.php

<?php

namespace yii\db\oci;

use yii\db\TableSchema;
use yii\db\oci\ColumnSchema;

class Schema extends \yii\db\Schema {

	private $_defaultSchema = '';

	public function createQueryBuilder() {
		return new QueryBuilder($this->db);
	}

	public function quoteTableName($name) {
		if (strpos($name, '.') === false) {
			return $this->quoteSimpleTableName($name);
		}
		$parts = explode('.', $name);
		foreach ($parts as $i => $part) {
			$parts[$i] = $this->quoteSimpleTableName($part);
		}
		return implode('.', $parts);
	}

	public function quoteSimpleTableName($name) {
		return '"' . $name . '"';
		//return $name;
	}

	public function quoteColumnName($name) {
		if (($pos = strrpos($name, '.')) !== false) {
			$prefix = $this->quoteTableName(substr($name, 0, $pos)) . '.';
			$name = substr($name, $pos + 1);
		} else {
			$prefix = '';
		}
		return $prefix . ($name === '*' ? $name : $this->quoteSimpleColumnName($name));
	}

	public function quoteSimpleColumnName($name) {
		return '"' . $name . '"';
		//return $name;
	}

	public function loadTableSchema($name) {
		$table = new TableSchema();
		$this->resolveTableNames($table, $name);
		// $this->findPrimaryKeys($table);        
		if ($this->findColumns($table)) {
			//$this->findForeignKeys($table);
			$this->findConstraints($table);
			return $table;
		}
	}

	/**
	 * Resolves the table name and schema name (if any).
	 *
	 * @param TableSchema $table the table metadata object
	 * @param string $name the table name
	 */
	protected function resolveTableNames($table, $name) {
		$parts = explode('.', str_replace('"', '', $name));
		if (isset($parts[1])) {
			$table->schemaName = $parts[0];
			$table->name = $parts[1];
		} else {
			$table->name = $parts[0];
		}

		if ($table->schemaName === null) {
			$table->schemaName = $this->getDefaultSchema();
		}
	}

	/**
	 *
	 * @return string default schema.
	 */
	public function getDefaultSchema() {
		if (!strlen($this->_defaultSchema)) {
			$this->setDefaultSchema(strtoupper($this->db->username));
		}

		return $this->_defaultSchema;
	}

	/**
	 *
	 * @param string $schema default schema.
	 */
	public function setDefaultSchema($schema) {
		$this->_defaultSchema = $schema;
	}

	public function getLastInsertID($sequenceName = '') {
		if ($this->db->isActive) {
			$sql = "select {$sequenceName}.currval from dual";
			return $this->db->createCommand($sql)->queryScalar();
		} else {
			throw new InvalidCallException('DB Connection is not active.');
		}
	}

	/**
	 * Collects the table column metadata.
	 * @param COciTableSchema $table the table metadata
	 * @return boolean whether the table exists in the database
	 */
	protected function findColumns($table) {
		$schemaName = $table->schemaName;
		$tableName = $table->name;

		$sql = <<<EOD
SELECT a.column_name, a.data_type ||
    case
        when data_precision is not null
            then '(' || a.data_precision ||
                    case when a.data_scale > 0 then ',' || a.data_scale else '' end
                || ')'
        when data_type = 'DATE' then ''
        when data_type = 'NUMBER' then ''
        else '(' || to_char(a.data_length) || ')'
    end as data_type,
    a.nullable, a.data_default,
    (   SELECT D.constraint_type
        FROM ALL_CONS_COLUMNS C
        inner join ALL_constraints D on D.OWNER = C.OWNER and D.constraint_name = C.constraint_name
        WHERE C.OWNER = B.OWNER
           and C.table_name = B.object_name
           and C.column_name = A.column_name
           and D.constraint_type = 'P') as Key,
    com.comments as column_comment
FROM ALL_TAB_COLUMNS A
inner join ALL_OBJECTS B ON b.owner = a.owner and ltrim(B.OBJECT_NAME) = ltrim(A.TABLE_NAME)
LEFT JOIN user_col_comments com ON (A.table_name = com.table_name AND A.column_name = com.column_name)
WHERE
    a.owner = '{$schemaName}'
	and (b.object_type = 'TABLE' or b.object_type = 'VIEW')
	and b.object_name = '{$tableName}'
ORDER by a.column_id
EOD;


		try {
			$columns = $this->db->createCommand($sql)->queryAll();
		} catch (\Exception $e) {
			return false;
		}

		foreach ($columns as $column) {
			$c = $this->createColumn($column);

			$table->columns[$c->name] = $c;
			if ($c->isPrimaryKey) {
				if ($table->primaryKey === null) {
					$table->primaryKey = $c->name;
				} elseif (is_string($table->primaryKey)) {
					$table->primaryKey = array($table->primaryKey, $c->name);
				} else {
					$table->primaryKey[] = $c->name;
				}
				$sql = "select sequence_name
from user_tables tabs
join user_triggers trigs
  on trigs.table_name = tabs.table_name
join user_dependencies deps
  on deps.name = trigs.trigger_name
join user_sequences seqs
  on seqs.sequence_name = deps.referenced_name
where tabs.table_name = '{$tableName}'
 and trigs.triggering_event = 'INSERT'";
				try {
					$seqName = $this->db->createCommand($sql)->queryScalar();
				} catch (\Exception $e) {
					return false;
				}
				$table->sequenceName = $seqName;
				$c->autoIncrement = true;
			}
		}
		return true;
	}

	protected function createColumn($column) {
		$c = new ColumnSchema;
		$c->name = $column['COLUMN_NAME'];
		$c->allowNull = $column['NULLABLE'] === 'Y';
		$c->isPrimaryKey = strpos($column['KEY'], 'P') !== false;
		$c->extract($column['DATA_TYPE'], $column['DATA_DEFAULT']);
		$c->comment = $column['COLUMN_COMMENT'] === null ? '' : $column['COLUMN_COMMENT'];

		return $c;
	}

	protected function findConstraints($table) {
		$sql = <<<EOD
		SELECT D.constraint_type as CONSTRAINT_TYPE, C.COLUMN_NAME, C.position, D.r_constraint_name,
                E.table_name as table_ref, f.column_name as column_ref,
            	C.table_name
        FROM ALL_CONS_COLUMNS C
        inner join ALL_constraints D on D.OWNER = C.OWNER and D.constraint_name = C.constraint_name
        left join ALL_constraints E on E.OWNER = D.r_OWNER and E.constraint_name = D.r_constraint_name
        left join ALL_cons_columns F on F.OWNER = E.OWNER and F.constraint_name = E.constraint_name and F.position = c.position
        WHERE C.OWNER = '{$table->schemaName}'
           and C.table_name = '{$table->name}'
           and D.constraint_type <> 'P'
        order by d.constraint_name, c.position
EOD;
		$command = $this->db->createCommand($sql);
		foreach ($command->queryAll() as $row) {
			if ($row['CONSTRAINT_TYPE'] === 'R') {   // foreign key
				$name = $row["COLUMN_NAME"];
				$table->foreignKeys[$name] = array($row["TABLE_REF"], $row["COLUMN_REF"]);
				/*
				  if (isset($table->columns[$name])) {
				  $table->columns[$name]->isForeignKey = true;
				  }
				 * 
				 */
			}
		}
	}

	/**
	 * Returns all table names in the database.
	 * @param string $schema the schema of the tables. Defaults to empty string, meaning the current or default schema.
	 * If not empty, the returned table names will be prefixed with the schema name.
	 * @return array all table names in the database.
	 */
	protected function findTableNames($schema = '') {
		if ($schema === '') {
			$sql = <<<EOD
SELECT table_name, '{$schema}' as table_schema FROM user_tables
EOD;
			$command = $this->db->createCommand($sql);
		} else {
			$sql = <<<EOD
SELECT object_name as table_name, owner as table_schema FROM all_objects
WHERE object_type = 'TABLE' AND owner=:schema
EOD;
			$command = $this->db > createCommand($sql);
			$command->bindParam(':schema', $schema);
		}

		$rows = $command->queryAll();
		$names = array();
		foreach ($rows as $row) {
			if ($schema === $this->getDefaultSchema() || $schema === '') {
				$names[] = $row['TABLE_NAME'];
			} else {
				$names[] = $row['TABLE_SCHEMA'] . '.' . $row['TABLE_NAME'];
			}
		}
		return $names;
	}

	/**
	 * Builds a SQL statement for renaming a DB table.
	 * @param string $table the table to be renamed. The name will be properly quoted by the method.
	 * @param string $newName the new table name. The name will be properly quoted by the method.
	 * @return string the SQL statement for renaming a DB table.
	 * @since 1.1.6
	 */
	public function renameTable($table, $newName) {
		return 'ALTER TABLE ' . $this->quoteTableName($table) . ' RENAME TO ' . $this->quoteTableName($newName);
	}

	/**
	 * Builds a SQL statement for changing the definition of a column.
	 * @param string $table the table whose column is to be changed. The table name will be properly quoted by the method.
	 * @param string $column the name of the column to be changed. The name will be properly quoted by the method.
	 * @param string $type the new column type. The {@link getColumnType} method will be invoked to convert abstract column type (if any)
	 * into the physical one. Anything that is not recognized as abstract type will be kept in the generated SQL.
	 * For example, 'string' will be turned into 'varchar(255)', while 'string not null' will become 'varchar(255) not null'.
	 * @return string the SQL statement for changing the definition of a column.
	 * @since 1.1.6
	 */
	public function alterColumn($table, $column, $type) {
		$type = $this->getColumnType($type);
		$sql = 'ALTER TABLE ' . $this->quoteTableName($table) . ' MODIFY '
				. $this->quoteColumnName($column) . ' '
				. $this->getColumnType($type);
		return $sql;
	}

	/**
	 * Builds a SQL statement for dropping an index.
	 * @param string $name the name of the index to be dropped. The name will be properly quoted by the method.
	 * @param string $table the table whose index is to be dropped. The name will be properly quoted by the method.
	 * @return string the SQL statement for dropping an index.
	 * @since 1.1.6
	 */
	public function dropIndex($name, $table) {
		return 'DROP INDEX ' . $this->quoteTableName($name);
	}

	/**
	 * Resets the sequence value of a table's primary key.
	 * The sequence will be reset such that the primary key of the next new row inserted
	 * will have the specified value or 1.
	 * @param CDbTableSchema $table the table schema whose primary key sequence will be reset
	 * @param mixed $value the value for the primary key of the next new row inserted. If this is not set,
	 * the next new row's primary key will have a value 1.
	 * @since 1.1.13
	 */
	public function resetSequence($table, $value = 1) {
		$seq = $table->name . "_SEQ";
		if ($table->sequenceName !== null) {
			$this->db->createCommand("DROP SEQUENCE " . $seq)->execute();

			$createSequenceSql = <<< SQL
create sequence $seq
start with $value
increment by 1
nomaxvalue
nocache
SQL;
			$this->db->createCommand($createSequenceSql)->execute();
		}
	}

}