uniquer.h

// Copyright (c) 2020-now by the Zeek Project. See LICENSE for details.
 
#pragma once
 
#include <hilti/base/cache.h>
#include <hilti/base/util.h>
 
namespace hilti::util {
 
template<typename ID>
class Uniquer : private Cache<ID, bool> {
public:
    ID get(ID name, bool normalize = true) {
        if ( normalize )
            name = ID(util::toIdentifier(name));
 
        auto x = name;
        int i = 1;
        while ( true ) {
            if ( ! this->has(x) ) {
                this->put(x, true);
                return x;
            }
 
            x = ID(util::fmt("%s_%d", name, ++i));
        }
    }
 
    void remove(const ID& id) { this->Cache<ID, bool>::remove(id); }
};
 
} // namespace hilti::util