<?php

/**
 * MARC record reader class.
 *
 * PHP version 7
 *
 * Copyright (C) The National Library of Finland 2020-2022.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 * @category VuFind
 * @package  MARC
 * @author   Ere Maijala <ere.maijala@helsinki.fi>
 * @license  http://opensource.org/licenses/gpl-2.0.php GNU General Public License
 * @link     https://vufind.org/wiki/development Wiki
 */

namespace VuFind\Marc;

use function in_array;
use function intval;
use function is_array;
use function is_string;

/**
 * MARC record reader class.
 *
 * @category VuFind
 * @package  MARC
 * @author   Ere Maijala <ere.maijala@helsinki.fi>
 * @license  http://opensource.org/licenses/gpl-2.0.php GNU General Public License
 * @link     https://vufind.org/wiki/development Wiki
 */
class MarcReader
{
    /**
     * Supported serialization formats
     *
     * @var array
     */
    protected $serializations = [
        'ISO2709' => Serialization\Iso2709::class,
        'JSON' => Serialization\MarcInJson::class,
        'MARCXML' => Serialization\MarcXml::class,
    ];

    /**
     * MARC leader
     *
     * @var string
     */
    protected $leader;

    /**
     * MARC is stored in a multidimensional array resembling MARC-in-JSON
     * specification by Ross Singer:
     * [
     *     'leader' => '...',
     *     'fields' => [
     *         [
     *             '001' => '12345'
     *         ],
     *         [
     *             '245' => [
     *                 'ind1' => '0',
     *                 'ind2' => '1',
     *                 'subfields' => [
     *                      ['a' => 'Title'],
     *                      ['k' => 'Form'],
     *                      ['k' => 'Another'],
     *                      ['p' => 'Part'],
     *                 ]
     *             ]
     *         ]
     *     ]
     * ]
     *
     * @var array
     * @see https://web.archive.org/web/20151112001548/http://dilettantes.code4lib.org/blog/2010/09/a-proposal-to-serialize-marc-in-json/
     */
    protected $data;

    /**
     * Any warnings encountered when parsing a record
     *
     * @var array
     */
    protected $warnings;

    /**
     * Constructor
     *
     * @param string|array $data MARC record in one of the supported formats, or an
     * associative array with 'leader' and 'fields' in the internal format
     */
    public function __construct($data)
    {
        $this->setData($data);
    }

    /**
     * Set MARC record data
     *
     * @param string|array $data MARC record in one of the supported formats, or an
     * associative array with 'leader' and 'fields' in the internal format
     *
     * @throws \Exception
     * @return void
     */
    public function setData($data): void
    {
        $this->warnings = [];
        if (is_array($data)) {
            if (
                !is_string($data['leader'] ?? null)
                || !is_array($data['fields'] ?? null)
            ) {
                throw new \Exception('Invalid data array format provided');
            }
            $this->data = $data;
            return;
        }
        $valid = false;
        foreach ($this->serializations as $serialization) {
            if ($serialization::canParse($data)) {
                $this->data = $serialization::fromString($data);
                if (isset($this->data['warnings'])) {
                    $this->warnings = $this->data['warnings'];
                    unset($this->data['warnings']);
                }
                $valid = true;
                break;
            }
        }
        if (!$valid) {
            throw new \Exception('MARC record format not recognized');
        }
        // Make sure leader is 24 characters, and reset meaningless offsets:
        if ($this->data['leader']) {
            $leader = str_pad(substr($this->data['leader'] ?? '', 0, 24), 24);
            $this->data['leader'] = '00000' . substr($leader, 5, 7) . '00000'
                . substr($leader, 17);
        }
    }

    /**
     * Serialize the record
     *
     * @param string $format Format to return (e.g. 'ISO2709' or 'MARCXML')
     *
     * @throws \Exception
     * @return string
     */
    public function toFormat(string $format): string
    {
        $serialization = $this->serializations[$format] ?? null;
        if (null === $serialization) {
            throw new \Exception("Unknown MARC format '$format' requested");
        }
        return $serialization::toString($this->data);
    }

    /**
     * Return leader
     *
     * @return string
     */
    public function getLeader(): string
    {
        return $this->data['leader'];
    }

    /**
     * Return an associative array for a data field, a string for a control field or
     * an empty array if field does not exist
     *
     * @param string $fieldTag      The MARC field tag to get
     * @param array  $subfieldCodes The MARC subfield codes to get, or empty for all
     *
     * @return array|string
     */
    public function getField(string $fieldTag, ?array $subfieldCodes = null)
    {
        $results = $this->getFields($fieldTag, $subfieldCodes);
        return $results[0] ?? [];
    }

    /**
     * Return an associative array of fields for data fields or an array of values
     * for control fields
     *
     * @param string $fieldTag      The MARC field tag to get
     * @param array  $subfieldCodes The MARC subfield codes to get, or empty for all.
     * Ignored for control fields.
     *
     * @return array
     */
    public function getFields(string $fieldTag, ?array $subfieldCodes = null): array
    {
        $result = [];

        foreach ($this->data['fields'] as $fieldData) {
            if ($fieldTag && $fieldTag !== (string)key($fieldData)) {
                continue;
            }
            $field = current($fieldData);
            if (!is_array($field)) {
                // Control field
                $result[] = $field;
                continue;
            }
            $subfields = [];
            foreach ($field['subfields'] ?? [] as $subfield) {
                if (
                    $subfieldCodes
                    && !in_array((string)key($subfield), $subfieldCodes)
                ) {
                    continue;
                }
                $subfields[] = [
                    'code' => (string)key($subfield),
                    'data' => current($subfield),
                ];
            }
            if ($subfields) {
                $result[] = [
                    'tag' => $fieldTag,
                    'i1' => $field['ind1'],
                    'i2' => $field['ind2'],
                    'subfields' => $subfields,
                ];
            }
        }

        return $result;
    }

    /**
     * Return all fields as an array.
     *
     * Control fields have the following elements:
     * - tag
     * - data
     *
     * Data fields have the following elements:
     * - tag
     * - i1
     * - i2
     * - subfields
     *
     * @return array
     */
    public function getAllFields()
    {
        $result = [];

        foreach ($this->data['fields'] as $fieldData) {
            $tag = (string)key($fieldData);
            $field = current($fieldData);
            if (is_string($field)) {
                // Control field
                $result[] = [
                    'tag' => $tag,
                    'data' => $field,
                ];
                continue;
            }
            $subfields = [];
            foreach ($field['subfields'] ?? [] as $subfield) {
                $subfields[] = [
                    'code' => (string)key($subfield),
                    'data' => current($subfield),
                ];
            }
            if ($subfields) {
                $result[] = [
                    'tag' => $tag,
                    'i1' => $field['ind1'],
                    'i2' => $field['ind2'],
                    'subfields' => $subfields,
                ];
            }
        }

        return $result;
    }

    /**
     * Return first subfield with the given code in the MARC field provided by
     * getField or getFields
     *
     * @param array  $field        Result from MarcReader::getFields
     * @param string $subfieldCode The MARC subfield code to get
     *
     * @return string
     */
    public function getSubfield(array $field, string $subfieldCode): string
    {
        foreach ($field['subfields'] ?? [] as $current) {
            if ($current['code'] == $subfieldCode) {
                return trim($current['data']);
            }
        }

        return '';
    }

    /**
     * Return all subfields with the given code in the MARC field provided by
     * getField or getFields. Returns all subfields if subfieldCode is empty.
     *
     * @param array  $field        Result from MarcReader::getFields
     * @param string $subfieldCode The MARC subfield code to get
     *
     * @return array
     */
    public function getSubfields(array $field, string $subfieldCode = ''): array
    {
        $result = [];
        foreach ($field['subfields'] ?? [] as $current) {
            if ('' === $subfieldCode || $current['code'] == $subfieldCode) {
                $result[] = trim($current['data']);
            }
        }

        return $result;
    }

    /**
     * Return an array of all values extracted from the specified field/subfield
     * combination.  If multiple subfields and a separator are specified, the
     * subfields will be concatenated together in the order listed -- each entry in
     * the array will correspond with a single MARC field.  If $separator is null,
     * the return array will contain separate entries for all subfields.
     *
     * @param string $fieldTag      The MARC field tag to get
     * @param array  $subfieldCodes The MARC subfield codes to get
     * @param string $separator     Subfield separator string. Set to null to disable
     * concatenation of subfields.
     *
     * @return array
     */
    public function getFieldsSubfields(
        string $fieldTag,
        array $subfieldCodes,
        ?string $separator = ' '
    ): array {
        $result = [];

        foreach ($this->getInternalFields($fieldTag) as $field) {
            if (!isset($field['subfields'])) {
                continue;
            }
            $subfields = [];
            foreach ($field['subfields'] ?? [] as $subfield) {
                if (
                    $subfieldCodes
                    && !in_array((string)key($subfield), $subfieldCodes)
                ) {
                    continue;
                }
                if (null !== $separator) {
                    $subfields[] = current($subfield);
                } else {
                    $result[] = current($subfield);
                }
            }
            if (null !== $separator && $subfields) {
                $result[] = implode($separator, $subfields);
            }
        }

        return $result;
    }

    /**
     * Return an associative array for a linked field such as 880 (Alternate Graphic
     * Representation) or an empty array if field does not exist
     *
     * @param string $fieldTag       The MARC field that contains the linked fields
     * @param string $linkedFieldTag The linked MARC field tag to get
     * @param string $occurrence     The occurrence number to get; empty string for
     * whatever comes first
     * @param array  $subfieldCodes  The MARC subfield codes to get, or empty for all
     *
     * @return array
     */
    public function getLinkedField(
        string $fieldTag,
        string $linkedFieldTag,
        string $occurrence = '',
        ?array $subfieldCodes = null
    ): array {
        $results
            = $this->getLinkedFields($fieldTag, $linkedFieldTag, $subfieldCodes);
        foreach ($results as $field) {
            if (empty($occurrence) || $occurrence === $field['link']['occurrence']) {
                return $field;
            }
        }
        return [];
    }

    /**
     * Return an array of associative arrays for a linked field such as 880
     * (Alternate Graphic Representation)
     *
     * @param string $fieldTag       The MARC field that contains the linked fields
     * @param string $linkedFieldTag The linked MARC field tag to get
     * @param array  $subfieldCodes  The MARC subfield codes to get, or empty for all
     *
     * @return array
     */
    public function getLinkedFields(
        string $fieldTag,
        string $linkedFieldTag,
        ?array $subfieldCodes = null
    ): array {
        $result = [];

        foreach ($this->getInternalFields($fieldTag) as $field) {
            if (is_string($field)) {
                // Control field
                continue;
            }
            $link
                = $this->parseLinkageField($this->getInternalSubfield($field, '6'));
            if ($link['field'] !== $linkedFieldTag) {
                continue;
            }
            $subfields = [];
            foreach ($field['subfields'] ?? [] as $subfield) {
                if (
                    $subfieldCodes
                    && !in_array((string)key($subfield), $subfieldCodes)
                ) {
                    continue;
                }
                $subfields[] = [
                    'code' => (string)key($subfield),
                    'data' => current($subfield),
                ];
            }
            if ($subfields) {
                $result[] = [
                    'tag' => $fieldTag,
                    'i1' => $field['ind1'],
                    'i2' => $field['ind2'],
                    'subfields' => $subfields,
                    'link' => $link,
                ];
            }
        }

        return $result;
    }

    /**
     * Return an array of all values extracted from the specified linked
     * field/subfield combination.  If multiple subfields and a separator are
     * specified, the subfields will be concatenated together in the order listed
     * -- each entry in the array will correspond with a single MARC field.  If
     * $separator is null, the return array will contain separate entries for all
     * subfields.
     *
     * @param string $fieldTag       The MARC field that contains the linked fields
     * @param string $linkedFieldTag The linked MARC field tag to get
     * @param array  $subfieldCodes  The MARC subfield codes to get
     * @param string $separator      Subfield separator string. Set to null to
     * disable concatenation of subfields.
     *
     * @return array
     */
    public function getLinkedFieldsSubfields(
        string $fieldTag,
        string $linkedFieldTag,
        array $subfieldCodes,
        ?string $separator = ' '
    ): array {
        $result = [];
        foreach ($this->getLinkedFields($fieldTag, $linkedFieldTag, $subfieldCodes) as $field) {
            $subfields = $this->getSubfields($field);
            if (null !== $separator) {
                $result[] = implode($separator, $subfields);
            } else {
                $result = array_merge($result, $subfields);
            }
        }
        return $result;
    }

    /**
     * Get linked field data from subfield 6
     *
     * @param array $field Field
     *
     * @return array
     */
    public function getFieldLink(array $field): array
    {
        return $this->parseLinkageField($this->getSubfield($field, '6'));
    }

    /**
     * Parse a linkage field
     *
     * @param string $link Linkage field
     *
     * @return array
     */
    public function parseLinkageField(string $link): array
    {
        $linkParts = explode('/', $link, 3);
        $targetParts = explode('-', $linkParts[0]);
        return [
            'field' => $targetParts[0],
            'occurrence' => $targetParts[1] ?? '',
            'script' => $linkParts[1] ?? '',
            'orientation' => $linkParts[2] ?? '',
        ];
    }

    /**
     * Return a copy of the record with the specified fields and/or subfields
     * removed.
     *
     * Each rule can have the following elements:
     *
     * tag       - Tag the rule applies to (a regular expression).
     * subfields - Subfields codes to remove (a regular expression, optional).
     *             Default is to remove all subfields (and the field).
     *
     * Examples:
     *
     * $result = $reader->getFilteredRecord(
     *   [
     *     [
     *       'tag' => '9..'
     *     ]
     *   ]
     * );
     *
     * $result = $reader->getFilteredRecord(
     *   [
     *     [
     *       'tag' => '...',
     *       'subfields' => '0'
     *     ]
     *   ]
     * );
     *
     * @param array $rules Array of filtering rules
     *
     * @return MarcReader
     */
    public function getFilteredRecord(array $rules): MarcReader
    {
        $resultFields = [];
        foreach ($this->data['fields'] as $fieldData) {
            $tag = (string)key($fieldData);
            $field = current($fieldData);
            $fieldRules = $this->getFilteringRulesForTag($rules, $tag);
            if ($fieldRules) {
                if (is_string($field)) {
                    // Control field, filter out completely
                    continue;
                }
                $field['subfields'] = $this->filterSubfields(
                    $fieldRules,
                    $field['subfields']
                );
                if (!$field['subfields']) {
                    // No subfields left, drop the field
                    continue;
                }
                $resultFields[] = [$tag => $field];
            } else {
                $resultFields[] = [$tag => $field];
            }
        }
        return new MarcReader(
            [
                'leader' => $this->data['leader'],
                'fields' => $resultFields,
            ]
        );
    }

    /**
     * Get filtering rules matching a field tag
     *
     * @param array  $rules Filtering rules
     * @param string $tag   Field tag
     *
     * @return array
     */
    protected function getFilteringRulesForTag(array $rules, string $tag): array
    {
        $result = [];
        foreach ($rules as $rule) {
            if (
                preg_match('/' . $rule['tag'] . '/', $tag)
                && (!isset($rule['subfields']) || intval($tag) >= 10)
            ) {
                $result[] = $rule;
            }
        }
        return $result;
    }

    /**
     * Filter subfields
     *
     * @param array $rules     Filtering rules
     * @param array $subfields Subfields
     *
     * @return array
     */
    protected function filterSubfields(array $rules, array $subfields): array
    {
        foreach ($rules as $rule) {
            if (!isset($rule['subfields'])) {
                // No subfields specified, filter out all of them
                return [];
            }
            $remaining = [];
            foreach ($subfields as $subfield) {
                $code = (string)key($subfield);
                if (!preg_match('/' . $rule['subfields'] . '/', $code)) {
                    $remaining[] = $subfield;
                }
            }
            if (!$remaining) {
                return [];
            }
            $subfields = $remaining;
        }

        return $subfields;
    }

    /**
     * Get any warnings encountered when parsing a record
     *
     * @return array
     */
    public function getWarnings(): array
    {
        return $this->warnings;
    }

    /**
     * Return fields by tag in internal format
     *
     * @param string $tag Field tag
     *
     * @return array
     */
    protected function getInternalFields(string $tag): array
    {
        $result = [];
        foreach ($this->data['fields'] as $field) {
            $fieldTag = (string)key($field);
            if ($fieldTag === $tag) {
                $result[] = current($field);
            }
        }
        return $result;
    }

    /**
     * Return first subfield with the given code in the internal MARC field
     *
     * @param array  $field        Internal MARC field
     * @param string $subfieldCode The MARC subfield code to get
     *
     * @return string
     */
    protected function getInternalSubfield(
        array $field,
        string $subfieldCode
    ): string {
        foreach ($field['subfields'] ?? [] as $subfield) {
            if ((string)key($subfield) === $subfieldCode) {
                return trim(current($subfield));
            }
        }
        return '';
    }
}