<?php
/**
 * BuddyDrive Items classes
 */

// Exit if accessed directly.
defined( 'ABSPATH' ) || exit;

/**
 * The "CRUD" BuddyDrive Items class
 *
 * @package BuddyDrive
 * @since 1.0
 */
class BuddyDrive_Item {

	public $user_id;
	public $id;
	public $type;
	public $parent_folder_id;
	public $title;
	public $content;
	public $mime_type;
	public $guid;
	public $metas;


	public function __construct( $id = 0 ){
		if ( ! empty( $id ) ) {
			$this->id = $id;
			$this->populate();
		}
	}

	/**
	 * request an item id
	 *
	 * @uses WP_Query
	 */
	public function populate() {

		$query_args = array(
			'post_status'	 => 'publish',
			'p' => intval( $this->id )
		);

		$this->query = new WP_Query( $query_args );

	}

	/**
	 * Saves or updates the BuddyDrive items
	 *
	 * @uses wp_update_post() to update an item if it exists
	 * @uses update_post_meta() to update/create privacy options
	 * @uses delete_post_meta() to eventually remove privacy options
	 * @uses wp_insert_post() to add a new item
	 * @return int $result the id of the item created
	 */
	public function save() {
		$this->id               = apply_filters_ref_array( 'buddydrive_item_id_before_save',        array( $this->id,                &$this ) );
		$this->type             = apply_filters_ref_array( 'buddydrive_item_type_before_save',      array( $this->type,              &$this ) );
		$this->user_id          = apply_filters_ref_array( 'buddydrive_item_user_id_before_save',   array( $this->user_id,           &$this ) );
		$this->parent_folder_id = apply_filters_ref_array( 'buddydrive_item_parent_id_before_save', array( $this->parent_folder_id,  &$this ) );
		$this->title            = apply_filters_ref_array( 'buddydrive_item_title_before_save',     array( $this->title,             &$this ) );
		$this->content          = apply_filters_ref_array( 'buddydrive_item_content_before_save',   array( $this->content,           &$this ) );
		$this->mime_type        = apply_filters_ref_array( 'buddydrive_item_mime_type_before_save', array( $this->mime_type,         &$this ) );
		$this->guid             = apply_filters_ref_array( 'buddydrive_item_guid_before_save',      array( $this->guid,              &$this ) );
		$this->metas            = apply_filters_ref_array( 'buddydrive_item_metas_before_save',     array( $this->metas,             &$this ) );


		// Use this, not the filters above
		do_action_ref_array( 'buddydrive_item_before_save', array( &$this ) );

		if ( ! $this->title || ! $this->type ) {
			return false;
		}

		// Defaults to private
		$post_status = 'buddydrive_private';

		if ( ! empty( $this->metas->privacy ) ) {
			$buddydrive_status = get_post_stati( array( 'buddydrive_privacy' => $this->metas->privacy ) );

			if ( is_array( $buddydrive_status ) ) {
				$post_status = reset( $buddydrive_status );
			}
		}

		// If we have an existing ID, update the post, otherwise insert it.
		if ( $this->id ) {
			$wp_update_post_args = array(
				'ID'		     => $this->id,
				'post_author'	 => $this->user_id,
				'post_title'	 => $this->title,
				'post_content'	 => $this->content,
				'post_type'		 => $this->type,
				'post_parent'    => $this->parent_folder_id,
				'post_mime_type' => $this->mime_type,
				'guid'           => $this->guid,
				'post_status'	 => $post_status,
			);

			if ( ! empty( $this->metas->password ) ) {
				$wp_update_post_args['post_password'] = $this->metas->password;
			}

			// If the file or folder is shared within groups
			if ( 'buddydrive_groups' === $post_status ) {
				// First get existing groups
				$existing_groups = (array) get_post_meta( $this->id, '_buddydrive_sharing_groups' );

				if ( ! empty( $this->metas->groups ) ) {
					$remove_group = array_diff( $existing_groups, (array) $this->metas->groups );
					$add_group    = array_diff( (array) $this->metas->groups, $existing_groups );
				}

				// No more groups ?
				if ( ( empty( $existing_groups ) && empty( $add_group ) ) || empty( $this->metas->groups ) ) {
					$wp_update_post_args['post_status'] = 'buddydrive_private';
					delete_post_meta( $this->id, '_buddydrive_sharing_groups' );
				}
			}

			// If the file or folder is shared with specific members
			if ( 'buddydrive_members' === $post_status ) {
				// First get existing groups
				$existing_members = (array) get_post_meta( $this->id, '_buddydrive_sharing_members' );

				if ( ! empty( $this->metas->members ) ) {
					$remove_member = array_diff( $existing_members, (array) $this->metas->members );
					$add_member    = array_diff( (array) $this->metas->members, $existing_members );
				}

				// No more groups ?
				if ( ( empty( $existing_members ) && empty( $add_member ) ) || empty( $this->metas->members ) ) {
					$wp_update_post_args['post_status'] = 'buddydrive_private';
					delete_post_meta( $this->id, '_buddydrive_sharing_members' );
				}
			}

			$result = wp_update_post( $wp_update_post_args );

			if ( $result ) {

				if ( ! empty( $remove_group ) ) {
					foreach ( $remove_group as $r_group_id ) {
						delete_post_meta( $this->id, '_buddydrive_sharing_groups', $r_group_id );
					}
				}

				if ( ! empty( $add_group ) ) {
					foreach ( $add_group as $a_group_id ) {
						add_post_meta( $this->id, '_buddydrive_sharing_groups', $a_group_id );
					}
				}

				if ( ! empty( $remove_member ) ) {
					foreach ( $remove_member as $r_member_id ) {
						delete_post_meta( $this->id, '_buddydrive_sharing_members', $r_member_id );
					}
				}

				if ( ! empty( $add_member ) ) {
					foreach ( $add_member as $a_member_id ) {
						add_post_meta( $this->id, '_buddydrive_sharing_members', $a_member_id );
					}
				}

				if ( ! empty( $this->metas->buddydrive_meta ) && is_array( $this->metas->buddydrive_meta ) ) {
					foreach( $this->metas->buddydrive_meta as $buddydrive_meta ) {
						if( empty( $buddydrive_meta->cvalue ) ) {
							delete_post_meta( $this->id, sanitize_key( $buddydrive_meta->cname ) );
						} else {
							$meta_value = is_array( $buddydrive_meta->cvalue ) ? array_map( 'esc_html', $buddydrive_meta->cvalue ) : wp_kses( $buddydrive_meta->cvalue, array() );
							update_post_meta( $this->id, sanitize_key( $buddydrive_meta->cname ), $meta_value );
						}
					}
				}

				do_action_ref_array( 'buddydrive_item_after_update', array( &$this ) );
			}

		} else {

			$wp_insert_post_args = array(
				'post_author'	 => $this->user_id,
				'post_title'	 => $this->title,
				'post_content'	 => $this->content,
				'post_type'		 => $this->type,
				'post_parent'    => $this->parent_folder_id,
				'post_mime_type' => $this->mime_type,
				'guid'           => $this->guid,
				'post_status'	 => $post_status,
			);

			if ( ! empty( $this->metas->password ) ) {
				$wp_insert_post_args['post_password'] = $this->metas->password;
			}

			$result = wp_insert_post( $wp_insert_post_args );

			if ( $result ) {

				$this->id = $result;

				if ( 'buddydrive_groups' === $post_status && ! empty( $this->metas->groups ) ) {
					foreach ( (array) $this->metas->groups as $group_id ) {
						add_post_meta( $this->id, '_buddydrive_sharing_groups', $group_id );
					}
				}

				if ( 'buddydrive_members' === $post_status && ! empty( $this->metas->members ) ) {
					foreach ( (array) $this->metas->members as $member_id ) {
						add_post_meta( $this->id, '_buddydrive_sharing_members', $member_id );
					}
				}

				if ( ! empty( $this->metas->buddydrive_meta ) && is_array( $this->metas->buddydrive_meta ) ) {

					foreach( $this->metas->buddydrive_meta as $buddydrive_meta ) {
						$meta_value = is_array( $buddydrive_meta->cvalue ) ? array_map( 'esc_html', $buddydrive_meta->cvalue ) : wp_kses( $buddydrive_meta->cvalue, array() );
						update_post_meta( $this->id, sanitize_key( $buddydrive_meta->cname ), $meta_value );
					}
				}

				do_action_ref_array( 'buddydrive_item_after_insert', array( &$this ) );
			}
		}

		do_action_ref_array( 'buddydrive_item_after_save', array( &$this ) );

		return $result;
	}

	/**
	 * The selection query
	 *
	 * @param array $args arguments to customize the query
	 * @uses wp_parse_args() to merge args with defaults one
	 * @uses bp_displayed_user_id() to get the displayed user id
	 * @uses bp_is_my_profile() to check if we're on current user profile
	 * @uses bp_is_active() to check for groups and friends component
	 * @uses friends_check_friendship() to check if current user is friend with item owner
	 * @uses friends_get_friend_user_ids() to get the friends of current user
	 * @uses paginate_links()
	 * @uses add_query_arg()
	 */
	public function get( $args ) {

		// Only run the query once
		if ( empty( $this->query ) ) {
			$defaults = array(
				'id'                => false,
				'name'              => false,
				'group_id'          => false,
				'user_id'           => false,
				'per_page'          => 10,
				'paged'             => 1,
				'type'              => false,
				'buddydrive_scope'  => false,
				'search'            => false,
				'buddydrive_parent' => 0,
				'exclude'           => false,
				'orderby'           => 'title',
				'order'             => 'ASC'
			);

			$r = wp_parse_args( $args, $defaults );
			$paged = ! empty( $_POST['page'] ) ? intval( $_POST['page'] ) : $r['paged'];

			$all_stati = array_keys( buddydrive_get_stati() );

			if ( ! empty( $r['id'] ) ){
				$query_args = array(
					'post_status'    => $all_stati,
					'post_type'      => $r['type'],
					'p'              => $r['id'],
					'posts_per_page' => $r['per_page'],
					'paged'          => $paged,
				);

			} elseif ( ! empty( $r['name'] ) && ! empty( $r['type'] ) ) {

				$query_args = array(
					'post_status'    => $all_stati,
					'post_type'      => $r['type'],
					'name'           => $r['name'],
					'posts_per_page' => $r['per_page'],
					'paged'          => $paged,
				);

			} else {
				// Get all public status
				$public_stati = array_keys( get_post_stati( array( 'public' => true ), 'objects' ) );

				$query_args = array(
					'post_status'    => array_intersect( $all_stati, $public_stati ),
					'post_type'      => $r['type'],
					'post_parent'    => $r['buddydrive_parent'],
					'posts_per_page' => $r['per_page'],
					'paged'          => $paged,
					'orderby'        => $r['orderby'],
					'order'          => $r['order'],
					'meta_query'     => array()
				);

				switch ( $r['buddydrive_scope'] ) {

					case 'files' :
						if ( ! empty( $r['user_id'] ) && (int) $r['user_id'] === (int) bp_displayed_user_id() ) {
							$query_args['author'] = $r['user_id'];
						}

						// Owners or Administrators can view all files
						if ( bp_is_my_profile() || bp_current_user_can( 'bp_moderate' ) ) {
							$query_args['post_status'] = $all_stati;

						// Restrict stati
						} else {
							// Include Password protected for any user
							$query_args['post_status'][] = 'buddydrive_password';

							/**
							 * Files attached to a public group can be viewed by anyone
							 * Other group stati needs the user to be a member of it to
							 * be able to view the file.
							 */
							if ( bp_is_active( 'groups' ) ) {
								// Include groups..
								$query_args['post_status'][] = 'buddydrive_groups';

								$visible_groups = buddydrive_get_visible_groups();
								if ( ! empty( $visible_groups ) ) {
									$query_args['meta_query'] = array(
										'relation' => 'OR',
										array(
											'key'     => '_buddydrive_sharing_groups',
											'value'   => $visible_groups,
											'compare' => 'IN' // Allows $group_id to be an array
										),
									);
								}

								$query_args['meta_query'][] = array(
										'key'     => '_buddydrive_sharing_groups',
										'compare' => 'NOT EXISTS'
								);
							}

							// If the current member is friend with the displayed one.
							if ( bp_is_active( 'friends' ) && friends_check_friendship( $r['user_id'], bp_loggedin_user_id() ) ) {
								$query_args['post_status'][] = 'buddydrive_friends';
							}
						}

						break;

					case 'friends' :
						if ( bp_is_active( 'friends' ) ) {
							$ids = friends_get_friend_user_ids( bp_loggedin_user_id() );

							if ( ! empty( $ids ) ) {
								// Files shared between friends and not only "by"
								$query_args['author__in']  = array_merge( $ids, array( bp_displayed_user_id() ) );
								$query_args['post_status'] = array( 'buddydrive_friends' );
							} else {
								$query_args = array();
							}

						} else {
							$query_args = array();
						}

						break;

					case 'groups' :
						if ( bp_is_active( 'groups' ) && ! empty( $r['group_id'] ) ) {
							if ( bp_is_user() ) {
								$user_groups = groups_get_user_groups( bp_displayed_user_id() );

								if ( ! empty( $user_groups['groups'] ) ) {
									$query_args['meta_query'][] = array(
										'key'     => '_buddydrive_sharing_groups',
										'value'   => $user_groups,
										'compare' => 'IN' // Allows $r['group_id'] to be an array
									);
								} else {
									$query_args = array();
								}
							} elseif ( bp_is_group() ) {
								$group = groups_get_current_group();

								if ( empty( $group->id ) || (int) $group->id !== (int) $r['group_id'] ) {
									$group = groups_get_group( array( 'group_id' => $r['group_id'] ) );
								}

								if ( 'public' !== $group->status && ! groups_is_user_member( bp_loggedin_user_id(), $group->id ) && ! bp_current_user_can( 'bp_moderate' ) ) {
									$query_args = array();
								} else {
									$query_args['post_status'] = array( 'buddydrive_groups' );

									if ( empty( $r['buddydrive_parent'] ) ) {
										$query_args['meta_query'][] = array(
											'key'     => '_buddydrive_sharing_groups',
											'value'   => $group->id,
											'compare' => 'IN' // Allows $r['group_id'] to be an array
										);
									}
								}
							} else {
								$query_args = array();
							}

						} else {
							$query_args = array();
						}

						break;

					case 'members' :
						if ( bp_is_my_profile() || bp_current_user_can( 'bp_moderate' ) ) {
							$query_args['post_status'] = array( 'buddydrive_members' );

							$query_args['meta_query'][] = array(
								'key'     => '_buddydrive_sharing_members',
								'value'   => bp_displayed_user_id(),
								'compare' => 'IN'
							);
						} else {
							$query_args = array();
						}

						break;

					case 'admin' :
						if ( ! bp_current_user_can( 'bp_moderate' ) ) {
							$query_args = array();
						} else {
							// Include all status
							$query_args['post_status'] = $all_stati;

							if ( ! empty( $r['user_id'] ) ) {
								$query_args['author'] = $r['user_id'];
							}

							if ( bp_is_active( 'groups' ) && ! empty( $r['group_id'] ) && empty( $r['buddydrive_parent'] ) ) {
								$query_args['meta_query'][] = array(
									'key'     => '_buddydrive_sharing_groups',
									'value'   => $r['group_id'],
									'compare' => 'IN' // Allows $group_id to be an array
								);
							}
						}

						break;

					default :
						// non public meta values are restricted to admins
						if ( 'public' !== $r['buddydrive_scope'] && ! bp_current_user_can( 'bp_moderate' ) ) {
							$query_args = array();
						}

						/**
						 * @since 2.0.0
						 */
						$query_args = apply_filters( 'buddydrive_item_get_default', $query_args, $r );

						break;
				}

			}

			if ( ! empty( $r['exclude'] ) ) {
				if ( ! is_array( $r['exclude'] ) ) {
					$r['exclude'] = explode( ',', $r['exclude'] );
				}

				$query_args['post__not_in'] = $r['exclude'];
			}

			// Search!
			if ( ! empty( $query_args ) && ! empty( $r['search'] ) ) {
				$query_args['s'] = $r['search'];
			}

			/**
			 * Use the 'buddydrive_item_get' filter to customize the query args
			 *
			 * @since 1.3.2
			 *
			 * @param array $query_args the arguments for the BuddyDrive query
			 * @param array $r          the requested arguments
			 */
			$buddydrive_args = apply_filters( 'buddydrive_item_get', $query_args, $r );

			// Query for BuddyDrive items
			$this->query = new WP_Query( $buddydrive_args );

			if ( ! empty( $buddydrive_args ) ) {
				// Let's also set up some pagination
				$this->pag_links = paginate_links( array(
					'base'      => add_query_arg( 'items_page', '%#%' ),
					'format'    => '',
					'total'     => ceil( (int) $this->query->found_posts / (int) $this->query->query_vars['posts_per_page'] ),
					'current'   => (int) $paged,
					'prev_text' => '&larr;',
					'next_text' => '&rarr;',
					'mid_size'  => 1
				) );
			}
		}
	}

	/**
	 * do we have items to show ?
	 */
	public function have_posts() {
		return $this->query->have_posts();
	}

	/**
	 * Part of our BuddyDrive loop
	 */
	public function the_post() {
		return $this->query->the_post();
	}

	/**
	 * list BuddyDrive Files attached to a folder if any
	 *
	 * @param  int $folder_id the BuddyDrive folder id
	 * @return array the list of file ids
	 */
	public function get_buddydrive_folder_children( $folder_id = false, $only_ids = true ) {
		if ( empty( $folder_id ) ) {
			return false;
		}

		$buddydrive_children = get_children( array(
			'post_type'   => buddydrive_get_file_post_type(),
			'post_parent' => (int) $folder_id,
		) );

		if ( true === $only_ids ) {
			return wp_list_pluck( $buddydrive_children, 'ID' );
		} else {
			foreach ( $buddydrive_children as $key => $child ) {
				$child->user_id   = $child->post_author;
				$child->title     = $child->post_title;
				$child->content   = $child->post_content;

				// Reset the children
				$buddydrive_children[ $key ] = $child;
			}
		}
		return $buddydrive_children;
	}

	/**
	 * Updates the privacy of files attached to a folder
	 *
	 * @since 2.0.0 The $parent_owner_id parameter has been added
	 *
	 * @param int    $folder_id       the BuddyFolder id
	 * @param object $metas           the privacy options of the folder
	 * @param int    $parent_owner_id the User ID of the folder.
	 * @uses BuddyDrive_Items::get_buddydrive_folder_children() to get the files attached to the folder
	 * @uses update_post_meta() to add privacy options
	 */
	public function update_children( $folder_id = false, $metas = false, $parent_owner_id = 0 ) {
		if ( empty( $folder_id ) || empty( $metas->privacy ) ) {
			return false;
		}

		$children = $this->get_buddydrive_folder_children( $folder_id, false );

		if ( ! empty( $children ) ) {
			foreach ( $children as $child ) {
				$args = array();

				/**
				 * if the parent owner is not the child owner, we need to
				 * reset the parent, so that the child owner can still
				 * access to its file no matter the privacy changes.
				 */
				if ( ! empty( $parent_owner_id ) && (int) $parent_owner_id !== (int) $child->user_id ) {
					$args['parent_folder_id'] = 0;
				}

				buddydrive_update_item( $args, $child );
			}
		}
	}

	/**
	 * Retrieves some items datas for an array of BuddyDrive items
	 *
	 * @param  array  $ids the list of ids
	 * @global object $wpdb
	 * @return object the post_title, ID and post_type of the wanted ids
	 */
	public function get_buddydrive_by_ids( $ids = array() ) {
		global $wpdb;

		$buddydrive_items = false;

		if ( ! empty( $ids ) ) {
			$ids = wp_parse_id_list( $ids );

			$in = '("' . implode( '","', $ids ) . '")';
			$buddydrive_items = $wpdb->get_results( "SELECT ID, post_title, post_type FROM {$wpdb->base_prefix}posts WHERE ID IN {$in}");

		}

		return $buddydrive_items;
	}

	/**
	 * Deletes a list of items or all the items of a given user
	 *
	 * @since 2.0.0 Returns an array containing the deleted ids instead of the deleted count
	 *
	 * @param  array $ids array of BuddyDrive Item ids
	 * @param  int $user_id the id of a user
	 * @global object $wpdb
	 *
	 * @return array list of deleted items
	 */
	public function delete( $ids = false, $user_id = false ) {
		global $wpdb;

		$buddydrive_ids = array();
		$spaces         = array();
		$new_space      = false;
		$ids            = array_filter( wp_parse_id_list( $ids ) );

		if ( ! empty( $ids ) ) {
			//we need to get the children
			$in = '("' . implode( '","', $ids ) . '")';
			$buddydrive_ids = $wpdb->get_col( "SELECT ID FROM {$wpdb->base_prefix}posts WHERE post_parent IN {$in}");

			$buddydrive_ids = array_merge( $buddydrive_ids, $ids );

		} elseif ( ! empty( $user_id ) && empty( $ids ) ) {
			// in case a user is deleted
			$buddydrive_ids = $wpdb->get_col( $wpdb->prepare( "SELECT ID FROM {$wpdb->base_prefix}posts WHERE post_author = %d AND post_type IN (%s, %s)", $user_id, buddydrive_get_folder_post_type(), buddydrive_get_file_post_type() ) );

			$new_user = (int) apply_filters( 'buddydrive_set_owner_on_user_deleted', 0 );

			// The new user must have the power to post in any group
			if ( ! empty( $new_user ) && user_can( $new_user, 'bp_moderate' ) && ! empty( $buddydrive_ids ) ) {
				$wpdb->query( $wpdb->prepare( "UPDATE {$wpdb->base_prefix}posts SET post_author = %d WHERE post_author = %d AND post_type IN (%s, %s)", $new_user, $user_id, buddydrive_get_folder_post_type(), buddydrive_get_file_post_type() ) );

				foreach ( $buddydrive_ids as $post_id ) {
					clean_post_cache( $post_id );
				}
			}
		}

		if ( empty( $buddydrive_ids ) ) {
			return false;
		}

		if ( empty( $new_user ) ) {
			foreach ( $buddydrive_ids as $key_id => $id ) {
				$buddyfile = buddydrive_get_buddyfile( $id, array( buddydrive_get_file_post_type(), buddydrive_get_folder_post_type() ) );

				// Delete the file or folder if the current user can
				if ( ! empty( $buddyfile ) ) {

					if ( ! empty( $buddyfile->post_parent ) ) {
						$parent_owner_id = get_post_field( 'post_author', $buddyfile->post_parent );

						/**
						 * The owner of the parent folder is not the same than the file and the current user do not own the file
						 * In this case we simply remove the file from the folder
						 */
						if ( (int) $buddyfile->user_id !== (int) $parent_owner_id && (int) bp_loggedin_user_id() !== (int) $buddyfile->user_id ) {
							buddydrive_update_item( array( 'parent_folder_id' => 0 ), $buddyfile );
							continue;
						}
					}

					if ( buddydrive_current_user_can( 'buddydrive_delete', array( 'owner_id' => $buddyfile->user_id ) ) ) {
						if ( ! empty( $buddyfile->path ) && file_exists( $buddyfile->path ) ) {

							if ( ! isset( $spaces[ $buddyfile->user_id ] ) ) {
								$spaces[ $buddyfile->user_id ] = filesize( $buddyfile->path );
							} else {
								$spaces[ $buddyfile->user_id ] += filesize( $buddyfile->path );
							}

							unlink( $buddyfile->path );
						}

						// Delete the thumbnail
						if ( 'public' === $buddyfile->check_for ) {
							buddydrive_delete_thumbnail( $buddyfile->ID );
						}

						wp_delete_post( $id, true );
					} else {
						unset( $buddydrive_ids[ $key_id ] );
					}

				// Else remove the id from the list
				} else {
					unset( $buddydrive_ids[ $key_id ] );
				}
			}
		}

		if ( ! empty( $spaces ) ) {

			foreach ( $spaces as $u_id => $space ) {
				$user_total_space = get_user_meta( $u_id, '_buddydrive_total_space', true );
				$user_total_space = intval( $user_total_space );

				if ( $space < $user_total_space ) {
					buddydrive_update_user_space( $u_id, -1 * absint( $space ) );
				} else {
					delete_user_meta( $u_id, '_buddydrive_total_space' );
				}
			}
		}

		return $buddydrive_ids;
	}

	/**
	 * Removes a BuddyDrive item from a group
	 *
	 * @since  2.0.0 Add the $group_id parameter
	 *
	 * @param  int     $item_id the BuddyDrive item if
	 * @param  string  $new_status the privacy option to fallback
	 * @return int     The removed item
	 */
	public function remove_from_group( $item_id = false, $new_status = 'private', $group_id = 0 ) {
		$updated = false;

		if ( empty( $item_id ) ) {
			return $updated;
		}

		if ( empty( $group_id ) && bp_is_group() ) {
			$group_id = bp_get_current_group_id();
		}

		$item = buddydrive_get_buddyfile( $item_id, array( buddydrive_get_folder_post_type(), buddydrive_get_file_post_type() ) );

		if ( empty( $item->group ) || 'buddydrive_groups' !== $item->post_status ) {
			return $updated;
		}

		if ( ! buddydrive_current_user_can( 'buddydrive_remove_group', array( 'item' => $item, 'group_id' => $group_id ) ) ) {
			return $updated;
		}

		$groups = array_diff( (array) $item->group, array( $group_id ) );

		if ( ! $groups ) {
			$args = array(
				'privacy' => $new_status,
			);
		} else {
			$args = array(
				'group' => $groups,
			);
		}

		if ( ! empty( $args ) ) {
			// Make sure if a file owned by a member is in a folder owned by another member
			// the parent is reset
			if ( ! empty( $item->post_parent ) ) {
				$parent_owner_id = get_post_field( 'post_author', $item->post_parent );

				if ( ! empty( $parent_owner_id ) && (int) $parent_owner_id !== (int) $item->user_id ) {
					$args['parent_folder_id'] = 0;
				}
			}

			$updated = buddydrive_update_item( $args, $item );
		}

		return $updated;
	}

	/**
	 * Handles the group deletion and restore a privacy to a BuddyDrive item
	 * @param  integer $group_id
	 * @param  string  $new_status
	 * @global $wpdb
	 * @return boolean success or not
	 */
	public function group_remove_items( $group_id = 0, $new_privacy = 'private' ) {
		global $wpdb;

		if ( empty( $group_id ) && bp_is_group() ) {
			$group_id = bp_get_current_group_id();
		}

		$buddydrive_in_group = $wpdb->get_col( $wpdb->prepare( "SELECT post_id FROM {$wpdb->postmeta} WHERE meta_key='_buddydrive_sharing_groups' AND meta_value = %d", $group_id ) );

		if ( empty( $buddydrive_in_group ) ) {
			return false;
		}

		foreach ( $buddydrive_in_group as $item ) {
			$this->remove_from_group( $item, $new_privacy, $group_id );
		}

		return true;
	}

	/**
	 * Count the number of BuddyDrive items for a user
	 *
	 * @since 2.0.0
	 *
	 * @param int          $user_id The User ID.
	 * @param string|array $type    The Single post type or array of post types to count the number of items for.
	 * @param string|array $status  The Single post status or array of post stati to count the number of items for.
	 */
	public static function count_user_items( $user_id = 0, $type = 'any', $status = 'any' ) {
		global $wpdb;

		if ( empty( $user_id ) ) {
			return false;
		}

		$item_type = array(  buddydrive_get_file_post_type(), buddydrive_get_folder_post_type() );
		if ( 'any' !== $type && ! empty( $type ) ) {
			$item_type = (array) $type;
		}

		$buddydrive_status = buddydrive_get_stati();
		$item_status = array_keys( $buddydrive_status );
		if ( 'any' !== $status && ! empty( $status ) && isset( $buddydrive_status[ $status ] ) ) {
			$item_status = (array) $status;
		}

		$sql = array(
			'select' => "SELECT COUNT(*) FROM {$wpdb->posts}",
			'where' => array(
				'user'   => $wpdb->prepare( "post_author = %d", $user_id ),
				'type'   => sprintf( "post_type IN ( '%s' )", join( "', '", esc_sql( $item_type ) ) ),
			),
		);

		if ( (int) $user_id !== (int) bp_loggedin_user_id() ) {
			$item_status = array_intersect( $item_status, array( 'buddydrive_public', 'buddydrive_password' ) );
		}

		$sql['where']['status'] = sprintf( "post_status IN ( '%s' )", join( "', '", esc_sql( $item_status ) ) );
		$sql['where'] = 'WHERE ' . join( ' AND ', $sql['where'] );

		$count = $wpdb->get_var( join( ' ', $sql ) );

		/**
		 * Filter the number of posts a user has written.
		 *
		 * @since 2.0.0
		 *
		 * @param int          $count   The user's post count.
		 * @param int          $user_id The User ID.
		 * @param string|array $type    The Single post type or array of post types to count the number of items for.
		 * @param string|array $status  The Single post status or array of post stati to count the number of items for.
		 */
		return apply_filters( 'buddydrive_count_user_items', $count, $user_id, $type, $status );
	}

	/**
	 * Get User stats
	 *
	 * @since 2.0.0
	 *
	 * @param int $user_id The User ID.
	 */
	public static function get_user_stats( $user_id = 0 ) {
		global $wpdb;

		if ( empty( $user_id ) ) {
			return false;
		}

		if ( ! bp_is_my_profile() && ! bp_current_user_can( 'bp_moderate' ) ) {
			return false;
		}

		$sql = array(
			'select' => "SELECT post_status, COUNT(ID) as num FROM {$wpdb->posts}",
			'where' => array(
				'user'   => $wpdb->prepare( "post_author = %d", $user_id ),
				'type'   => $wpdb->prepare( "post_type = %s", buddydrive_get_file_post_type() ),
			),
			'groupby' => 'GROUP BY post_status',
		);

		$sql['where'] = 'WHERE ' . join( ' AND ', $sql['where'] );

		$stats = $wpdb->get_results( join( ' ', $sql ) );

		/**
		 * Filter the stats of a user.
		 *
		 * @since 2.0.0
		 *
		 * @param int          $count   The user's post count.
		 * @param int          $user_id The User ID.
		 */
		return apply_filters( 'buddydrive_get_user_stats', $stats, $user_id );
	}
}

if ( class_exists( 'BP_Attachment' ) ) :
/**
 * The Attachments class
 *
 * @since 1.3.0
 */
class BuddyDrive_Attachment extends BP_Attachment {
	/**
	 * The constuctor
	 *
	 * @since 1.3.0
	 */
	public function __construct() {
		parent::__construct( array(
			'action'               => 'buddydrive_upload',
			'file_input'           => 'buddyfile-upload',
			'base_dir'             => 'buddydrive',
			'upload_error_strings' => buddydrive_get_upload_error_strings(),
			'allowed_mime_types'   => buddydrive_get_allowed_upload_exts(),
		) );
	}

	/**
	 * Get BuddyDrive upload data
	 *
	 * @since 1.3.0
	 *
	 * @return array an associative array to inform about the upload base dir and url
	 */
	public function get_upload_data() {
		// Make sure to run this part once
		if ( empty( $this->upload_data['dir'] ) ) {
			/**
			 * In previous version it was possible to change
			 * the upload data before upload dir has been created
			 * using this filter. Keeping it for backcompat
			 *
			 * @since  1.2.0
			 *
			 * @param array an associative array to inform about the upload base dir and url
			 */
			$this->upload_data = apply_filters( 'buddydrive_get_upload_data', array(
				'dir'      => $this->upload_path,
				'url'      => $this->url,
				'thumbdir' => $this->upload_path . '-thumbnails',
				'thumburl' => $this->url . '-thumbnails',
			) );

			if ( $this->upload_data['dir'] != $this->upload_path ) {
				$this->upload_path = $this->upload_data['dir'];
			}

			if ( $this->upload_data['url'] != $this->url ) {
				$this->url = $this->upload_data['url'];
			}
		}

		return $this->upload_data;
	}

	/**
	 * Create the BuddyDrive dir and add an .htaccess in it
	 *
	 * @since 1.3.0
	 *
	 * @return bool true on dir created, false otherwise
	 */
	public function create_dir() {
		// Let's be sure old filter is fired
		$this->get_upload_data();

		// Create a public folder for thumbnails
		if ( ! empty( $this->upload_data['thumbdir'] ) && ! is_dir( $this->upload_data['thumbdir'] ) ) {
			wp_mkdir_p( $this->upload_data['thumbdir'] );
		}

		// Check if upload path already exists
		if ( ! is_dir( $this->upload_path ) ) {

			// If path does not exist, attempt to create it
			if ( ! wp_mkdir_p( $this->upload_path ) ) {
				return false;
			}

			// then we need to check for .htaccess and eventually create it
			if ( ! file_exists( $this->upload_path .'/.htaccess' ) ) {
				$this->required_wp_files['misc'] = 'misc';
				$this->includes();

				// Defining the rule, we need to make it unreachable and use php to reach it
				$rules = array( 'Order Allow,Deny','Deny from all' );

				// creating the .htaccess file
				insert_with_markers( $this->upload_path .'/.htaccess', 'Buddydrive', $rules );
				unset( $this->required_wp_files['misc'] );
			}
		}

		// Directory exists
		return true;
	}

	/**
	 * BuddyDrive specific upload rules
	 *
	 * @since 1.3.0
	 *
	 * @param  array $file the temporary file attributes (before it has been moved)
	 * @return array the file
	 */
	public function validate_upload( $file = array() ) {
		// there's already an error
		if ( ! empty( $file['error'] ) ) {
			return $file;
		}

		// This codes are restricted to BuddyDrive
		$buddydrive_errors = array(
			9  => 1,
			10 => 1,
			11 => 1,
		);

		// what's left in user's quota ?
		$space_left = buddydrive_get_user_space_left( 'diff' );
		$file_size = filesize( $file['tmp_name'] );

		// File is bigger than space left
		if ( $space_left < $file_size ) {
			$file['error'] = 9;
		}

		// File is bigger than the max allowed for BuddyDrive files
		if ( $file_size > buddydrive_max_upload_size( true ) ) {
			$file['error'] = 10;
		}

		// No more space left
		if ( $space_left <= 0 ) {
			$file['error'] = 11;
		}

		if ( ! isset( $buddydrive_errors[ $file['error'] ] ) ) {
			/**
			 * Validation for custom errors
			 *
			 * @since 1.2.2
			 *
			 * @param $file the file data
			 */
			return apply_filters( 'buddydrive_upload_errors', $file );
		}

		return $file;
	}

	/**
	 * Set the directory when uploading a file
	 *
	 * @since 1.3.0
	 * @since 1.3.3 Add the $upload_dir parameter
	 *
	 * @return array upload data (path, url, basedir...)
	 */
	public function upload_dir_filter( $upload_dir = array() ) {
		$upload_data = parent::upload_dir_filter( $upload_dir );
		/**
		 * Filters BuddyDrive's upload data.
		 *
		 * If people used to filter 'buddydrive_upload_datas', we need
		 * to have it here
		 *
		 * @since 1.0
		 *
		 * @param array $value Array containing the path, URL, and other helpful settings.
		 */
		return apply_filters( 'buddydrive_upload_datas', $upload_data );
	}

	/**
	 * Build script datas for the Uploader UI
	 *
	 * @since 1.3.0
	 *
	 * @return array the javascript localization data
	 */
	public function script_data() {
		// Get default script data
		$script_data = parent::script_data();

		$script_data['bp_params'] = array(
			'object'  => buddydrive_get_file_post_type(),
			'item_id' => bp_loggedin_user_id(),
			'privacy' => buddydrive_get_default_privacy(),
		);

		if ( buddydrive_current_group_is_enabled() ) {
			$script_data['bp_params']['privacy']         = 'groups';
			$script_data['bp_params']['privacy_item_id'] = bp_get_current_group_id();
		}

		// Include our specific css
		$script_data['extra_css'] = array( 'buddydrive-app-style' );

		// css front-end complementary styles
		if ( wp_style_is( 'buddydrive-front-end-style', 'registered' ) ) {
			$script_data['extra_css'] = array( 'buddydrive-front-end-style' );
		}

		// Include our specific js
		$script_data['extra_js']  = array( 'buddydrive-app-js' );

		return apply_filters( 'buddydrive_attachment_script_data', $script_data );
	}
}
endif;