<?php
/* Prohibit direct script loading */
defined('ABSPATH') || die('No direct script access allowed!');
use Joomunited\WPMediaFolder\WpmfHelper;

/**
 * Class WpmfFolderAccess
 */
class WpmfFolderAccess
{
    /**
     * WpmfFolderAccess constructor.
     */
    public function __construct()
    {
        add_action('init', array($this, 'createUserFolder'));
        add_filter('wp_generate_attachment_metadata', array($this, 'autoAddAttachmentToFolder'), 10, 2);
    }

    /**
     * Create user folder
     *
     * @return void
     */
    public function createUserFolder()
    {
        // insert term when user login and enable option 'Display only media by User/User'
        global $current_user;
        $user_roles = $current_user->roles;
        $role = array_shift($user_roles);
        /**
         * Filter check capability of current user when create user folder
         *
         * @param boolean The current user has the given capability
         * @param string  Action name
         *
         * @return boolean
         *
         * @ignore Hook already documented
         */
        $wpmf_capability = apply_filters('wpmf_user_can', current_user_can('upload_files'), 'create_user_folder');
        if (($role !== 'administrator' && $wpmf_capability) || $role === 'employer') {
            $wpmf_create_folder = get_option('wpmf_create_folder');
            if ($wpmf_create_folder === 'user') {
                $slug       = sanitize_title($current_user->data->user_login) . '-wpmf';
                $check_term = get_term_by('slug', $slug, WPMF_TAXO);
                if (empty($check_term)) {
                    $parent = $this->getUserParentFolder();
                    $this->doCreateUserFolder($current_user->data->user_login, $slug, $parent, $current_user);
                }

                $google_connect = WpmfHelper::isConnected('google_drive');
                $dropbox_connect = WpmfHelper::isConnected('dropbox');
                $onedrive_connect = WpmfHelper::isConnected('onedrive');
                $onedrive_business_connect = WpmfHelper::isConnected('onedrive_business');
                if ($google_connect) {
                    $is_exist = get_term_by('slug', $current_user->data->user_login . '-wpmf-google_drive', WPMF_TAXO);
                    if (empty($is_exist)) {
                        $parent = WpmfHelper::getCloudRootFolderID('google_drive');
                        if ($parent) {
                            $this->doCreateUserFolder($current_user->data->user_login, $current_user->data->user_login . '-wpmf-google_drive', $parent, $current_user);
                        }
                    }
                }

                if ($dropbox_connect) {
                    $is_exist = get_term_by('slug', $current_user->data->user_login . '-wpmf-dropbox', WPMF_TAXO);
                    if (empty($is_exist)) {
                        $parent = WpmfHelper::getCloudRootFolderID('dropbox');
                        if ($parent) {
                            $this->doCreateUserFolder($current_user->data->user_login, $current_user->data->user_login . '-wpmf-dropbox', $parent, $current_user);
                        }
                    }
                }

                if ($onedrive_connect) {
                    $is_exist = get_term_by('slug', $current_user->data->user_login . '-wpmf-onedrive', WPMF_TAXO);
                    if (empty($is_exist)) {
                        $parent = WpmfHelper::getCloudRootFolderID('onedrive');
                        if ($parent) {
                            $this->doCreateUserFolder($current_user->data->user_login, $current_user->data->user_login . '-wpmf-onedrive', $parent, $current_user);
                        }
                    }
                }

                if ($onedrive_business_connect) {
                    $is_exist = get_term_by('slug', $current_user->data->user_login . '-wpmf-onedrive_business', WPMF_TAXO);
                    if (empty($is_exist)) {
                        $parent = WpmfHelper::getCloudRootFolderID('onedrive_business');
                        if ($parent) {
                            $this->doCreateUserFolder($current_user->data->user_login, $current_user->data->user_login . '-wpmf-onedrive_business', $parent, $current_user);
                        }
                    }
                }
            } elseif ($wpmf_create_folder === 'role') {
                $slug       = sanitize_title($role) . '-wpmf-role';
                $check_term = get_term_by('slug', $slug, WPMF_TAXO);
                if (empty($check_term)) {
                    $inserted = wp_insert_term($role, WPMF_TAXO, array('parent' => 0, 'slug' => $slug));
                    if (!is_wp_error($inserted)) {
                        $role = WpmfHelper::getRoles($current_user->data->ID);
                        add_term_meta((int)$inserted['term_id'], 'wpmf_folder_role_permissions', array($role, 'add_media', 'move_media', 'view_folder', 'add_folder', 'update_folder', 'remove_folder', 'view_media', 'remove_media', 'update_media'));
                        do_action('wpmf_create_folder', $inserted['term_id'], $role, 0, array('trigger' => 'media_library_action'));
                    }
                }
            }

            $google_connect = WpmfHelper::isConnected('google_drive');
            $dropbox_connect = WpmfHelper::isConnected('dropbox');
            $onedrive_connect = WpmfHelper::isConnected('onedrive');
            $onedrive_business_connect = WpmfHelper::isConnected('onedrive_business');
            if ($google_connect) {
                $options = get_option('_wpmfAddon_cloud_config');
                if (!empty($options['access_by']) && $options['access_by'] === 'user') {
                    $is_exist = get_term_by('slug', $current_user->data->user_login . '-wpmf-google_drive', WPMF_TAXO);
                    $name = $current_user->data->user_login;
                    $slug = $current_user->data->user_login . '-wpmf-google_drive';
                    $type = '';
                } else {
                    $is_exist = get_term_by('slug', $role . '-wpmf-role-google_drive', WPMF_TAXO);
                    $name = $role;
                    $slug = $role . '-wpmf-role-google_drive';
                    $type = 'role';
                }

                if (empty($is_exist)) {
                    $parent = WpmfHelper::getCloudRootFolderID('google_drive');
                    if ($parent) {
                        $this->doCreateUserFolder($name, $slug, $parent, $current_user, $type);
                    }
                }
            }

            if ($dropbox_connect) {
                $options = get_option('_wpmfAddon_dropbox_config');
                if (!empty($options['access_by']) && $options['access_by'] === 'user') {
                    $is_exist = get_term_by('slug', $current_user->data->user_login . '-wpmf-dropbox', WPMF_TAXO);
                    $name = $current_user->data->user_login;
                    $slug = $current_user->data->user_login . '-wpmf-dropbox';
                    $type = '';
                } else {
                    $is_exist = get_term_by('slug', $role . '-wpmf-role-dropbox', WPMF_TAXO);
                    $name = $role;
                    $slug = $role . '-wpmf-role-dropbox';
                    $type = 'role';
                }

                if (empty($is_exist)) {
                    $parent = WpmfHelper::getCloudRootFolderID('dropbox');
                    if ($parent) {
                        $this->doCreateUserFolder($name, $slug, $parent, $current_user, $type);
                    }
                }
            }

            if ($onedrive_connect) {
                $options = get_option('_wpmfAddon_onedrive_config');
                if (!empty($options['access_by']) && $options['access_by'] === 'user') {
                    $is_exist = get_term_by('slug', $current_user->data->user_login . '-wpmf-onedrive', WPMF_TAXO);
                    $name = $current_user->data->user_login;
                    $slug = $current_user->data->user_login . '-wpmf-onedrive';
                    $type = '';
                } else {
                    $is_exist = get_term_by('slug', $role . '-wpmf-role-onedrive', WPMF_TAXO);
                    $name = $role;
                    $slug = $role . '-wpmf-role-onedrive';
                    $type = 'role';
                }

                if (empty($is_exist)) {
                    $parent = WpmfHelper::getCloudRootFolderID('onedrive');
                    if ($parent) {
                        $this->doCreateUserFolder($name, $slug, $parent, $current_user, $type);
                    }
                }
            }

            if ($onedrive_business_connect) {
                $options = get_option('_wpmfAddon_onedrive_business_config');
                if (!empty($options['access_by']) && $options['access_by'] === 'user') {
                    $is_exist = get_term_by('slug', $current_user->data->user_login . '-wpmf-onedrive_business', WPMF_TAXO);
                    $name = $current_user->data->user_login;
                    $slug = $current_user->data->user_login . '-wpmf-onedrive_business';
                    $type = '';
                } else {
                    $is_exist = get_term_by('slug', $role . '-wpmf-role-onedrive_business', WPMF_TAXO);
                    $name = $role;
                    $slug = $role . '-wpmf-role-onedrive_business';
                    $type = 'role';
                }

                if (empty($is_exist)) {
                    $parent = WpmfHelper::getCloudRootFolderID('onedrive_business');
                    if ($parent) {
                        $this->doCreateUserFolder($name, $slug, $parent, $current_user, $type);
                    }
                }
            }
        }
    }

    /**
     * Do create user folder
     *
     * @param string  $name         Folder name
     * @param string  $slug         Folder slug
     * @param integer $parent       Folder parent
     * @param object  $current_user Current_user
     * @param string  $type         Type
     *
     * @return void
     */
    public function doCreateUserFolder($name, $slug, $parent, $current_user, $type = 'user')
    {
        $inserted = wp_insert_term(
            $name,
            WPMF_TAXO,
            array('parent' => $parent, 'slug' => $slug)
        );
        if (!is_wp_error($inserted)) {
            if ($type === 'user') {
                wp_update_term($inserted['term_id'], WPMF_TAXO, array('term_group' => $current_user->data->ID));
                add_term_meta((int)$inserted['term_id'], 'wpmf_folder_user_permissions', array($current_user->data->ID, 'add_media', 'move_media', 'view_folder', 'add_folder', 'update_folder', 'remove_folder', 'view_media', 'remove_media', 'update_media'));
                add_term_meta((int)$inserted['term_id'], 'inherit_folder', 0);
            } else {
                $role = WpmfHelper::getRoles($current_user->data->ID);
                add_term_meta((int)$inserted['term_id'], 'wpmf_folder_role_permissions', array($role, 'add_media', 'move_media', 'view_folder', 'add_folder', 'update_folder', 'remove_folder', 'view_media', 'remove_media', 'update_media'));
                add_term_meta((int)$inserted['term_id'], 'inherit_folder', 0);
            }

            // do_action('wpmf_create_folder', $inserted['term_id'], $name, $parent, array('trigger' => 'media_library_action'));
        }
    }

    /**
     * Auto add attachment to folder
     *
     * @param array   $data    Meta data
     * @param integer $post_id Attachment ID
     *
     * @return array
     */
    public function autoAddAttachmentToFolder($data, $post_id)
    {
        $active_media = get_option('wpmf_active_media');
        if (isset($active_media) && (int) $active_media === 1) {
            $wpmf_create_folder = get_option('wpmf_create_folder');
            if ($wpmf_create_folder === 'user') {
                global $wpdb, $current_user;
                if (!empty($current_user->ID)) {
                    $user_login = $current_user->data->user_login;
                    $user_roles = $current_user->roles;
                    $role = array_shift($user_roles);
                    if ($role === 'employer') {
                        $parent = $this->getUserParentFolder();
                        $user_folder = $wpdb->get_row($wpdb->prepare('SELECT * FROM ' . $wpdb->terms . ' as t INNER JOIN ' . $wpdb->term_taxonomy . ' AS tt ON tt.term_id = t.term_id WHERE t.name = %s AND t.term_group = %d AND tt.parent = %d AND tt.taxonomy = %s', array($user_login, (int) $current_user->ID, (int) $parent, WPMF_TAXO)));
                        if (!empty($user_folder)) {
                            wp_set_object_terms($post_id, $user_folder->term_id, WPMF_TAXO, true);
                        }
                    }
                }
            }
        }

        return $data;
    }

    /**
     * Get user parent folder
     *
     * @return integer|mixed|void
     */
    public function getUserParentFolder()
    {
        $wpmf_checkbox_tree = get_option('wpmf_checkbox_tree');
        $parent = 0;
        if (!empty($wpmf_checkbox_tree)) {
            $current_parrent = get_term($wpmf_checkbox_tree, WPMF_TAXO);
            if (!empty($current_parrent)) {
                $parent = $wpmf_checkbox_tree;
            }
        }

        return $parent;
    }
}