dcf4f77fac
No real need for keeping a separate header for such a simple class. This requires changing a bit `OrSuggestions<T>::operator*` to not throw an `Error` to prevent a cyclic dependency. But since this error is only thrown on programmer error, we can replace the whole method by a direct call to `std::get` which will raise its own assertion if needs be.
103 lines
1.9 KiB
C++
103 lines
1.9 KiB
C++
#pragma once
|
|
|
|
#include "comparator.hh"
|
|
#include "types.hh"
|
|
#include <set>
|
|
|
|
namespace nix {
|
|
|
|
int levenshteinDistance(std::string_view first, std::string_view second);
|
|
|
|
/**
|
|
* A potential suggestion for the cli interface.
|
|
*/
|
|
class Suggestion {
|
|
public:
|
|
int distance; // The smaller the better
|
|
std::string suggestion;
|
|
|
|
std::string to_string() const;
|
|
|
|
GENERATE_CMP(Suggestion, me->distance, me->suggestion)
|
|
};
|
|
|
|
class Suggestions {
|
|
public:
|
|
std::set<Suggestion> suggestions;
|
|
|
|
std::string to_string() const;
|
|
|
|
Suggestions trim(
|
|
int limit = 5,
|
|
int maxDistance = 2
|
|
) const;
|
|
|
|
static Suggestions bestMatches (
|
|
std::set<std::string> allMatches,
|
|
std::string query
|
|
);
|
|
|
|
Suggestions& operator+=(const Suggestions & other);
|
|
};
|
|
|
|
std::ostream & operator<<(std::ostream & str, const Suggestion &);
|
|
std::ostream & operator<<(std::ostream & str, const Suggestions &);
|
|
|
|
// Either a value of type `T`, or some suggestions
|
|
template<typename T>
|
|
class OrSuggestions {
|
|
public:
|
|
using Raw = std::variant<T, Suggestions>;
|
|
|
|
Raw raw;
|
|
|
|
T* operator ->()
|
|
{
|
|
return &**this;
|
|
}
|
|
|
|
T& operator *()
|
|
{
|
|
return std::get<T>(raw);
|
|
}
|
|
|
|
operator bool() const noexcept
|
|
{
|
|
return std::holds_alternative<T>(raw);
|
|
}
|
|
|
|
OrSuggestions(T t)
|
|
: raw(t)
|
|
{
|
|
}
|
|
|
|
OrSuggestions()
|
|
: raw(Suggestions{})
|
|
{
|
|
}
|
|
|
|
static OrSuggestions<T> failed(const Suggestions & s)
|
|
{
|
|
auto res = OrSuggestions<T>();
|
|
res.raw = s;
|
|
return res;
|
|
}
|
|
|
|
static OrSuggestions<T> failed()
|
|
{
|
|
return OrSuggestions<T>::failed(Suggestions{});
|
|
}
|
|
|
|
const Suggestions & getSuggestions()
|
|
{
|
|
static Suggestions noSuggestions;
|
|
if (const auto & suggestions = std::get_if<Suggestions>(&raw))
|
|
return *suggestions;
|
|
else
|
|
return noSuggestions;
|
|
}
|
|
|
|
};
|
|
|
|
}
|