/*
 * 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.
 */

#ifndef AAPT_NAME_MANGLER_H
#define AAPT_NAME_MANGLER_H

#include "Resource.h"

#include "util/Maybe.h"

#include <set>
#include <string>

namespace aapt {

struct NameManglerPolicy {
    /**
     * Represents the package we are trying to build. References pointing
     * to this package are not mangled, and mangled references inherit this package name.
     */
    std::u16string targetPackageName;

    /**
     * We must know which references to mangle, and which to keep (android vs. com.android.support).
     */
    std::set<std::u16string> packagesToMangle;
};

class NameMangler {
private:
    NameManglerPolicy mPolicy;

public:
    NameMangler(NameManglerPolicy policy) : mPolicy(policy) {
    }

    Maybe<ResourceName> mangleName(const ResourceName& name) {
        if (mPolicy.targetPackageName == name.package ||
                mPolicy.packagesToMangle.count(name.package) == 0) {
            return {};
        }

        return ResourceName{
                mPolicy.targetPackageName,
                name.type,
                mangleEntry(name.package, name.entry)
        };
    }

    bool shouldMangle(const std::u16string& package) const {
        if (package.empty() || mPolicy.targetPackageName == package) {
            return false;
        }
        return mPolicy.packagesToMangle.count(package) != 0;
    }

    /**
     * Returns a mangled name that is a combination of `name` and `package`.
     * The mangled name should contain symbols that are illegal to define in XML,
     * so that there will never be name mangling collisions.
     */
    static std::u16string mangleEntry(const std::u16string& package, const std::u16string& name) {
        return package + u"$" + name;
    }

    /**
     * Unmangles the name in `outName`, storing the correct name back in `outName`
     * and the package in `outPackage`. Returns true if the name was unmangled or
     * false if the name was never mangled to begin with.
     */
    static bool unmangle(std::u16string* outName, std::u16string* outPackage) {
        size_t pivot = outName->find(u'$');
        if (pivot == std::string::npos) {
            return false;
        }

        outPackage->assign(outName->data(), pivot);
        outName->assign(outName->data() + pivot + 1, outName->size() - (pivot + 1));
        return true;
    }
};

} // namespace aapt

#endif // AAPT_NAME_MANGLER_H