<?php
/**
 * @brief		File Storage Extension: ProfileField
 * @author		<a href='https://www.invisioncommunity.com'>Invision Power Services, Inc.</a>
 * @copyright	(c) Invision Power Services, Inc.
 * @license		https://www.invisioncommunity.com/legal/standards/
 * @package		Invision Community
 * @since		29 Aug 2014
 */

namespace IPS\core\extensions\core\FileStorage;

/* To prevent PHP errors (extending class does not exist) revealing path */
if ( !defined( '\IPS\SUITE_UNIQUE_KEY' ) )
{
	header( ( isset( $_SERVER['SERVER_PROTOCOL'] ) ? $_SERVER['SERVER_PROTOCOL'] : 'HTTP/1.0' ) . ' 403 Forbidden' );
	exit;
}

/**
 * File Storage Extension: ProfileField
 */
class _ProfileField
{
	/**
	 * Count stored files
	 *
	 * @return	int
	 */
	public function count()
	{
		$count = 0;
		foreach( \IPS\Db::i()->select( '*', 'core_pfields_data', array( 'pf_type=?', 'Upload' ) ) AS $field )
		{
			$count += \IPS\Db::i()->select( 'COUNT(*)', 'core_pfields_content', array( "field_{$field['pf_id']}<>? OR field_{$field['pf_id']} IS NOT NULL", '' ) )->first();
		}
		
		return $count;
	}
	
	/**
	 * Move stored files
	 *
	 * @param	int			$offset					This will be sent starting with 0, increasing to get all files stored by this extension
	 * @param	int			$storageConfiguration	New storage configuration ID
	 * @param	int|NULL	$oldConfiguration		Old storage configuration ID
	 * @throws	\UnderflowException					When file record doesn't exist. Indicating there are no more files to move
	 * @return	void|int							An offset integer to use on the next cycle, or nothing
	 */
	public function move( $offset, $storageConfiguration, $oldConfiguration=NULL )
	{
		if( !\IPS\Db::i()->select( 'COUNT(*)', 'core_pfields_data', array( 'pf_type=?', 'Upload' ) )->first() )
		{
			throw new \Underflowexception;
		}

		foreach( \IPS\Db::i()->select( '*', 'core_pfields_data', array( 'pf_type=?', 'Upload' ) ) AS $field )
		{
			$cfield	= \IPS\Db::i()->select( '*', 'core_pfields_content', array( "field_{$field['pf_id']}<>? OR field_{$field['pf_id']} IS NOT NULL", '' ), 'member_id', array( $offset, 1 ) )->first();
			
			try
			{
				$file = \IPS\File::get( $oldConfiguration ?: 'core_ProfileField', $cfield[ 'field_' . $field['pf_id'] ] )->move( $storageConfiguration );
				
				if ( (string) $file !== $cfield[ 'field_' . $field['pf_id'] ] )
				{
					\IPS\Db::i()->update( 'core_pfields_content', array( "field_{$field['pf_id']}=?", (string) $file ), array( 'member_id=?', $cfield['member_id'] ) );
				}
			}
			catch( \Exception $e )
			{
				/* Any issues are logged and the \IPS\Db::i()->update not run as the exception is thrown */
			}
		}
	}
	
	/**
	 * Fix all URLs
	 *
	 * @param	int			$offset					This will be sent starting with 0, increasing to get all files stored by this extension
	 * @return void
	 */
	public function fixUrls( $offset )
	{
		if( !\IPS\Db::i()->select( 'COUNT(*)', 'core_pfields_data', array( 'pf_type=?', 'Upload' ) )->first() )
		{
			throw new \Underflowexception;
		}

		foreach( \IPS\Db::i()->select( '*', 'core_pfields_data', array( 'pf_type=?', 'Upload' ) ) AS $field )
		{
			$cfield	= \IPS\Db::i()->select( '*', 'core_pfields_content', array( "field_{$field['pf_id']}<>? OR field_{$field['pf_id']} IS NOT NULL", '' ), 'member_id', array( $offset, 1 ) )->first();
			
			try
			{
				if ( $new = \IPS\File::repairUrl( $cfield[ 'field_' . $field['pf_id'] ] ) )
				{
					\IPS\Db::i()->update( 'core_pfields_content', array( "field_{$field['pf_id']}=?", $new ), array( 'member_id=?', $cfield['member_id'] ) );
				}
			}
			catch( \Exception $e )
			{
				/* Any issues are logged and the \IPS\Db::i()->update not run as the exception is thrown */
			}
		}
	}

	/**
	 * Check if a file is valid
	 *
	 * @param	string	$file		The file path to check
	 * @return	bool
	 */
	public function isValidFile( $file )
	{
		$valid = FALSE;
		foreach( \IPS\Db::i()->select( '*', 'core_pfields_data', array( 'pf_type=?', 'Upload' ) ) AS $field )
		{
			try
			{
				\IPS\Db::i()->select( '*', 'core_pfields_content', array( "field_{$field['pf_id']}=?", (string) $file ) )->first();
				
				$valid = TRUE;
				break;
			}
			catch( \UnderflowException $e ) {}
		}
		
		return $valid;
	}
}