#pragma once

#include <iostream>

#include <vector>

#include <variant>

#include <string>

#include "swift/extractor/trap/TrapLabel.h"

namespace codeql {

class SwiftMangledName {

public:

using Part = std::variant<UntypedTrapLabel, std::string, unsigned>;

explicit operator bool() const { return !value.empty(); }

const std::string& str() const { return value; }

operator std::string_view() const { return value; }

std::string hash() const;

// let's avoid copying as long as we don't need it

SwiftMangledName() = default;

SwiftMangledName(const SwiftMangledName&) = delete;

SwiftMangledName& operator=(const SwiftMangledName&) = delete;

SwiftMangledName(SwiftMangledName&&) = default;

SwiftMangledName& operator=(SwiftMangledName&&) = default;

template <typename... Args>

SwiftMangledName(Args&&... args) {

(operator<<(std::forward<Args>(args)), ...);

}

SwiftMangledName& operator<<(UntypedTrapLabel label) &;

SwiftMangledName& operator<<(unsigned i) &;

template <typename Tag>

SwiftMangledName& operator<<(TrapLabel<Tag> label) & {

return operator<<(static_cast<UntypedTrapLabel>(label));

}

// streaming string-like stuff will add a new part it only if strictly required, otherwise it will

// append to the last part if it is a string

template <typename T>

SwiftMangledName& operator<<(T&& arg) & {

value += arg;

return *this;

}

template <typename T>

SwiftMangledName&& operator<<(T&& arg) && {

return std::move(operator<<(std::forward<T>(arg)));

}

private:

std::string value;

};

} // namespace codeql