<?php
/**
 * Attachment Data Extra fields
 *
 * @package Attachment Fields
 */

if ( ! class_exists( 'UABB_Attachment' ) ) {
	/**
	 * This class initializes UABB Attachment
	 *
	 * @class UABB_Attachment
	 */
	class UABB_Attachment {

		/**
		 * Constructor function that initializes required actions and hooks
		 *
		 * @since 1.0
		 */
		public function __construct() {

			add_filter( 'attachment_fields_to_edit', array( $this, 'uabb_attachment_field_cta' ), 10, 2 );
			add_filter( 'attachment_fields_to_save', array( $this, 'uabb_attachment_field_cta_save' ), 10, 2 );
		}

		/**
		 * Add CTA Link field to media uploader
		 *
		 * @param array  $form_fields array, fields to include in attachment form.
		 * @param object $post object, attachment record in database.
		 * @return $form_fields, modified form fields
		 */
		public function uabb_attachment_field_cta( $form_fields, $post ) {
			$form_fields['uabb-cta-link']   = array(
				'label' => __( 'Image Link', 'uabb' ),
				'input' => 'text',
				'value' => get_post_meta( $post->ID, 'uabb-cta-link', true ),
			);
			$form_fields['uabb-categories'] = array(
				'label' => __( ' UABB - Categories  (Ex: Cat1, Cat2) ', 'uabb' ),
				'input' => 'text',
				'value' => get_post_meta( $post->ID, 'uabb-categories', true ),
			);

			return $form_fields;
		}


		/**
		 * Save values of CTA Link field in media uploader
		 *
		 * @param array $post array, the post data for database.
		 * @param array $attachment array, attachment fields from $_POST form.
		 * @return array $post array, modified post data
		 */
		public function uabb_attachment_field_cta_save( $post, $attachment ) {
			if ( isset( $attachment['uabb-cta-link'] ) ) {
				update_post_meta( $post['ID'], 'uabb-cta-link', $attachment['uabb-cta-link'] );
			}
			if ( isset( $attachment['uabb-categories'] ) ) {
				update_post_meta( $post['ID'], 'uabb-categories', $attachment['uabb-categories'] );
			}

			return $post;
		}


	}
	new UABB_Attachment();
}