<?php

class UM_Files {

	function __construct() {

		add_action('init',  array(&$this, 'setup_paths'), 1);
		
		$this->fonticon = array(
			'pdf' => array('icon' => 'um-icon-file-pdf', 'color' => '#D24D4D' ),
			'txt' => array('icon' => 'um-icon-file-text' ),
			'csv' => array('icon' => 'um-icon-file-text' ),
			'doc' => array('icon' => 'um-icon-file-text', 'color' => '#2C95D5' ),
			'docx' => array('icon' => 'um-icon-file-text', 'color' => '#2C95D5' ),
			'odt' => array('icon' => 'um-icon-file-text', 'color' => '#2C95D5' ),
			'ods' => array('icon' => 'um-icon-file-binary', 'color' => '#51BA6A' ),
			'xls' => array('icon' => 'um-icon-file-binary', 'color' => '#51BA6A' ),
			'xlsx' => array('icon' => 'um-icon-file-binary', 'color' => '#51BA6A' ),
			'zip' => array('icon' => 'um-icon-file-zip' ),
			'rar' => array('icon' => 'um-icon-file-zip' ),
		);
		
		$this->default_file_fonticon = 'um-icon-doc';
	
	}
	
	/***
	***	@allowed image types
	***/
	function allowed_image_types() {
	
		$array['png'] = 'PNG';
		$array['jpeg'] = 'JPEG';
		$array['jpg'] = 'JPG';
		$array['gif'] = 'GIF';
		
		$array = apply_filters('um_allowed_image_types', $array);
		return $array;
	}
	
	/***
	***	@allowed file types
	***/
	function allowed_file_types() {
	
		$array['pdf'] = 'PDF';
		$array['txt'] = 'Text';
		$array['csv'] = 'CSV';
		$array['doc'] = 'DOC';
		$array['docx'] = 'DOCX';
		$array['odt'] = 'ODT';
		$array['ods'] = 'ODS';
		$array['xls'] = 'XLS';
		$array['xlsx'] = 'XLSX';
		$array['zip'] = 'ZIP';
		$array['rar'] = 'RAR';
		
		$array = apply_filters('um_allowed_file_types', $array);
		return $array;
	}
	
	/***
	***	@Get extension icon
	***/
	function get_fonticon_by_ext( $extension ) {
		if (isset($this->fonticon[$extension]['icon'])){
			return $this->fonticon[$extension]['icon'];
		} else {
			return $this->default_file_fonticon;
		}
	}
	
	/***
	***	@Get extension icon background
	***/
	function get_fonticon_bg_by_ext( $extension ) {
		if (isset($this->fonticon[$extension]['color'])){
			return $this->fonticon[$extension]['color'];
		} else {
			return '#666';
		}
	}
	
	/***
	***	@Setup upload directory
	***/
	function setup_paths(){
	
		$this->upload_dir = wp_upload_dir();
		
		$this->upload_basedir = $this->upload_dir['basedir'] . '/ultimatemember/';
		$this->upload_baseurl = $this->upload_dir['baseurl'] . '/ultimatemember/';
		
		$this->upload_temp = $this->upload_basedir . 'temp/';
		$this->upload_temp_url = $this->upload_baseurl . 'temp/';

		if (!file_exists( $this->upload_basedir )) {
			@mkdir( $this->upload_basedir, 0777, true);
		}

		if (!file_exists( $this->upload_temp )) {
			@mkdir( $this->upload_temp , 0777, true);
		}
		
	}
	
	/***
	***	@Generate unique temp directory
	***/
	function unique_dir(){
		global $ultimatemember;
		$unique_number = $ultimatemember->validation->generate();
		$array['dir'] = $this->upload_temp . $unique_number . '/';
		$array['url'] = $this->upload_temp_url . $unique_number . '/';
		return $array;
	}
	
	/***
	***	@get path only without file name
	***/
	function path_only( $file ) {
		return trailingslashit( dirname( $file ) );
	}
	
	/***
	***	@fix image orientation
	***/
	function fix_image_orientation($rotate, $source){
		$exif = @exif_read_data($source);

		if (isset($exif['Orientation'])) {
			switch ($exif['Orientation']) {
				case 3:
					$rotate = imagerotate($rotate, 180, 0);
					break;

				case 6:
					$rotate = imagerotate($rotate, -90, 0);
					break;

				case 8:
					$rotate = imagerotate($rotate, 90, 0);
					break;
			}
		}
		
		return $rotate;
	}
	
	/***
	***	@Process an image
	***/
	function create_and_copy_image($source, $destination, $quality = 100) {
		
		$info = @getimagesize($source);
		
		if ($info['mime'] == 'image/jpeg'){
		
			$image = imagecreatefromjpeg($source);
	
		} else if ($info['mime'] == 'image/gif'){
		
			$image = imagecreatefromgif($source);

		} else if ($info['mime'] == 'image/png'){
		
			$image = imagecreatefrompng($source);

		}

		list($w, $h) = @getimagesize( $source );
		if ( $w > um_get_option('image_max_width') ) {
		
			$ratio = round( $w / $h, 2 );
			$new_w = um_get_option('image_max_width');
			$new_h = round( $new_w / $ratio, 2 );
			
			$image_p = imagecreatetruecolor( $new_w, $new_h );
			imagecopyresampled( $image_p, $image, 0, 0, 0, 0, $new_w, $new_h, $w, $h );
			$image_p = $this->fix_image_orientation($image_p, $source);
			imagejpeg( $image_p, $destination, $quality);
		
		} else {
			
			$image = $this->fix_image_orientation($image, $source);
			imagejpeg( $image, $destination, $quality);
			
		}

	}
	
	/***
	***	@Process a file
	***/
	function upload_temp_file($source, $destination) {
		
		move_uploaded_file($source, $destination);
				
	}

	/***
	***	@Process a temp upload
	***/
	function new_image_upload_temp($source, $destination, $quality = 100){
	
		$unique_dir = $this->unique_dir();
		
		$this->make_dir( $unique_dir['dir'] );

		$this->create_and_copy_image($source, $unique_dir['dir'] . $destination, $quality);
		
		$url = $unique_dir['url'] . $destination;

		return $url;
		
	}
	
	/***
	***	@Process a temp upload for files
	***/
	function new_file_upload_temp($source, $destination ){
	
		$unique_dir = $this->unique_dir();
		
		$this->make_dir( $unique_dir['dir'] );

		$this->upload_temp_file($source, $unique_dir['dir'] . $destination);
		
		$url = $unique_dir['url'] . $destination;

		return $url;
		
	}
	
	/***
	***	@Make a Folder
	***/
	function make_dir( $dir ){
	
		@mkdir( $dir, 0777, true);
		
	}
	
	/***
	***	@Get extension by mime type
	***/
	function get_extension_by_mime_type($mime){
		$split = explode('/',$mime);
		return $split[1];
	}
	
	/***
	***	@Get file data
	***/
	function get_file_data($file){
	
		$array['size'] = filesize($file);

		return $array;
	}
	
	/***
	***	@Get image data
	***/
	function get_image_data($file){
	
		$array['size'] = filesize($file);
		
		$array['image'] = @getimagesize($file);
		
		if ( $array['image'] > 0 ) {
		
			$array['invalid_image'] = false;
			
			list($width, $height, $type, $attr) = @getimagesize($file);
			
			$array['width'] = $width;
			$array['height'] = $height;
			$array['ratio'] = $width / $height;
			
			$array['extension'] = $this->get_extension_by_mime_type( $array['image']['mime'] );
		
		} else {
		
			$array['invalid_image'] = true;
			
		}
		
		return $array;
	}
	
	/***
	***	@Check image upload and handle errors
	***/
	function check_image_upload($file, $field) {
		global $ultimatemember;
		$error = null;
		
		$fileinfo = $this->get_image_data($file);
		$data = $ultimatemember->fields->get_field($field);
		
		if ( $fileinfo['invalid_image'] == true ) {
			$error = sprintf(__('Your image is invalid or too large!','ultimatemember') );
		} elseif ( !$this->in_array( $fileinfo['extension'], $data['allowed_types'] ) ) {
			$error = $data['extension_error'];
		} elseif ( isset($data['min_size']) && ( $fileinfo['size'] < $data['min_size'] ) ) {
			$error = $data['min_size_error'];
		} elseif ( isset($data['min_width']) && ( $fileinfo['width'] < $data['min_width'] ) ) {
			$error = sprintf(__('Your photo is too small. It must be at least %spx wide.'), $data['min_width']);
		} elseif ( isset($data['min_height']) && ( $fileinfo['height'] < $data['min_height'] ) ) {
			$error = sprintf(__('Your photo is too small. It must be at least %spx wide.'), $data['min_height']);
		}
		
		return $error;
	}
	
	/***
	***	@Check file upload and handle errors
	***/
	function check_file_upload($file, $extension, $field) {
		global $ultimatemember;
		$error = null;

		$fileinfo = $this->get_file_data($file);
		$data = $ultimatemember->fields->get_field($field);
		
		if ( !$this->in_array( $extension, $data['allowed_types'] ) ) {
			$error = $data['extension_error'];
		} elseif ( isset($data['min_size']) && ( $fileinfo['size'] < $data['min_size'] ) ) {
			$error = $data['min_size_error'];
		}
		
		return $error;
	}
	
	/***
	***	@If a value exists in comma seperated list
	***/
	function in_array($value, $array){
		if (in_array($value, explode(',',$array)))
			return true;
		return false;
	}
	
	/***
	***	@delete a main user photo
	***/
	function delete_core_user_photo( $user_id, $type ) {
	
		delete_user_meta( $user_id, $type );
		
		$dir = $this->upload_basedir . $user_id . '/';
		$prefix = $type;
		chdir($dir);
		$matches = glob($prefix.'*',GLOB_MARK);
		
		if( is_array($matches) && !empty($matches)) {
			foreach($matches as $match) {
				if( is_file($dir.$match) ) unlink($dir.$match);
			}
		}
		
		if ( count(glob("$dir/*")) === 0) {
			rmdir( $dir );
		}
		
	}

	/***
	***	@resize a local image
	***/
	function resize_image( $file, $crop ) {
	
		$targ_x1 = $crop[0];
		$targ_y1 = $crop[1];
		$targ_x2 = $crop[2];
		$targ_y2 = $crop[3];

		$img_r = imagecreatefromjpeg($file);
		$dst_r = imagecreatetruecolor( $targ_x2, $targ_y2 );

		imagecopy( $dst_r, $img_r, 0, 0, $targ_x1, $targ_y1, $targ_x2, $targ_y2 );
		imagejpeg( $dst_r, $this->path_only( $file ) . basename( $file ), 100);
		
		$split = explode('/ultimatemember/temp/', $file);
		return $this->upload_temp_url . $split[1];
		
	}
	
	/***
	***	@new user upload
	***/
	function new_user_upload( $user_id, $source, $key ) {
	
		// if he does not have uploads dir yet
		if ( !file_exists( $this->upload_basedir . $user_id . '/' ) ) {
			@mkdir( $this->upload_basedir . $user_id . '/' , 0777, true);
		}
		
		// name and extension stuff
		$source_name = basename( $source );
		
		if ( $key == 'profile_photo' ) {
			$source_name = 'profile_photo.jpg';
		}
		
		if ( $key == 'cover_photo' ) {
			$source_name = 'cover_photo.jpg';
		}
		
		$ext = '.' . pathinfo($source_name, PATHINFO_EXTENSION);
		$name = str_replace( $ext, '', $source_name );
		$filename = $name . $ext;

		// copy file
		copy( $source, $this->upload_basedir . $user_id . '/' . $filename );
		
		// thumbs
		if ( $key == 'profile_photo' ) {
		
			list($w, $h) = @getimagesize( $source );
			
			$sizes = um_get_option('photo_thumb_sizes');
			foreach( $sizes as $size ) {
			
				if ( file_exists(  $this->upload_basedir . $user_id . '/' . $name . '-' . $size . $ext ) ) {
					unlink( $this->upload_basedir . $user_id . '/' . $name . '-' . $size . $ext );
				}
				
				if ( $size < $w ) {

				$thumb_s = imagecreatefromjpeg( $source );
				$thumb = imagecreatetruecolor( $size, $size );
				imagecopyresampled( $thumb, $thumb_s, 0, 0, 0, 0, $size, $size, $w, $h );
				imagejpeg( $thumb, $this->upload_basedir . $user_id . '/' . $name . '-' . $size . $ext, 100);
				
				}
			
			}
		
		}
		
		if ( $key == 'cover_photo' ) {
		
			list($w, $h) = @getimagesize( $source );
			
			$sizes = um_get_option('cover_thumb_sizes');
			foreach( $sizes as $size ) {
			
				$ratio = round( $w / $h, 2 );
				$height = round( $size / $ratio, 2 );
				
				if ( file_exists(  $this->upload_basedir . $user_id . '/' . $name . '-' . $size . $ext ) ) {
					unlink( $this->upload_basedir . $user_id . '/' . $name . '-' . $size . $ext );
				}
				
				if ( $size < $w ) {

				$thumb_s = imagecreatefromjpeg( $source );
				$thumb = imagecreatetruecolor( $size, $height );
				imagecopyresampled( $thumb, $thumb_s, 0, 0, 0, 0, $size, $height, $w, $h );
				imagejpeg( $thumb, $this->upload_basedir . $user_id . '/' . $name . '-' . $size . $ext, 100);
				
				}
			
			}
		
		}

		// clean up temp
		$dir = dirname( $source );
		unlink( $source );
		rmdir( $dir );

		// update user's meta
		update_user_meta( $user_id, $key, $filename );

		// the url of upload
		return $this->upload_baseurl . $user_id . '/' . $filename;
		
	}
	
}