<?php

/**
 * Sets, gets, and removes theme mods.
 *
 * ## EXAMPLES
 *
 *     # Set the 'background_color' theme mod to '000000'.
 *     $ wp theme mod set background_color 000000
 *     Success: Theme mod background_color set to 000000.
 *
 *     # Get single theme mod in JSON format.
 *     $ wp theme mod get background_color --format=json
 *     [{"key":"background_color","value":"dd3333"}]
 *
 *     # Remove all theme mods.
 *     $ wp theme mod remove --all
 *     Success: Theme mods removed.
 */
class Theme_Mod_Command extends WP_CLI_Command {

    private $fields = [ 'key', 'value' ];

    /**
     * Gets one or more theme mods.
     *
     * ## OPTIONS
     *
     * [<mod>...]
     * : One or more mods to get.
     *
     * [--field=<field>]
     * : Returns the value of a single field.
     *
     * [--all]
     * : List all theme mods
     *
     * [--format=<format>]
     * : Render output in a particular format.
     * ---
     * default: table
     * options:
     *   - table
     *   - json
     *   - csv
     *   - yaml
     * ---
     *
     * ## EXAMPLES
     *
     *     # Get all theme mods.
     *     $ wp theme mod get --all
     *     +------------------+---------+
     *     | key              | value   |
     *     +------------------+---------+
     *     | background_color | dd3333  |
     *     | link_color       | #dd9933 |
     *     | main_text_color  | #8224e3 |
     *     +------------------+---------+
     *
     *     # Get single theme mod in JSON format.
     *     $ wp theme mod get background_color --format=json
     *     [{"key":"background_color","value":"dd3333"}]
     *
     *     # Get value of a single theme mod.
     *     $ wp theme mod get background_color --field=value
     *     dd3333
     *
     *     # Get multiple theme mods.
     *     $ wp theme mod get background_color header_textcolor
     *     +------------------+--------+
     *     | key              | value  |
     *     +------------------+--------+
     *     | background_color | dd3333 |
     *     | header_textcolor |        |
     *     +------------------+--------+
     */
    public function get( $args = array(), $assoc_args = array() ) {

        if ( ! \WP_CLI\Utils\get_flag_value( $assoc_args, 'all' ) && empty( $args ) ) {
            WP_CLI::error( 'You must specify at least one mod or use --all.' );
        }

        if ( \WP_CLI\Utils\get_flag_value( $assoc_args, 'all' ) ) {
            $args = array();
        }

        $list = array();
        $mods = get_theme_mods();
        if ( ! is_array( $mods ) ) {
            // If no mods are set (perhaps new theme), make sure foreach still works.
            $mods = array();
        }
        foreach ( $mods as $k => $v ) {
            // If mods were given, skip the others.
            if ( ! empty( $args ) && ! in_array( $k, $args, true ) ) {
                continue;
            }

            if ( is_array( $v ) ) {
                $list[] = [
                    'key'   => $k,
                    'value' => '=>',
                ];
                foreach ( $v as $_k => $_v ) {
                    $list[] = [
                        'key'   => "    $_k",
                        'value' => $_v,
                    ];
                }
            } else {
                $list[] = [
                    'key'   => $k,
                    'value' => $v,
                ];
            }
        }

        // For unset mods, show blank value.
        foreach ( $args as $mod ) {
            if ( ! isset( $mods[ $mod ] ) ) {
                $list[] = [
                    'key'   => $mod,
                    'value' => '',
                ];
            }
        }

        $formatter = new \WP_CLI\Formatter( $assoc_args, $this->fields, 'thememods' );
        $formatter->display_items( $list );
    }

    /**
     * Gets a list of theme mods.
     *
     * ## OPTIONS
     *
     * [--field=<field>]
     * : Returns the value of a single field.
     *
     * [--format=<format>]
     * : Render output in a particular format.
     * ---
     * default: table
     * options:
     *   - table
     *   - json
     *   - csv
     *   - yaml
     * ---
     *
     * ## EXAMPLES
     *
     *     # Gets a list of theme mods.
     *     $ wp theme mod list
     *     +------------------+---------+
     *     | key              | value   |
     *     +------------------+---------+
     *     | background_color | dd3333  |
     *     | link_color       | #dd9933 |
     *     | main_text_color  | #8224e3 |
     *     +------------------+---------+
     *
     * @subcommand list
     */
    public function list_( $args = array(), $assoc_args = array() ) {

        $assoc_args['all'] = 1;

        $this->get( $args, $assoc_args );
    }

    /**
     * Removes one or more theme mods.
     *
     * ## OPTIONS
     *
     * [<mod>...]
     * : One or more mods to remove.
     *
     * [--all]
     * : Remove all theme mods.
     *
     * ## EXAMPLES
     *
     *     # Remove all theme mods.
     *     $ wp theme mod remove --all
     *     Success: Theme mods removed.
     *
     *     # Remove single theme mod.
     *     $ wp theme mod remove background_color
     *     Success: 1 mod removed.
     *
     *     # Remove multiple theme mods.
     *     $ wp theme mod remove background_color header_textcolor
     *     Success: 2 mods removed.
     */
    public function remove( $args = array(), $assoc_args = array() ) {

        if ( ! \WP_CLI\Utils\get_flag_value( $assoc_args, 'all' ) && empty( $args ) ) {
            WP_CLI::error( 'You must specify at least one mod or use --all.' );
        }

        if ( \WP_CLI\Utils\get_flag_value( $assoc_args, 'all' ) ) {
            remove_theme_mods();
            WP_CLI::success( 'Theme mods removed.' );
            return;
        }

        foreach ( $args as $mod ) {
            remove_theme_mod( $mod );
        }

        $count           = count( $args );
        $success_message = ( 1 === $count ) ? '%d mod removed.' : '%d mods removed.';
        WP_CLI::success( sprintf( $success_message, $count ) );
    }

    /**
     * Sets the value of a theme mod.
     *
     * ## OPTIONS
     *
     * <mod>
     * : The name of the theme mod to set or update.
     *
     * <value>
     * : The new value.
     *
     * ## EXAMPLES
     *
     *     # Set theme mod
     *     $ wp theme mod set background_color 000000
     *     Success: Theme mod background_color set to 000000.
     */
    public function set( $args = array(), $assoc_args = array() ) {
        list( $mod, $value ) = $args;

        set_theme_mod( $mod, $value );

        if ( get_theme_mod( $mod ) === $value ) {
            WP_CLI::success( "Theme mod {$mod} set to {$value}." );
        } else {
            WP_CLI::success( "Could not update theme mod {$mod}." );
        }
    }
}