<?php

/**
 * Class MC4WP_Form_Output_Manager
 *
 * @ignore
 * @access private
 */
class MC4WP_Form_Output_Manager
{
    /**
     * @var int The # of forms outputted
     */
    public $count = 0;

    /**
     * @const string
     */
    const SHORTCODE = 'mc4wp_form';

    /**
     * Add hooks
     */
    public function add_hooks()
    {
        // enable shortcodes in form content
        add_filter('mc4wp_form_content', 'do_shortcode');
        add_action('init', array( $this, 'register_shortcode' ));
    }

    /**
     * Registers the [mc4wp_form] shortcode
     */
    public function register_shortcode()
    {
        add_shortcode(self::SHORTCODE, array( $this, 'shortcode' ));
    }

    /**
     * @param array $attributes
     * @param string $content
     * @return string
     */
    public function shortcode($attributes = array(), $content = '')
    {
        $default_attributes = array(
            'id'            => '',
            'lists'         => '',
            'email_type'    => '',
            'element_id'    => '',
            'element_class' => '',
        );

        $attributes = shortcode_atts(
            $default_attributes,
            $attributes,
            self::SHORTCODE
        );

        $config = array(
            'element_id'    => $attributes['element_id'],
            'lists'         => $attributes['lists'],
            'email_type'    => $attributes['email_type'],
            'element_class' => $attributes['element_class'],
        );

        return $this->output_form($attributes['id'], $config, false);
    }

    /**
     * @param int   $id
     * @param array $config
     * @param bool $echo
     *
     * @return string
     */
    public function output_form($id = 0, $config = array(), $echo = true)
    {
        try {
            $form = mc4wp_get_form($id);
        } catch (Exception $e) {
            if (current_user_can('manage_options')) {
                return sprintf('<strong>Mailchimp for WordPress error:</strong> %s', $e->getMessage());
            }

            return '';
        }

        ++$this->count;

        // set a default element_id if none is given
        if (empty($config['element_id'])) {
            $config['element_id'] = 'mc4wp-form-' . $this->count;
        }

        $form_html = $form->get_html($config['element_id'], $config);

        try {
            // start new output buffer
            ob_start();

            /**
             * Runs just before a form element is outputted.
             *
             * @since 3.0
             *
             * @param MC4WP_Form $form
             */
            do_action('mc4wp_output_form', $form);

            // output the form (in output buffer)
            echo $form_html;

            // grab all contents in current output buffer & then clean + end it.
            $html = ob_get_clean();
        } catch (Error $e) {
            $html = $form_html;
        }

        // echo content if necessary
        if ($echo) {
            echo $html;
        }

        return $html;
    }
}