<?php
defined('ABSPATH') or die();

/* Posts */

/*
 * Update 1.0.3
 * 
 * Since 1.0.0
 * 
 * Display column value;
 */
function ad_column_custom_post_column( $column, $post_id ) 
{
	$p = get_post( $post_id );

	if( is_object($p) && isset($p->ID) ) 
	{
		$data = ad_column_custom_get_column( $p->post_type, $column );

		$name = isset($data['name']) ? $data['name'] : '';
		$type = isset($data['type']) ? $data['type'] : '';

		if( $name == '' ) return;
		
		if( $type == 'post_field' )
		{
			if( substr($name, 0, 5 ) != 'post_' ) {
				$name = 'post_' . $name;
			}

			echo esc_attr( isset($p->$name) ? $p->$name : '' );
		}
		else if( $type == 'post_meta' )
		{
			$value = get_post_meta( $post_id, $column );

			if( is_object($value) ) {
				$value = array_values( (array) $value );
			}
			
			if( is_array($value) ) {
				$value = implode(', ', $value);
			}

			echo esc_attr( $value );
		}
	}
}

/*
 * Since 1.0.0
 * 
 * Display column title;
 */
function ad_column_custom_add_post_column( $columns = array(), $type = 'page' )
{
	$data = get_option( 'ad_column_' . $type );

	if( is_array($data) == false || count($data) == 0 || count($columns) == 0 ) {
		return $columns;
	}

	$list = array();

	foreach( $data as $i => $item ) {
		if( $item['after'] == 'first' ) {
			$list[ $item['name'] ] = esc_attr( __( $item['title'] ) );
			unset($data[$i]);
		}
	}

	foreach ( $columns as $key => $value ) {
		$list[ $key ] = $value;
	
		foreach( $data as $i => $item ) {
			if( $item['after'] == $key ) {
				$list[ $item['name'] ] = esc_attr( __( $item['title'] ) );
				unset($data[$i]);
			}
		}
	}
	
	return $list;
}

/*
 * Update 1.0.2
 * 
 * Since 1.0.1
 */
function ad_column_custom_list_page()
{
	/* posts */
	foreach( ad_column_custom_get_post_types() as $post_type => $title ) {
		add_filter( 'manage_'. $post_type .'s_columns' , 'ad_column_custom_add_post_column', 10, 2);
		add_action( 'manage_'. $post_type .'s_custom_column' , 'ad_column_custom_post_column', 10, 2 );
	}

	/* user */
	add_filter('manage_users_columns', 'ad_column_custom_add_user_column');
	add_filter('manage_users_custom_column', 'ad_column_custom_user_column', 10, 3);
}
add_action( 'admin_init' , 'ad_column_custom_list_page');

/* Users */

/*
 * Since 1.0.0
 */
function ad_column_custom_add_user_column($columns) 
{
	return ad_column_custom_add_post_column( $columns, 'user' );
}

/*
 * Since 1.0.0
 */
function ad_column_custom_user_column($value, $column, $user_id) 
{
	if( $user_id>0 ) {
		$data = ad_column_custom_get_column( 'user', $column );
		if( isset($data['name']) ) {
			$value = esc_attr( get_user_meta( $user_id, $column, true ) );
		}
	}
	
	return $value;
}

/*
 * Since 1.0.1
 */
function ad_column_custom_manage_columns( $columns = array() )
{
	// $columns['url'] = 'URL';

	return $columns;
}
add_filter( 'manage_edit-page_columns', 'ad_column_custom_manage_columns' );

/**
 * Can't hide these for they are special.
 * 
 * wp-admin/includes/class-wp-screen.php
 * 
 * Line 1164 in function `render_list_table_columns_preferences`;
 * 
 */
function ad_column_custom_get_special_columns()
{
	$special = array( '_title', 'cb', 'comment', 'media', 'name', 'title', 'username', 'blogname' );

	return $special;
}

/*
 * Update 1.0.2
 * 
 * Since 1.0.1
 */
function ad_column_custom_get_default_columns( $type = '' )
{
	$columns = array();

	if( $type == 'user' ) 
	{
		$names = array( 'username', 'name', 'email', 'role', 'posts' );
	}
	/* posts */
	else {
		$names = array( 'title', 'author' );

		if( $type == 'post' ) {
			$names = array_merge( $names, array( 'categories', 'tags' ) );
		}

		$names = array_merge( $names, array( 'comments', 'date', 'modified' ) );
	}

	foreach( $names as $name ) {
		$columns[] = array(
			'title' => __( ucwords($name) ),
			'name' => $name,
			'default' => 1
		);
	}

	return $columns;
}

/*
 * Since 1.0.1
 */
function ad_column_custom_get_columns( $type = '', $columns = array() )
{
	if( count($columns) == 0 ) {
		$columns = ad_column_custom_get_default_columns( $type );
	}

	$data = get_option('ad_column_' . $type);

	if( is_array($data) && count($data) ) {
		$list = array();

		foreach( $data as $i => $item ) {
			if( $item['after'] == 'first' ) {
				$item['default'] = 0;
				$list[] = $item;
				unset($data[$i]);
			}
		}

		foreach( $columns as $column ) {
			$list[] = $column;

			foreach( $data as $i => $item ) {
				if( $item['after'] == $column['name'] ) {
					$item['default'] = 0;
					$list[] = $item;
					unset($data[$i]);
				}
			}
		}

		$columns = $list;
	}

	return $columns;
}

/*
 * Since 1.0.1
 */
function ad_column_custom_get_column( $type = '', $name = '', $key = '' )
{
	global $ad_column_data;

	if( empty($ad_column_data[ $type ]) ) {
		$ad_column_data[ $type ] = get_option('ad_column_' . $type);
	}

	$data = $ad_column_data[ $type ];

	if( is_array($data) && count($data)>0 ) {
		foreach( $data as $item ) {
			if( $item['name'] == $name ) {

				if( $key!='' && isset($item[$key]) ) {
					return $item[$key];
				}

				return $item;
			}
		}

		return $data;
	}

	if( $key!='' ) {
		return '';
	}

	return array();
}

/*
 * Since 1.0.2
 */
function ad_column_custom_get_column_types( $type = '' )
{
	if( $type == 'user' ) 
	{
		$column_types = array(
			'user_meta' 	=> __( 'User Meta', 'admin-column-custom' ),
		);
	}
	/* posts */
	else {
		$column_types = array(
			'post_field' 	=> __( 'Post Field', 'admin-column-custom' ),
			'post_meta' 	=> __( 'Post Meta', 'admin-column-custom' ),
		);
	}

	return $column_types;
}