/**
 * Copyright (c) 2021-2022 Huawei Device Co., Ltd.
 * 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.
 */

#ifndef LIBPANDAFILE_CLASS_DATA_ACCESSOR_H_
#define LIBPANDAFILE_CLASS_DATA_ACCESSOR_H_

#include "file.h"
#include "file_items.h"
#include "field_data_accessor.h"
#include "method_data_accessor.h"

namespace panda::panda_file {

// NOLINTNEXTLINE(cppcoreguidelines-special-member-functions, hicpp-special-member-functions)
class ClassDataAccessor {
public:
    ClassDataAccessor(const File &panda_file, File::EntityId class_id);

    ~ClassDataAccessor() = default;

    File::EntityId GetSuperClassId() const
    {
        return File::EntityId(super_class_off_);
    }

    bool IsInterface() const
    {
        return (access_flags_ & ACC_INTERFACE) != 0;
    }

    bool IsPublic() const
    {
        return (access_flags_ & ACC_PUBLIC) != 0;
    }

    bool IsProtected() const
    {
        return (access_flags_ & ACC_PROTECTED) != 0;
    }

    bool IsPrivate() const
    {
        return (access_flags_ & ACC_PRIVATE) != 0;
    }

    bool IsFinal() const
    {
        return (access_flags_ & ACC_FINAL) != 0;
    }

    uint32_t GetAccessFlags() const
    {
        return access_flags_;
    }

    uint32_t GetFieldsNumber() const
    {
        return num_fields_;
    }

    uint32_t GetMethodsNumber() const
    {
        return num_methods_;
    }

    uint32_t GetIfacesNumber() const
    {
        return num_ifaces_;
    }

    uint32_t GetAnnotationsNumber();

    uint32_t GetRuntimeAnnotationsNumber();

    File::EntityId GetInterfaceId(size_t idx) const;

    template <class Callback>
    void EnumerateInterfaces(const Callback &cb);

    template <class Callback>
    void EnumerateRuntimeAnnotations(const Callback &cb);

    template <class Callback>
    void EnumerateAnnotations(const Callback &cb);

    template <class Callback>
    void EnumerateTypeAnnotations(const Callback &cb);

    template <class Callback>
    void EnumerateRuntimeTypeAnnotations(const Callback &cb);

    template <class Callback>
    bool EnumerateRuntimeAnnotationsWithEarlyStop(const Callback &cb);

    template <class Callback>
    bool EnumerateAnnotationsWithEarlyStop(const Callback &cb);

    std::optional<SourceLang> GetSourceLang();

    std::optional<File::EntityId> GetSourceFileId();

    template <class Callback>
    void EnumerateFields(const Callback &cb);

    template <class Callback>
    void EnumerateMethods(const Callback &cb);

    size_t GetSize()
    {
        if (size_ == 0) {
            SkipMethods();
        }

        return size_;
    }

    const File &GetPandaFile() const
    {
        return panda_file_;
    }

    File::EntityId GetClassId() const
    {
        return class_id_;
    }

    const uint8_t *GetDescriptor() const
    {
        return name_.data;
    }

    std::string DemangledName()
    {
        const uint8_t *descriptor = GetDescriptor();
        switch (*descriptor) {
            case 'V':
                return "void";
            case 'Z':
                return "u1";
            case 'B':
                return "i8";
            case 'H':
                return "u8";
            case 'S':
                return "i16";
            case 'C':
                return "u16";
            case 'I':
                return "i32";
            case 'U':
                return "u32";
            case 'J':
                return "i64";
            case 'Q':
                return "u64";
            case 'F':
                return "f32";
            case 'D':
                return "f64";
            case 'A':
                return "any";
            default: {
                break;
            }
        }

        std::string name = utf::Mutf8AsCString(descriptor);
        if (name[0] == '[') {
            return name;
        }

        std::replace(name.begin(), name.end(), '/', '.');

        ASSERT(name.size() > 2);  // 2 - L and ;

        name.erase(0, 1);
        name.pop_back();

        return name;
    }

private:
    void SkipSourceLang();

    void SkipRuntimeAnnotations();

    void SkipAnnotations();

    void SkipSourceFile();

    void SkipFields();

    void SkipMethods();

    void SkipRuntimeTypeAnnotations();

    void SkipTypeAnnotations();

    const File &panda_file_;
    File::EntityId class_id_;

    File::StringData name_ {};
    uint32_t super_class_off_;
    uint32_t access_flags_;
    uint32_t num_fields_ {0};
    uint32_t num_methods_ {0};
    uint32_t num_ifaces_ {0};

    Span<const uint8_t> ifaces_offsets_sp_ {nullptr, nullptr};
    Span<const uint8_t> source_lang_sp_ {nullptr, nullptr};
    Span<const uint8_t> runtime_annotations_sp_ {nullptr, nullptr};
    Span<const uint8_t> annotations_sp_ {nullptr, nullptr};
    Span<const uint8_t> runtime_type_annotation_sp_ {nullptr, nullptr};
    Span<const uint8_t> type_annotation_sp_ {nullptr, nullptr};
    Span<const uint8_t> source_file_sp_ {nullptr, nullptr};
    Span<const uint8_t> fields_sp_ {nullptr, nullptr};
    Span<const uint8_t> methods_sp_ {nullptr, nullptr};

    size_t size_ {0};
};

}  // namespace panda::panda_file

#endif  // LIBPANDAFILE_CLASS_DATA_ACCESSOR_H_