/*

 * Copyright (c) 2023, kleines Filmröllchen <filmroellchen@serenityos.org>

 *

 * SPDX-License-Identifier: BSD-2-Clause

 */

#include "VorbisComment.h"

#include <AK/Endian.h>

#include <AK/MemoryStream.h>

#include <AK/Span.h>

#include <AK/String.h>

namespace Audio {

static StringView vorbis_field_for_role(Person::Role role)

{

    static HashMap<Person::Role, StringView> role_map {

        { Person::Role::Artist, "ARTIST"sv },

        { Person::Role::Performer, "PERFORMER"sv },

        { Person::Role::Lyricist, "LYRICIST"sv },

        { Person::Role::Conductor, "CONDUCTOR"sv },

        { Person::Role::Publisher, "PUBLISHER"sv },

        { Person::Role::Engineer, "ENCODED-BY"sv },

        { Person::Role::Composer, "COMPOSER"sv },

    };

    return role_map.get(role).value();

}

// "Content vector format"

static ErrorOr<void> read_vorbis_field(Metadata& metadata_to_write_into, String const& unparsed_user_comment)

{

    // Technically the field name has to be ASCII, but we just accept all UTF-8.

    auto field_name_and_contents = TRY(unparsed_user_comment.split_limit('=', 2));

    if (field_name_and_contents.size() != 2)

        return Error::from_string_literal("User comment does not contain '='");

    auto contents = field_name_and_contents.take_last();

    auto field_name = TRY(field_name_and_contents.take_first().to_uppercase());

    // Some of these are taken from https://age.hobba.nl/audio/tag_frame_reference.html

    if (field_name == "TITLE"sv) {

        if (metadata_to_write_into.title.has_value())

            TRY(metadata_to_write_into.add_miscellaneous(field_name, contents));

        else

            metadata_to_write_into.title = contents;

    } else if (field_name == "VERSION"sv) {

        if (metadata_to_write_into.subtitle.has_value())

            TRY(metadata_to_write_into.add_miscellaneous(field_name, contents));

        else

            metadata_to_write_into.subtitle = contents;

    } else if (field_name == "ALBUM"sv) {

        if (metadata_to_write_into.album.has_value())

            TRY(metadata_to_write_into.add_miscellaneous(field_name, contents));

        else

            metadata_to_write_into.album = contents;

    } else if (field_name == "COPYRIGHT"sv) {

        if (metadata_to_write_into.copyright.has_value())

            TRY(metadata_to_write_into.add_miscellaneous(field_name, contents));

        else

            metadata_to_write_into.copyright = contents;

    } else if (field_name == "ISRC"sv) {

        if (metadata_to_write_into.isrc.has_value())

            TRY(metadata_to_write_into.add_miscellaneous(field_name, contents));

        else

            metadata_to_write_into.isrc = contents;

    } else if (field_name == "GENRE"sv) {

        if (metadata_to_write_into.genre.has_value())

            TRY(metadata_to_write_into.add_miscellaneous(field_name, contents));

        else

            metadata_to_write_into.genre = contents;

    } else if (field_name == "COMMENT"sv) {

        if (metadata_to_write_into.comment.has_value())

            TRY(metadata_to_write_into.add_miscellaneous(field_name, contents));

        else

            metadata_to_write_into.comment = contents;

    } else if (field_name == "TRACKNUMBER"sv) {

        if (metadata_to_write_into.track_number.has_value())

            TRY(metadata_to_write_into.add_miscellaneous(field_name, contents));

        else if (auto maybe_number = contents.to_number<unsigned>(); maybe_number.has_value())

            metadata_to_write_into.track_number = maybe_number.release_value();

        else

            TRY(metadata_to_write_into.add_miscellaneous(field_name, contents));

    } else if (field_name == "DATE"sv) {

        if (metadata_to_write_into.unparsed_time.has_value())

            TRY(metadata_to_write_into.add_miscellaneous(field_name, contents));

        else

            metadata_to_write_into.unparsed_time = contents;

    } else if (field_name == vorbis_field_for_role(Person::Role::Performer)) {

        TRY(metadata_to_write_into.add_person(Person::Role::Performer, contents));

    } else if (field_name == vorbis_field_for_role(Person::Role::Artist)) {

        TRY(metadata_to_write_into.add_person(Person::Role::Artist, contents));

    } else if (field_name == vorbis_field_for_role(Person::Role::Composer)) {

        TRY(metadata_to_write_into.add_person(Person::Role::Composer, contents));

    } else if (field_name == vorbis_field_for_role(Person::Role::Conductor)) {

        TRY(metadata_to_write_into.add_person(Person::Role::Conductor, contents));

    } else if (field_name == vorbis_field_for_role(Person::Role::Lyricist)) {

        TRY(metadata_to_write_into.add_person(Person::Role::Lyricist, contents));

    } else if (field_name == "ORGANIZATION"sv) {

        TRY(metadata_to_write_into.add_person(Person::Role::Publisher, contents));

    } else if (field_name == vorbis_field_for_role(Person::Role::Publisher)) {

        TRY(metadata_to_write_into.add_person(Person::Role::Publisher, contents));

    } else if (field_name == vorbis_field_for_role(Person::Role::Engineer)) {

        TRY(metadata_to_write_into.add_person(Person::Role::Engineer, contents));

    } else {

        TRY(metadata_to_write_into.add_miscellaneous(field_name, contents));

    }

    return {};

}

ErrorOr<Metadata, LoaderError> load_vorbis_comment(ByteBuffer const& vorbis_comment)

{

    FixedMemoryStream stream { vorbis_comment };

    auto vendor_length = TRY(stream.read_value<LittleEndian<u32>>());

    Vector<u8> raw_vendor_string;

    TRY(raw_vendor_string.try_resize(vendor_length));

    TRY(stream.read_until_filled(raw_vendor_string));

    auto vendor_string = TRY(String::from_utf8(StringView { raw_vendor_string.span() }));

    Metadata metadata;

    metadata.encoder = move(vendor_string);

    auto user_comment_count = TRY(stream.read_value<LittleEndian<u32>>());

    for (size_t i = 0; i < user_comment_count; ++i) {

        auto user_comment_length = TRY(stream.read_value<LittleEndian<u32>>());

        Vector<u8> raw_user_comment;

        TRY(raw_user_comment.try_resize(user_comment_length));

        TRY(stream.read_until_filled(raw_user_comment));

        auto unparsed_user_comment = TRY(String::from_utf8(StringView { raw_user_comment.span() }));

        TRY(read_vorbis_field(metadata, unparsed_user_comment));

    }

    return metadata;

}

struct VorbisCommentPair {

    String field_name;

    String contents;

};

static ErrorOr<Vector<VorbisCommentPair>> make_vorbis_user_comments(Metadata const& metadata)

{

    Vector<VorbisCommentPair> user_comments;

    auto add_if_present = [&](auto field_name, auto const& value) -> ErrorOr<void> {

        if (value.has_value())

            TRY(user_comments.try_append(VorbisCommentPair { field_name, TRY(String::formatted("{}", value.value())) }));

        return {};

    };

Unexecuted instantiation: VorbisComment.cpp:AK::ErrorOr<void, AK::Error> Audio::make_vorbis_user_comments(Audio::Metadata const&)::$_0::operator()<AK::String, AK::Optional<AK::String> >(AK::String, AK::Optional<AK::String> const&) const

Unexecuted instantiation: VorbisComment.cpp:AK::ErrorOr<void, AK::Error> Audio::make_vorbis_user_comments(Audio::Metadata const&)::$_0::operator()<AK::String, AK::Optional<unsigned int> >(AK::String, AK::Optional<unsigned int> const&) const

    TRY(add_if_present("TITLE"_string, metadata.title));

    TRY(add_if_present("VERSION"_string, metadata.subtitle));

    TRY(add_if_present("ALBUM"_string, metadata.album));

    TRY(add_if_present("COPYRIGHT"_string, metadata.copyright));

    TRY(add_if_present("ISRC"_string, metadata.isrc));

    TRY(add_if_present("GENRE"_string, metadata.genre));

    TRY(add_if_present("COMMENT"_string, metadata.comment));

    TRY(add_if_present("TRACKNUMBER"_string, metadata.track_number));

    TRY(add_if_present("DATE"_string, metadata.unparsed_time));

    for (auto const& person : metadata.people)

        TRY(user_comments.try_append(VorbisCommentPair { TRY(String::from_utf8(vorbis_field_for_role(person.role))), person.name }));

    for (auto const& field : metadata.miscellaneous) {

        for (auto const& value : field.value)

            TRY(user_comments.try_append(VorbisCommentPair { field.key, value }));

    }

    return user_comments;

}

ErrorOr<void> write_vorbis_comment(Metadata const& metadata, Stream& target)

{

    auto encoder = metadata.encoder.value_or({}).bytes();

    TRY(target.write_value<LittleEndian<u32>>(encoder.size()));

    TRY(target.write_until_depleted(encoder));

    auto vorbis_user_comments = TRY(make_vorbis_user_comments(metadata));

    TRY(target.write_value<LittleEndian<u32>>(vorbis_user_comments.size()));

    for (auto const& field : vorbis_user_comments) {

        auto const serialized_field = TRY(String::formatted("{}={}", field.field_name, field.contents));

        TRY(target.write_value<LittleEndian<u32>>(serialized_field.bytes().size()));

        TRY(target.write_until_depleted(serialized_field.bytes()));

    }

    return {};

}

}