/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "Resource.h"
#include "ResourceUtils.h"
#include "ResourceValues.h"
#include "ValueVisitor.h"
#include "util/Util.h"

#include <algorithm>
#include <androidfw/ResourceTypes.h>
#include <limits>
#include <set>

namespace aapt {

template <typename Derived>
void BaseValue<Derived>::accept(RawValueVisitor* visitor) {
    visitor->visit(static_cast<Derived*>(this));
}

template <typename Derived>
void BaseItem<Derived>::accept(RawValueVisitor* visitor) {
    visitor->visit(static_cast<Derived*>(this));
}

RawString::RawString(const StringPool::Ref& ref) : value(ref) {
}

bool RawString::equals(const Value* value) const {
    const RawString* other = valueCast<RawString>(value);
    if (!other) {
        return false;
    }
    return *this->value == *other->value;
}

RawString* RawString::clone(StringPool* newPool) const {
    RawString* rs = new RawString(newPool->makeRef(*value));
    rs->mComment = mComment;
    rs->mSource = mSource;
    return rs;
}

bool RawString::flatten(android::Res_value* outValue) const {
    outValue->dataType = android::Res_value::TYPE_STRING;
    outValue->data = util::hostToDevice32(static_cast<uint32_t>(value.getIndex()));
    return true;
}

void RawString::print(std::ostream* out) const {
    *out << "(raw string) " << *value;
}

Reference::Reference() : referenceType(Type::kResource) {
}

Reference::Reference(const ResourceNameRef& n, Type t) :
        name(n.toResourceName()), referenceType(t) {
}

Reference::Reference(const ResourceId& i, Type type) : id(i), referenceType(type) {
}

Reference::Reference(const ResourceNameRef& n, const ResourceId& i) :
        name(n.toResourceName()), id(i), referenceType(Type::kResource) {
}

bool Reference::equals(const Value* value) const {
    const Reference* other = valueCast<Reference>(value);
    if (!other) {
        return false;
    }
    return referenceType == other->referenceType && privateReference == other->privateReference &&
            id == other->id && name == other->name;
}

bool Reference::flatten(android::Res_value* outValue) const {
    outValue->dataType = (referenceType == Reference::Type::kResource) ?
            android::Res_value::TYPE_REFERENCE : android::Res_value::TYPE_ATTRIBUTE;
    outValue->data = util::hostToDevice32(id ? id.value().id : 0);
    return true;
}

Reference* Reference::clone(StringPool* /*newPool*/) const {
    return new Reference(*this);
}

void Reference::print(std::ostream* out) const {
    *out << "(reference) ";
    if (referenceType == Reference::Type::kResource) {
        *out << "@";
        if (privateReference) {
            *out << "*";
        }
    } else {
        *out << "?";
    }

    if (name) {
        *out << name.value();
    }

    if (id && !Res_INTERNALID(id.value().id)) {
        *out << " " << id.value();
    }
}

bool Id::equals(const Value* value) const {
    return valueCast<Id>(value) != nullptr;
}

bool Id::flatten(android::Res_value* out) const {
    out->dataType = android::Res_value::TYPE_INT_BOOLEAN;
    out->data = util::hostToDevice32(0);
    return true;
}

Id* Id::clone(StringPool* /*newPool*/) const {
    return new Id(*this);
}

void Id::print(std::ostream* out) const {
    *out << "(id)";
}

String::String(const StringPool::Ref& ref) : value(ref) {
}

bool String::equals(const Value* value) const {
    const String* other = valueCast<String>(value);
    if (!other) {
        return false;
    }
    return *this->value == *other->value;
}

bool String::flatten(android::Res_value* outValue) const {
    // Verify that our StringPool index is within encode-able limits.
    if (value.getIndex() > std::numeric_limits<uint32_t>::max()) {
        return false;
    }

    outValue->dataType = android::Res_value::TYPE_STRING;
    outValue->data = util::hostToDevice32(static_cast<uint32_t>(value.getIndex()));
    return true;
}

String* String::clone(StringPool* newPool) const {
    String* str = new String(newPool->makeRef(*value));
    str->mComment = mComment;
    str->mSource = mSource;
    return str;
}

void String::print(std::ostream* out) const {
    *out << "(string) \"" << *value << "\"";
}

StyledString::StyledString(const StringPool::StyleRef& ref) : value(ref) {
}

bool StyledString::equals(const Value* value) const {
    const StyledString* other = valueCast<StyledString>(value);
    if (!other) {
        return false;
    }

    if (*this->value->str == *other->value->str) {
        const std::vector<StringPool::Span>& spansA = this->value->spans;
        const std::vector<StringPool::Span>& spansB = other->value->spans;
        return std::equal(spansA.begin(), spansA.end(), spansB.begin(),
                          [](const StringPool::Span& a, const StringPool::Span& b) -> bool {
            return *a.name == *b.name && a.firstChar == b.firstChar && a.lastChar == b.lastChar;
        });
    }
    return false;
}

bool StyledString::flatten(android::Res_value* outValue) const {
    if (value.getIndex() > std::numeric_limits<uint32_t>::max()) {
        return false;
    }

    outValue->dataType = android::Res_value::TYPE_STRING;
    outValue->data = util::hostToDevice32(static_cast<uint32_t>(value.getIndex()));
    return true;
}

StyledString* StyledString::clone(StringPool* newPool) const {
    StyledString* str = new StyledString(newPool->makeRef(value));
    str->mComment = mComment;
    str->mSource = mSource;
    return str;
}

void StyledString::print(std::ostream* out) const {
    *out << "(styled string) \"" << *value->str << "\"";
    for (const StringPool::Span& span : value->spans) {
        *out << " "<< *span.name << ":" << span.firstChar << "," << span.lastChar;
    }
}

FileReference::FileReference(const StringPool::Ref& _path) : path(_path) {
}

bool FileReference::equals(const Value* value) const {
    const FileReference* other = valueCast<FileReference>(value);
    if (!other) {
        return false;
    }
    return *path == *other->path;
}

bool FileReference::flatten(android::Res_value* outValue) const {
    if (path.getIndex() > std::numeric_limits<uint32_t>::max()) {
        return false;
    }

    outValue->dataType = android::Res_value::TYPE_STRING;
    outValue->data = util::hostToDevice32(static_cast<uint32_t>(path.getIndex()));
    return true;
}

FileReference* FileReference::clone(StringPool* newPool) const {
    FileReference* fr = new FileReference(newPool->makeRef(*path));
    fr->file = file;
    fr->mComment = mComment;
    fr->mSource = mSource;
    return fr;
}

void FileReference::print(std::ostream* out) const {
    *out << "(file) " << *path;
}

BinaryPrimitive::BinaryPrimitive(const android::Res_value& val) : value(val) {
}

BinaryPrimitive::BinaryPrimitive(uint8_t dataType, uint32_t data) {
    value.dataType = dataType;
    value.data = data;
}

bool BinaryPrimitive::equals(const Value* value) const {
    const BinaryPrimitive* other = valueCast<BinaryPrimitive>(value);
    if (!other) {
        return false;
    }
    return this->value.dataType == other->value.dataType && this->value.data == other->value.data;
}

bool BinaryPrimitive::flatten(android::Res_value* outValue) const {
    outValue->dataType = value.dataType;
    outValue->data = util::hostToDevice32(value.data);
    return true;
}

BinaryPrimitive* BinaryPrimitive::clone(StringPool* /*newPool*/) const {
    return new BinaryPrimitive(*this);
}

void BinaryPrimitive::print(std::ostream* out) const {
    switch (value.dataType) {
        case android::Res_value::TYPE_NULL:
            *out << "(null)";
            break;
        case android::Res_value::TYPE_INT_DEC:
            *out << "(integer) " << static_cast<int32_t>(value.data);
            break;
        case android::Res_value::TYPE_INT_HEX:
            *out << "(integer) 0x" << std::hex << value.data << std::dec;
            break;
        case android::Res_value::TYPE_INT_BOOLEAN:
            *out << "(boolean) " << (value.data != 0 ? "true" : "false");
            break;
        case android::Res_value::TYPE_INT_COLOR_ARGB8:
        case android::Res_value::TYPE_INT_COLOR_RGB8:
        case android::Res_value::TYPE_INT_COLOR_ARGB4:
        case android::Res_value::TYPE_INT_COLOR_RGB4:
            *out << "(color) #" << std::hex << value.data << std::dec;
            break;
        default:
            *out << "(unknown 0x" << std::hex << (int) value.dataType << ") 0x"
                 << std::hex << value.data << std::dec;
            break;
    }
}

Attribute::Attribute(bool w, uint32_t t) :
        typeMask(t),
        minInt(std::numeric_limits<int32_t>::min()),
        maxInt(std::numeric_limits<int32_t>::max()) {
    mWeak = w;
}

template <typename T>
T* addPointer(T& val) {
    return &val;
}

bool Attribute::equals(const Value* value) const {
    const Attribute* other = valueCast<Attribute>(value);
    if (!other) {
        return false;
    }

    if (symbols.size() != other->symbols.size()) {
        return false;
    }

    if (typeMask != other->typeMask || minInt != other->minInt || maxInt != other->maxInt) {
        return false;
    }

    std::vector<const Symbol*> sortedA;
    std::transform(symbols.begin(), symbols.end(),
                   std::back_inserter(sortedA), addPointer<const Symbol>);
    std::sort(sortedA.begin(), sortedA.end(), [](const Symbol* a, const Symbol* b) -> bool {
        return a->symbol.name < b->symbol.name;
    });

    std::vector<const Symbol*> sortedB;
    std::transform(other->symbols.begin(), other->symbols.end(),
                   std::back_inserter(sortedB), addPointer<const Symbol>);
    std::sort(sortedB.begin(), sortedB.end(), [](const Symbol* a, const Symbol* b) -> bool {
        return a->symbol.name < b->symbol.name;
    });

    return std::equal(sortedA.begin(), sortedA.end(), sortedB.begin(),
                      [](const Symbol* a, const Symbol* b) -> bool {
        return a->symbol.equals(&b->symbol) && a->value == b->value;
    });
}

Attribute* Attribute::clone(StringPool* /*newPool*/) const {
    return new Attribute(*this);
}

void Attribute::printMask(std::ostream* out) const {
    if (typeMask == android::ResTable_map::TYPE_ANY) {
        *out << "any";
        return;
    }

    bool set = false;
    if ((typeMask & android::ResTable_map::TYPE_REFERENCE) != 0) {
        if (!set) {
            set = true;
        } else {
            *out << "|";
        }
        *out << "reference";
    }

    if ((typeMask & android::ResTable_map::TYPE_STRING) != 0) {
        if (!set) {
            set = true;
        } else {
            *out << "|";
        }
        *out << "string";
    }

    if ((typeMask & android::ResTable_map::TYPE_INTEGER) != 0) {
        if (!set) {
            set = true;
        } else {
            *out << "|";
        }
        *out << "integer";
    }

    if ((typeMask & android::ResTable_map::TYPE_BOOLEAN) != 0) {
        if (!set) {
            set = true;
        } else {
            *out << "|";
        }
        *out << "boolean";
    }

    if ((typeMask & android::ResTable_map::TYPE_COLOR) != 0) {
        if (!set) {
            set = true;
        } else {
            *out << "|";
        }
        *out << "color";
    }

    if ((typeMask & android::ResTable_map::TYPE_FLOAT) != 0) {
        if (!set) {
            set = true;
        } else {
            *out << "|";
        }
        *out << "float";
    }

    if ((typeMask & android::ResTable_map::TYPE_DIMENSION) != 0) {
        if (!set) {
            set = true;
        } else {
            *out << "|";
        }
        *out << "dimension";
    }

    if ((typeMask & android::ResTable_map::TYPE_FRACTION) != 0) {
        if (!set) {
            set = true;
        } else {
            *out << "|";
        }
        *out << "fraction";
    }

    if ((typeMask & android::ResTable_map::TYPE_ENUM) != 0) {
        if (!set) {
            set = true;
        } else {
            *out << "|";
        }
        *out << "enum";
    }

    if ((typeMask & android::ResTable_map::TYPE_FLAGS) != 0) {
        if (!set) {
            set = true;
        } else {
            *out << "|";
        }
        *out << "flags";
    }
}

void Attribute::print(std::ostream* out) const {
    *out << "(attr) ";
    printMask(out);

    if (!symbols.empty()) {
        *out << " [" << util::joiner(symbols, ", ") << "]";
    }

    if (minInt != std::numeric_limits<int32_t>::min()) {
        *out << " min=" << minInt;
    }

    if (maxInt != std::numeric_limits<int32_t>::max()) {
        *out << " max=" << maxInt;
    }

    if (isWeak()) {
        *out << " [weak]";
    }
}

static void buildAttributeMismatchMessage(DiagMessage* msg, const Attribute* attr,
                                          const Item* value) {
    *msg << "expected";
    if (attr->typeMask & android::ResTable_map::TYPE_BOOLEAN) {
        *msg << " boolean";
    }

    if (attr->typeMask & android::ResTable_map::TYPE_COLOR) {
        *msg << " color";
    }

    if (attr->typeMask & android::ResTable_map::TYPE_DIMENSION) {
        *msg << " dimension";
    }

    if (attr->typeMask & android::ResTable_map::TYPE_ENUM) {
        *msg << " enum";
    }

    if (attr->typeMask & android::ResTable_map::TYPE_FLAGS) {
        *msg << " flags";
    }

    if (attr->typeMask & android::ResTable_map::TYPE_FLOAT) {
        *msg << " float";
    }

    if (attr->typeMask & android::ResTable_map::TYPE_FRACTION) {
        *msg << " fraction";
    }

    if (attr->typeMask & android::ResTable_map::TYPE_INTEGER) {
        *msg << " integer";
    }

    if (attr->typeMask & android::ResTable_map::TYPE_REFERENCE) {
        *msg << " reference";
    }

    if (attr->typeMask & android::ResTable_map::TYPE_STRING) {
        *msg << " string";
    }

    *msg << " but got " << *value;
}

bool Attribute::matches(const Item* item, DiagMessage* outMsg) const {
    android::Res_value val = {};
    item->flatten(&val);

    // Always allow references.
    const uint32_t mask = typeMask | android::ResTable_map::TYPE_REFERENCE;
    if (!(mask & ResourceUtils::androidTypeToAttributeTypeMask(val.dataType))) {
        if (outMsg) {
            buildAttributeMismatchMessage(outMsg, this, item);
        }
        return false;

    } else if (ResourceUtils::androidTypeToAttributeTypeMask(val.dataType) &
            android::ResTable_map::TYPE_INTEGER) {
        if (static_cast<int32_t>(util::deviceToHost32(val.data)) < minInt) {
            if (outMsg) {
                *outMsg << *item << " is less than minimum integer " << minInt;
            }
            return false;
        } else if (static_cast<int32_t>(util::deviceToHost32(val.data)) > maxInt) {
            if (outMsg) {
                *outMsg << *item << " is greater than maximum integer " << maxInt;
            }
            return false;
        }
    }
    return true;
}

bool Style::equals(const Value* value) const {
    const Style* other = valueCast<Style>(value);
    if (!other) {
        return false;
    }
    if (bool(parent) != bool(other->parent) ||
            (parent && other->parent && !parent.value().equals(&other->parent.value()))) {
        return false;
    }

    if (entries.size() != other->entries.size()) {
        return false;
    }

    std::vector<const Entry*> sortedA;
    std::transform(entries.begin(), entries.end(),
                   std::back_inserter(sortedA), addPointer<const Entry>);
    std::sort(sortedA.begin(), sortedA.end(), [](const Entry* a, const Entry* b) -> bool {
        return a->key.name < b->key.name;
    });

    std::vector<const Entry*> sortedB;
    std::transform(other->entries.begin(), other->entries.end(),
                   std::back_inserter(sortedB), addPointer<const Entry>);
    std::sort(sortedB.begin(), sortedB.end(), [](const Entry* a, const Entry* b) -> bool {
        return a->key.name < b->key.name;
    });

    return std::equal(sortedA.begin(), sortedA.end(), sortedB.begin(),
                      [](const Entry* a, const Entry* b) -> bool {
        return a->key.equals(&b->key) && a->value->equals(b->value.get());
    });
}

Style* Style::clone(StringPool* newPool) const {
    Style* style = new Style();
    style->parent = parent;
    style->parentInferred = parentInferred;
    style->mComment = mComment;
    style->mSource = mSource;
    for (auto& entry : entries) {
        style->entries.push_back(Entry{
                entry.key,
                std::unique_ptr<Item>(entry.value->clone(newPool))
        });
    }
    return style;
}

void Style::print(std::ostream* out) const {
    *out << "(style) ";
    if (parent && parent.value().name) {
        if (parent.value().privateReference) {
            *out << "*";
        }
        *out << parent.value().name.value();
    }
    *out << " ["
        << util::joiner(entries, ", ")
        << "]";
}

static ::std::ostream& operator<<(::std::ostream& out, const Style::Entry& value) {
    if (value.key.name) {
        out << value.key.name.value();
    } else if (value.key.id) {
        out << value.key.id.value();
    } else {
        out << "???";
    }
    out << " = ";
    value.value->print(&out);
    return out;
}

bool Array::equals(const Value* value) const {
    const Array* other = valueCast<Array>(value);
    if (!other) {
        return false;
    }

    if (items.size() != other->items.size()) {
        return false;
    }

    return std::equal(items.begin(), items.end(), other->items.begin(),
                      [](const std::unique_ptr<Item>& a, const std::unique_ptr<Item>& b) -> bool {
        return a->equals(b.get());
    });
}

Array* Array::clone(StringPool* newPool) const {
    Array* array = new Array();
    array->mComment = mComment;
    array->mSource = mSource;
    for (auto& item : items) {
        array->items.emplace_back(std::unique_ptr<Item>(item->clone(newPool)));
    }
    return array;
}

void Array::print(std::ostream* out) const {
    *out << "(array) ["
        << util::joiner(items, ", ")
        << "]";
}

bool Plural::equals(const Value* value) const {
    const Plural* other = valueCast<Plural>(value);
    if (!other) {
        return false;
    }

    if (values.size() != other->values.size()) {
        return false;
    }

    return std::equal(values.begin(), values.end(), other->values.begin(),
                      [](const std::unique_ptr<Item>& a, const std::unique_ptr<Item>& b) -> bool {
        if (bool(a) != bool(b)) {
            return false;
        }
        return bool(a) == bool(b) || a->equals(b.get());
    });
}

Plural* Plural::clone(StringPool* newPool) const {
    Plural* p = new Plural();
    p->mComment = mComment;
    p->mSource = mSource;
    const size_t count = values.size();
    for (size_t i = 0; i < count; i++) {
        if (values[i]) {
            p->values[i] = std::unique_ptr<Item>(values[i]->clone(newPool));
        }
    }
    return p;
}

void Plural::print(std::ostream* out) const {
    *out << "(plural)";
    if (values[Zero]) {
        *out << " zero=" << *values[Zero];
    }

    if (values[One]) {
        *out << " one=" << *values[One];
    }

    if (values[Two]) {
        *out << " two=" << *values[Two];
    }

    if (values[Few]) {
        *out << " few=" << *values[Few];
    }

    if (values[Many]) {
        *out << " many=" << *values[Many];
    }
}

static ::std::ostream& operator<<(::std::ostream& out, const std::unique_ptr<Item>& item) {
    return out << *item;
}

bool Styleable::equals(const Value* value) const {
    const Styleable* other = valueCast<Styleable>(value);
    if (!other) {
        return false;
    }

    if (entries.size() != other->entries.size()) {
        return false;
    }

    return std::equal(entries.begin(), entries.end(), other->entries.begin(),
                      [](const Reference& a, const Reference& b) -> bool {
        return a.equals(&b);
    });
}

Styleable* Styleable::clone(StringPool* /*newPool*/) const {
    return new Styleable(*this);
}

void Styleable::print(std::ostream* out) const {
    *out << "(styleable) " << " ["
        << util::joiner(entries, ", ")
        << "]";
}

bool operator<(const Reference& a, const Reference& b) {
    int cmp = a.name.valueOrDefault({}).compare(b.name.valueOrDefault({}));
    if (cmp != 0) return cmp < 0;
    return a.id < b.id;
}

bool operator==(const Reference& a, const Reference& b) {
    return a.name == b.name && a.id == b.id;
}

bool operator!=(const Reference& a, const Reference& b) {
    return a.name != b.name || a.id != b.id;
}

struct NameOnlyComparator {
    bool operator()(const Reference& a, const Reference& b) const {
        return a.name < b.name;
    }
};

void Styleable::mergeWith(Styleable* other) {
    // Compare only names, because some References may already have their IDs assigned
    // (framework IDs that don't change).
    std::set<Reference, NameOnlyComparator> references;
    references.insert(entries.begin(), entries.end());
    references.insert(other->entries.begin(), other->entries.end());
    entries.clear();
    entries.reserve(references.size());
    entries.insert(entries.end(), references.begin(), references.end());
}

} // namespace aapt