Files
codeql/swift/extractor/infra/SwiftMangledName.h
Paolo Tranquilli 70ff401f21 Swift: replace internal swift mangler with our own
Our mangler is split in two version:
* `SwiftTrapMangler`, with the same behaviour as the previous
  `SwiftMangler`, constructing mangled names with trap label references
* `SwiftRecursiveMangler` that replaces trap label references with
  recursive calls to its own `mangle` functions, effectively rolling out
  the entire chain of references

The latter is used to create lazy trap file names. Hashing is used to
avoid excessively long filenames.
2023-05-31 09:52:20 +02:00

61 lines
1.6 KiB
C++

#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