<?php

use Gravity_Forms\Gravity_Tools\Hermes\Enum\Field_Type_Validation_Enum;
use Gravity_Forms\Gravity_Tools\Hermes\Models\Model;
use Gravity_Forms\Gravity_Tools\Hermes\Utils\Relationship;
use Gravity_Forms\Gravity_Tools\Hermes\Utils\Relationship_Collection;
use tad\FunctionMocker\FunctionMocker;

require_once __DIR__ . '/vendor/autoload.php';

FunctionMocker::init();

/**
 * Bootstrap for WP Unit Tests
 */
require __DIR__ . '/wp-loader.php';
$core_loader = new WpLoader();
$core_loader->init();

class FakeUserModel extends Model {
	protected $type = 'user';
	protected $access_cap = 'manage_options';
	protected $forced_table_name = 'users';

	public function fields() {
		return array(
			'ID' => Field_Type_Validation_Enum::INT,
			'user_login' => Field_Type_Validation_Enum::STRING,
			'user_nicename' => Field_Type_Validation_Enum::STRING,
			'user_email' => Field_Type_Validation_Enum::EMAIL,
			'user_url' => Field_Type_Validation_Enum::STRING,
			'user_registered' => Field_Type_Validation_Enum::DATE,
			'user_status' => Field_Type_Validation_Enum::INT,
			'display_name' => Field_Type_Validation_Enum::STRING,
		);
	}

	public function relationships() {
		return new Relationship_Collection( array(
			new Relationship( 'user', 'deal', 'manage_options', false, 'one_to_many' ),
		) );
	}
}

class FakeWebsiteModel extends \Gravity_Forms\Gravity_Tools\Hermes\Models\Model {
	protected $type = 'website';
	protected $access_cap = 'manage_options';

	public function fields() {
		return array(
			'id' => Field_Type_Validation_Enum::INT,
			'type' => Field_Type_Validation_Enum::STRING,
			'url' => Field_Type_Validation_Enum::STRING,
		);
	}

	public function relationships() {
		return new Relationship_Collection( array() );
	}
}

class FakePhoneModel extends \Gravity_Forms\Gravity_Tools\Hermes\Models\Model {
	protected $type = 'phone';
	protected $access_cap = 'manage_options';

	public function fields() {
		return array(
			'id' => Field_Type_Validation_Enum::INT,
			'type' => Field_Type_Validation_Enum::STRING,
			'number' => Field_Type_Validation_Enum::INT,
			'countryCode' => Field_Type_Validation_Enum::STRING,
		);
	}

	public function relationships() {
		return new Relationship_Collection( array() );
	}
}

class FakeEmailModel extends \Gravity_Forms\Gravity_Tools\Hermes\Models\Model {

	protected $type = 'email';

	protected $access_cap = 'manage_options';

	public function fields() {
		return array(
			'id'      => Field_Type_Validation_Enum::INT,
			'address' => Field_Type_Validation_Enum::STRING,
			'type'    => Field_Type_Validation_Enum::STRING,
		);
	}

	public function relationships() {
		return new Relationship_Collection( array() );
	}
}

class FakeContactModel extends \Gravity_Forms\Gravity_Tools\Hermes\Models\Model {

	protected $type = 'contact';

	protected $access_cap = 'manage_options';

	public function fields() {
		return array(
			'id'              => Field_Type_Validation_Enum::INT,
			'firstName'       => Field_Type_Validation_Enum::STRING,
			'lastName'        => Field_Type_Validation_Enum::STRING,
			'profile_picture' => Field_Type_Validation_Enum::INT,
			'foobar'          => function ( $value ) {
				if ( $value === 'foo' ) {
					return 'foo';
				}

				return null;
			},
		);
	}

	public function meta_fields() {
		return array(
			'secondary_phone'   => Field_Type_Validation_Enum::STRING,
			'alternate_website' => Field_Type_Validation_Enum::STRING,
		);
	}

	public function transformations() {
		return array(
			'transformMakeThumb' => function ( $thumb_id, $thumb_type = 'hexagon', $thumb_size = 'medium' ) {
				return sprintf( 'thumbnail_url:%s/%s/%s', $thumb_id, $thumb_type, $thumb_size );
			},
		);
	}

	public function relationships() {
		return new \Gravity_Forms\Gravity_Tools\Hermes\Utils\Relationship_Collection(
			array(
				new Relationship( 'contact', 'email', 'manage_options', false, 'one_to_many' ),
				new Relationship( 'contact', 'phone', 'manage_options', false, 'one_to_many' ),
				new Relationship( 'contact', 'website', 'manage_options', false, 'one_to_many' ),
			)
		);
	}
}

class FakeCompanyModel extends \Gravity_Forms\Gravity_Tools\Hermes\Models\Model {

	protected $type = 'company';

	public function fields() {
		return array(
			'id'          => Field_Type_Validation_Enum::STRING,
			'companyName' => Field_Type_Validation_Enum::STRING,
		);
	}

	protected $access_cap = 'manage_options';

	public function transformations() {
		return array(
			'transformMakeFoo' => function ( $arg, $value ) {
				return sprintf( "IM%s", $arg );
			},
		);
	}
	public function relationships() {
		$relationships = array(
			new Relationship( 'company', 'contact', 'manage_options' ),
			new Relationship( 'company', 'email', 'manage_options', false, 'one_to_many' ),
			new Relationship( 'company', 'phone', 'manage_options', false, 'one_to_many'  ),
			new Relationship( 'company', 'website', 'manage_options', false, 'one_to_many'  ),
		);

		return new Relationship_Collection( $relationships );
	}
}

class FakeGroupModel extends \Gravity_Forms\Gravity_Tools\Hermes\Models\Model {

	protected $type = 'group';

	protected $fields = array(
		'label',
	);

	public function fields() {
		return array(
			'label' => Field_Type_Validation_Enum::STRING,
		);
	}

	protected $access_cap = 'manage_options';

	public function relationships() {
		return new \Gravity_Forms\Gravity_Tools\Hermes\Utils\Relationship_Collection(
			array(
				new \Gravity_Forms\Gravity_Tools\Hermes\Utils\Relationship( 'group', 'contact', 'manage_options' ),
			)
		);
	}
}

class FakeDealModel extends \Gravity_Forms\Gravity_Tools\Hermes\Models\Model {

	protected $type = 'deal';

	public function fields() {
		return array(
			'label' => Field_Type_Validation_Enum::STRING,
			'value' => Field_Type_Validation_Enum::INT,
			'id' => Field_Type_Validation_Enum::INT,
		);
	}

	protected $access_cap = 'manage_options';

	public function relationships() {
		return new \Gravity_Forms\Gravity_Tools\Hermes\Utils\Relationship_Collection(
			array(
				new \Gravity_Forms\Gravity_Tools\Hermes\Utils\Relationship( 'deal', 'stage', 'manage_options', true, 'one_to_many' ),
				new Relationship( 'deal', 'user', 'manage_options', true, 'one_to_many' ),
			)
		);
	}
}

class FakeStageModel extends \Gravity_Forms\Gravity_Tools\Hermes\Models\Model {

	protected $type = 'stage';

	public function fields() {
		return array(
			'id' => Field_Type_Validation_Enum::INT,
			'label' => Field_Type_Validation_Enum::STRING,
		);
	}

	protected $access_cap = 'manage_options';

	public function relationships() {
		return new \Gravity_Forms\Gravity_Tools\Hermes\Utils\Relationship_Collection(
			array()
		);
	}
}

function gravitytools_tests_reset_db() {
	echo "\r\n";
	echo '=========================================' . "\r\n";
	echo 'Cleaning up test database for next run...' . "\r\n";
	global $wpdb;

	$tables = array(
		'contact',
		'company',
		'deal',
		'pipeline',
		'company_contact',
		'deal_company',
		'deal_contact',
		'address',
		'social',
		'meta',
		'email',
		'phone',
		'website',
		'stage',
	);

	foreach ( $tables as $table ) {
		$table_name = sprintf( '%s%s_%s', $wpdb->prefix, 'gravitycrm', $table );
		$sql        = sprintf( 'TRUNCATE TABLE %s', $table_name );
		$wpdb->query( $sql );
	}

	echo 'Done cleaning test database!' . "\r\n";
	echo '=========================================' . "\r\n";
	echo "\r\n";
}