<?php

namespace Megumi\WP\Post;

/*
	License:
		Released under the GPL license
		http://www.gnu.org/copyleft/gpl.html

	Copyright 2013 ( email : wokamoto1973@gmail.com )

	This program is free software; you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation; either version 2 of the License, or
	( at your option ) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program; if not, write to the Free Software
	Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

if ( defined( 'ABSPATH' ) ) :

	require_once( ABSPATH . 'wp-admin/includes/image.php' );
	require_once( ABSPATH . 'wp-admin/includes/post.php' );

	class Helper {

		private $post;

		private $postid = false;

		private $attachment_id = array();
		private $tags   = array();
		private $medias = array();
		private $metas  = array();
		private $fields = array();
		private $media_count = 0;
		private $terms  = array();

		function __construct( $args = array() )
		{
			if ( is_object( $args ) ) {
				$args = (array) $args;
			}

			$this->post = get_default_post_to_edit();
			$this->post->post_category = null;
			if ( is_array( $args ) && count( $args ) > 0 ) {
				$this->set( $args );
			}
		}

		// Get PostID
		public function get_post_id()
		{
			return $this->postid;
		}

		// Get Attachment ID
		public function get_attachment_id()
		{
			return $this->attachment_id;
		}

		// Set Post Data
		private function set( $args )
		{
			if ( is_object( $args ) ) {
				$args = (array) $args;
			}

			if ( ! is_array( $args ) ) {
				return false;
			}

			if ( isset( $args['post_date'] ) && $args['post_date'] ) {
				$args['post_date'] = date( "Y-m-d H:i:s", strtotime( $args['post_date'] ) );
			}

			// setup author
			if ( isset( $args['post_author'] ) && $args['post_author'] && ! intval( $args['post_author'] ) ) {
				$u = get_user_by( 'login', $args['post_author'] );
				if ( $u ) {
					$args['post_author'] = $u->ID;
				} else {
					unset( $args['post_author'] );
				}
			}

			$post = $this->post;
			foreach ( $post as $key => &$val ) {
				if ( 'ID' !== $key && isset( $args[ $key ] ) ) {
					$val = $args[ $key ];
				}
			}
			$this->post = $post;

			if ( isset( $args['ID'] ) && intval( $args['ID'] ) ) {
				$this->post->ID = $args['ID'];
			}

			if ( isset( $args['post_tags'] ) ) {
				$this->add_tags(
					is_array( $args['post_tags'] )
					? $args['post_tags']
					: explode( ',', $args['post_tags'] )
				);
			}

			return true;
		}

		// Add Post
		public function insert()
		{
			if ( ! isset( $this->post ) ) {
				return false;
			}

			$postid = wp_insert_post( $this->post, true );
			if ( $postid && ! is_wp_error( $postid ) ) {
				$this->postid   = $postid;
				$this->post->ID = $postid;
				return $this->add_related_meta( $postid ) ? $postid : false;
			} else {
				$this->postid   = $postid;
				$this->post->ID = 0;
				return $postid;
			}
		}

		// Update Post
		// public function update()
		// {
		// 	if ( ! isset( $this->post ) ) {
		// 		return false;
		// 	}
		//
		// 	$postid =
		// 		$this->postid
		// 		? wp_update_post( $this->post )
		// 		: wp_insert_post( $this->post );
		// 	if ( $postid && ! is_wp_error( $postid ) ) {
		// 		$this->postid   = $postid;
		// 		$this->post->ID = $postid;
		// 		return $this->add_related_meta( $postid ) ? $postid : false;
		// 	} else {
		// 		$this->postid   = false;
		// 		$this->post->ID = 0;
		// 		return false;
		// 	}
		// }

		private function add_related_meta( $postid )
		{
			if ( ! $postid || is_wp_error( $postid ) ) {
				return false;
			}

			$this->postid   = $postid;

			// add Tags
			if ( count( $this->tags ) > 0 ) {
				$this->add_tags( $this->tags );
			}

			$this->tags = array();

			// add medias
			foreach ( $this->medias as $key => $val ) {
				$this->add_media( $key, $val[0], $val[1], $val[2], $val[3] );
			}
			$this->medias = array();

			// add terms
			foreach ( $this->terms as $taxonomy => $terms ) {
				$this->add_terms( $taxonomy, $terms );
			}
			$this->terms = array();

			// add Custom Fields
			foreach ( $this->metas as $key => $val ) {
				if ( is_array( $val ) ) {
					$this->add_meta( $key, $val[0], isset( $val[1] ) ? $val[1] : true );
				} else {
					$this->add_meta( $key, $val );
				}
			}
			$this->metas = array();

			// add ACF Fields
			foreach ( $this->fields as $key => $val ) {
				$this->add_field( $key, $val );
			}
			$this->fields = array();

			return true;
		}

		// Add Tag
		public function add_tags( $tags = array() )
		{
			$tags = is_array( $tags ) ? $tags : explode( ',', $tags );
			foreach ( $tags as $tag ) {
				if ( ! empty( $tag ) && ! array_search( $tag, $this->tags ) ) {
					$this->tags[] = $tag;
				}
			}
			unset( $tags );

			if ( $this->postid ) {
				$tags = implode( ',', $this->tags );
				$this->tags = array();
				return wp_add_post_tags( $this->postid, $tags );
			}
		}

		// add terms
		public function add_terms( $taxonomy, $terms )
		{
			if ( ! $this->postid ) {
				if ( ! isset( $this->terms[ $taxonomy ] ) ) {
					$this->terms[ $taxonomy ] = array();
				}
				foreach ( (array) $terms as $term ) {
					if ( false === array_search( $term, $this->terms[ $taxonomy ] ) ) {
						$this->terms[ $taxonomy ][] = $term;
					}
				}
			} else {
				return wp_set_object_terms( $this->postid, $terms, $taxonomy );
			}
		}

		// Add Media
		public function add_media( $filename, $title = null, $content = null, $excerpt = null, $thumbnail = false )
		{
			if ( ! $this->postid ) {
				$this->medias[ $filename ] = array(
					$title,
					$content,
					$excerpt,
					$thumbnail,
				);
				return;
			}

			if ( filter_var( $filename, FILTER_VALIDATE_URL ) ) {
				$filename = $this->remote_get_file( $filename );
			}

			if (  $filename && file_exists( $filename )  ) {
				$mime_type = '';
				$wp_filetype = wp_check_filetype( basename( $filename ), null );
				if ( isset( $wp_filetype['type'] ) && $wp_filetype['type'] ) {
					$mime_type = $wp_filetype['type'];
				}
				unset( $wp_filetype );

				$title = isset( $title ) ? $title : preg_replace( '/\.[^.]+$/', '', basename( $filename ) );
				$content = isset( $content ) ? $content : $title;
				$excerpt = isset( $excerpt ) ? $excerpt : $content;
				$attachment = array(
					'post_mime_type' => $mime_type,
					'post_parent'    => $this->postid,
					'post_author'    => $this->post->post_author,
					'post_title'     => $title,
					'post_content'   => $content,
					'post_excerpt'   => $excerpt,
					'post_status'    => 'inherit',
					'menu_order'     => $this->media_count + 1,
				 );

				if ( isset( $this->post->post_name ) && $this->post->post_name ) {
					$attachment['post_name'] = $this->post->post_name;
				}

				$attachment_id = wp_insert_attachment( $attachment, $filename, $this->postid );
				unset( $attachment );

				if ( ! is_wp_error( $attachment_id ) ) {
					$this->media_count++;
					$this->attachment_id[] = $attachment_id;
					$attachment_data = wp_generate_attachment_metadata( $attachment_id, $filename );
					wp_update_attachment_metadata( $attachment_id,  $attachment_data );
					unset( $attachment_data );
					if ( $thumbnail ) {
						set_post_thumbnail( $this->postid, $attachment_id );
					}
					return $attachment_id;
				} else {
					return false;
				}
			} else {
				return false;
			}
		}

		// Add Custom Field
		public function add_meta( $metakey, $val, $unique = true )
		{
			if ( ! $this->postid ) {
				$this->metas[ $metakey ] = array( $val, $unique );
			} else {
				return $val ? add_post_meta( $this->postid, $metakey, $val, $unique ) : false;
			}
		}

		// Add Advanced Custom Field
		public function add_field( $field_key, $val )
		{
			if ( ! $this->postid ) {
				$this->fields[ $field_key ] = $val;
			} else {
				return $val ? update_field( $field_key, $val, $this->postid ) : false;
			}
		}

		public function get_post()
		{
			return $this->post;
		}

		private function remote_get_file( $url = null, $file_dir = '' )
		{
			if ( ! $url ) {
				return false;
			}

			if ( empty( $file_dir ) ) {
				 $upload_dir = wp_upload_dir();
				 $file_dir = isset( $upload_dir['path'] ) ? $upload_dir['path'] : '';
			}
			$file_dir = trailingslashit( $file_dir );

			// make directory
			if ( ! file_exists( $file_dir ) ) {
				$dirs = explode( '/', $file_dir );
				$subdir = '/';
				foreach ( $dirs as $dir ) {
					if ( ! empty( $dir ) ) {
						$subdir .= $dir . '/';
						if ( ! file_exists( $subdir ) ) {
							wp_mkdir_p( $subdir );
						}
					}
				}
			}

			// remote get!
			$photo = $file_dir . basename( $url );
			if ( ! file_exists( $photo ) ) {
				$response = wp_remote_get( esc_url_raw( $url ) );
				if ( ! is_wp_error( $response ) && $response['response']['code'] === 200 ) {
					$photo_data = $response['body'];
					file_put_contents( $photo, $photo_data );
					unset( $photo_data );
				} else {
					$photo = false;
				}
				unset( $response );
			}
			return file_exists( $photo ) ? $photo : false;
		}
	}

endif;