<?php

namespace WP_CLI\I18n;

use DirectoryIterator;
use Gettext\Translations;
use IteratorIterator;
use SplFileInfo;
use WP_CLI;
use WP_CLI\Utils;
use WP_CLI_Command;

class MakeMoCommand extends WP_CLI_Command {
    /**
     * Create MO files from PO files.
     *
     * ## OPTIONS
     *
     * <source>
     * : Path to an existing PO file or a directory containing multiple PO files.
     *
     * [<destination>]
     * : Path to the destination file or directory for the resulting MO files. Defaults to the source directory.
     *
     * ## EXAMPLES
     *
     *     # Create MO files for all PO files in the current directory.
     *     $ wp i18n make-mo .
     *
     *     # Create a MO file from a single PO file in a specific directory.
     *     $ wp i18n make-mo example-plugin-de_DE.po languages
     *
     *     # Create a MO file from a single PO file to a specific file destination
     *     $ wp i18n make-mo example-plugin-de_DE.po languages/bar.mo
     *
     * @when before_wp_load
     *
     * @throws WP_CLI\ExitException
     */
    public function __invoke( $args, $assoc_args ) {
        $source = realpath( $args[0] );
        if ( ! $source || ( ! is_file( $source ) && ! is_dir( $source ) ) ) {
            WP_CLI::error( 'Source file or directory does not exist.' );
        }

        $destination      = is_file( $source ) ? dirname( $source ) : $source;
        $custom_file_name = null;
        if ( isset( $args[1] ) ) {
            $destination          = $args[1];
            $destination_pathinfo = pathinfo( $destination );
            // Destination is a file, make sure source is also a file
            if ( ! empty( $destination_pathinfo['filename'] ) && ! empty( $destination_pathinfo['extension'] ) ) {
                if ( ! is_file( $source ) ) {
                    WP_CLI::error( 'Destination file not supported when source is a directory.' );
                }
                $destination      = $destination_pathinfo['dirname'];
                $custom_file_name = $destination_pathinfo['filename'] . '.' . $destination_pathinfo['extension'];
            }
        }

        if ( ! is_dir( $destination ) && ! mkdir( $destination, 0777, true ) ) {
            WP_CLI::error( 'Could not create destination directory.' );
        }

        if ( is_file( $source ) ) {
            $files = [ new SplFileInfo( $source ) ];
        } else {
            $files = new IteratorIterator( new DirectoryIterator( $source ) );
        }

        $result_count = 0;
        /** @var DirectoryIterator $file */
        foreach ( $files as $file ) {
            if ( 'po' !== $file->getExtension() ) {
                continue;
            }

            if ( ! $file->isFile() || ! $file->isReadable() ) {
                WP_CLI::warning( sprintf( 'Could not read file %s', $file->getFilename() ) );
                continue;
            }

            $file_basename = basename( $file->getFilename(), '.po' );
            $file_name     = $file_basename . '.mo';
            if ( $custom_file_name ) {
                $file_name = $custom_file_name;
            }
            $destination_file = "{$destination}/{$file_name}";

            $translations = Translations::fromPoFile( $file->getPathname() );
            if ( ! $translations->toMoFile( $destination_file ) ) {
                WP_CLI::warning( sprintf( 'Could not create file %s', $destination_file ) );
                continue;
            }

            ++$result_count;
        }

        WP_CLI::success( sprintf( 'Created %d %s.', $result_count, Utils\pluralize( 'file', $result_count ) ) );
    }
}