/*
 * Copyright 2020 Google LLC
 *
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */

#ifndef SKSL_BUILTINMAP
#define SKSL_BUILTINMAP

#include "include/private/SkSLProgramElement.h"
#include "include/private/SkTHash.h"

#include <functional>
#include <memory>
#include <string>

namespace SkSL {

/**
 * Represents the builtin elements in the Context.
 */
class BuiltinMap {
public:
    BuiltinMap(BuiltinMap* parent) : fParent(parent) {}

    void insertOrDie(std::string key, std::unique_ptr<ProgramElement> element);

    const ProgramElement* find(const std::string& key);

    const ProgramElement* findAndInclude(const std::string& key);

    void resetAlreadyIncluded();

    void foreach(const std::function<void(const std::string&, const ProgramElement&)>& fn) const;

private:
    struct BuiltinElement {
        std::unique_ptr<ProgramElement> fElement;
        bool fAlreadyIncluded = false;
    };

    SkTHashMap<std::string, BuiltinElement> fElements;
    BuiltinMap* fParent = nullptr;
};

} // namespace SkSL

#endif