mirror of
https://github.com/10h30/ultimatemember.git
synced 2026-07-11 18:56:10 +09:00
ultimatemember
328 lines
6.9 KiB
PHP
328 lines
6.9 KiB
PHP
<?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/';
|
|
|
|
// create plugin uploads directory
|
|
if (!file_exists( $this->upload_basedir )) {
|
|
@mkdir( $this->upload_basedir, 0777, true);
|
|
}
|
|
|
|
}
|
|
|
|
/***
|
|
*** @Return upload directory
|
|
***/
|
|
function upload_dir(){
|
|
return $this->upload_basedir;
|
|
}
|
|
|
|
/***
|
|
*** @Return upload URL
|
|
***/
|
|
function upload_url(){
|
|
return $this->upload_baseurl;
|
|
}
|
|
|
|
/***
|
|
*** @Generate unique temp directory
|
|
***/
|
|
function unique_dir(){
|
|
global $ultimatemember;
|
|
$unique_number = $ultimatemember->validation->generate();
|
|
$array['dir'] = $this->upload_basedir . 'temp/'. $unique_number . '/';
|
|
$array['url'] = $this->upload_baseurl . 'temp/'. $unique_number . '/';
|
|
return $array;
|
|
}
|
|
|
|
/***
|
|
*** @Fix image orientation
|
|
***/
|
|
function fix_image_orientation(&$image, $source){
|
|
$exif = @exif_read_data($source);
|
|
|
|
if (isset($exif['Orientation'])) {
|
|
switch ($exif['Orientation']) {
|
|
case 3:
|
|
$image = imagerotate($image, 180, 0);
|
|
break;
|
|
|
|
case 6:
|
|
$image = imagerotate($image, -90, 0);
|
|
break;
|
|
|
|
case 8:
|
|
$image = imagerotate($image, 90, 0);
|
|
break;
|
|
}
|
|
}
|
|
|
|
return $image;
|
|
}
|
|
|
|
/***
|
|
*** @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);
|
|
|
|
}
|
|
|
|
$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 = $data['invalid_image'];
|
|
} 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'];
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
} |