mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
9779c18da3
LLVM is presumably moving towards `std::string_view` - `StringRef::startswith` is deprecated on tip. `SmallString::startswith` was just renamed there (maybe with some small deprecation inbetween, but if so, we've missed it). The `SmallString::startswith` references were moved to `.str().starts_with()`, rather than adding the `starts_with` on `stable/20230725` as we only had a few of them. Open to switching that over if anyone feels strongly though.
96 lines
2.9 KiB
C++
96 lines
2.9 KiB
C++
//===--- ASTScriptConfiguration.cpp ---------------------------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2019 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
///
|
|
/// Configuration parsing for an AST script.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "ASTScriptConfiguration.h"
|
|
|
|
#include "swift/Basic/QuotedString.h"
|
|
#include "llvm/ADT/ArrayRef.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace swift;
|
|
using namespace scripting;
|
|
|
|
std::unique_ptr<ASTScriptConfiguration>
|
|
ASTScriptConfiguration::parse(CompilerInstance &compiler,
|
|
ArrayRef<const char *> args) {
|
|
bool hadError = false;
|
|
|
|
std::unique_ptr<ASTScriptConfiguration> result(
|
|
new ASTScriptConfiguration(compiler));
|
|
|
|
#define emitError(WHAT) do { \
|
|
llvm::errs() << "error: " << WHAT << "\n"; \
|
|
hadError = true; \
|
|
} while (0)
|
|
|
|
auto popArg = [&](const char *what) -> StringRef {
|
|
// Gracefully handle the case of running out of arguments.
|
|
if (args.empty()) {
|
|
assert(what && "expected explanation here!");
|
|
emitError(what);
|
|
return "";
|
|
}
|
|
|
|
auto arg = args.front();
|
|
args = args.slice(1);
|
|
return arg;
|
|
};
|
|
|
|
auto setScriptFile = [&](StringRef filename) {
|
|
if (!result->ScriptFile.empty()) {
|
|
emitError("multiple script files ("
|
|
<< QuotedString(result->ScriptFile)
|
|
<< ", "
|
|
<< QuotedString(filename)
|
|
<< ")");
|
|
}
|
|
result->ScriptFile = filename;
|
|
};
|
|
|
|
// Parse the arguments.
|
|
while (!hadError && !args.empty()) {
|
|
StringRef arg = popArg(nullptr);
|
|
if (!arg.starts_with("-")) {
|
|
setScriptFile(arg);
|
|
} else if (arg == "-f") {
|
|
StringRef filename = popArg("expected path after -f");
|
|
if (!hadError)
|
|
setScriptFile(filename);
|
|
} else if (arg == "-h" || arg == "--help") {
|
|
llvm::errs()
|
|
<< "usage: swift-ast-script <script-args> -- <compiler-args\n"
|
|
"accepted script arguments:\n"
|
|
" <filename>\n"
|
|
" -f <filename>\n"
|
|
" Specify the file to use as the script; required argument\n";
|
|
hadError = true;
|
|
} else {
|
|
emitError("unknown argument " << QuotedString(arg));
|
|
}
|
|
}
|
|
|
|
// Check well-formedness.
|
|
if (!hadError) {
|
|
if (result->ScriptFile.empty()) {
|
|
emitError("script file is required");
|
|
}
|
|
}
|
|
|
|
if (hadError)
|
|
result.reset();
|
|
return result;
|
|
}
|