Handle wiaas documents

backend/app/plugins/wiaas/includes/document/class-wiaas-document-upload.php

@@ -0,0 +1,88 @@
+<?php
+
+class Wiaas_Document_Upload {
+
+	private static $allowed_extensions = array(
+		'pdf','docx','doc','xlsx','xls','odt','ods', 'zip'
+	);
+
+	public static function init() {
+		add_filter( 'upload_dir', array( __CLASS__, 'upload_dir' ) );
+	}
+
+
+	/**
+	 * Allowed extensions for wiaas documents uploads
+	 * @return array
+	 */
+	public static function allowed_extensions() {
+		return self::$allowed_extensions;
+	}
+
+
+	/**
+	 * Handle wiaas document upload
+	 *
+	 * @param string $file_id
+	 *
+	 * @return int|WP_Error
+	 */
+	public static function upload_document_version($file_id = 'file') {
+
+		wc_maybe_define_constant( 'WIAAS_DOCUMENT_UPLOAD', true );
+
+		// validate file
+
+		if (!isset($_FILES[$file_id])) {
+			return new WP_Error( 'wiaas_upload_missing_file', 'No file!' );
+		}
+
+		$file_name = $_FILES[$file_id]['name'];
+		$name = esc_attr($file_name);
+
+		$ext = explode('.', $name);
+		$ext = end($ext);
+		$ext = strtolower($ext);
+
+		if(!in_array($ext, self::$allowed_extensions)) {
+			return new WP_Error( 'wiaas_upload_error', 'Invalid file extension!' );
+		}
+
+		$file = wp_handle_upload(
+			$_FILES[$file_id],
+			array( 'test_form' => false ),
+			current_time('mysql'));
+
+		if ( isset($file['error']) )
+			return new WP_Error( 'wiaas_upload_error', $file['error'] );
+
+		return wiaas_get_doc_version_from_path($file['file']);
+	}
+
+	/**
+	 * Updates wordpress upload dir so that wiaas documents are uploaded to wiaas specific upload dir
+	 * @param $pathdata
+	 *
+	 * @return mixed
+	 */
+	public static function upload_dir( $pathdata ) {
+
+		if (defined('WIAAS_DOCUMENT_UPLOAD') ) {
+			if ( empty( $pathdata['subdir'] ) ) {
+				$pathdata['path']   = $pathdata['path'] . wiaas_documents_base_dir();
+				$pathdata['url']    = $pathdata['url'] . wiaas_documents_base_dir();
+				$pathdata['subdir'] = wiaas_documents_base_dir();
+			} else {
+				$new_subdir = wiaas_documents_base_dir() . $pathdata['subdir'];
+
+				$pathdata['path']   = str_replace( $pathdata['subdir'], $new_subdir, $pathdata['path'] );
+				$pathdata['url']    = str_replace( $pathdata['subdir'], $new_subdir, $pathdata['url'] );
+				$pathdata['subdir'] = str_replace( $pathdata['subdir'], $new_subdir, $pathdata['subdir'] );
+			}
+		}
+
+		return $pathdata;
+	}
+}
+
+Wiaas_Document_Upload::init();