mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
528d642fc5
While here, also sort and deduplicate output entries, and stop having the Python script try to The Python script is still messing up the files in some cases, but I haven't tracked it down. Instead, I have a small Swift program that does the same thing more easily.
88 lines
3.0 KiB
C++
88 lines
3.0 KiB
C++
//===--- Edit.cpp - Misc edit utilities -----------------------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2017 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "swift/Basic/Edit.h"
|
|
#include "swift/Basic/SourceManager.h"
|
|
#include <algorithm>
|
|
|
|
using namespace swift;
|
|
|
|
void SourceEdits::addEdit(SourceManager &SM, CharSourceRange Range,
|
|
StringRef Text) {
|
|
SourceLoc Loc = Range.getStart();
|
|
unsigned BufID = SM.findBufferContainingLoc(Loc);
|
|
unsigned Offset = SM.getLocOffsetInBuffer(Loc, BufID);
|
|
unsigned Length = Range.getByteLength();
|
|
StringRef Path = SM.getIdentifierForBuffer(BufID);
|
|
|
|
// NOTE: We cannot store SourceManager here since this logic is used by a
|
|
// DiagnosticConsumer where the SourceManager may not remain valid. This is
|
|
// the case when e.g build swift interfaces, we create a fresh
|
|
// CompilerInstance for a limited scope, but diagnostics are passed outside of
|
|
// it.
|
|
Edits.push_back({Path.str(), Text.str(), Offset, Length});
|
|
}
|
|
|
|
void swift::
|
|
writeEditsInJson(const SourceEdits &AllEdits, llvm::raw_ostream &OS) {
|
|
// Sort the edits so they occur from the last to the first within a given
|
|
// source file. That's the order in which applying non-overlapping edits
|
|
// will succeed.
|
|
std::vector<SourceEdits::Edit> allEdits(AllEdits.getEdits().begin(),
|
|
AllEdits.getEdits().end());
|
|
std::sort(allEdits.begin(), allEdits.end(),
|
|
[&](const SourceEdits::Edit &lhs, const SourceEdits::Edit &rhs) {
|
|
// Sort first based on the path. This keeps the edits for a given
|
|
// file together.
|
|
if (lhs.Path < rhs.Path)
|
|
return true;
|
|
else if (lhs.Path > rhs.Path)
|
|
return false;
|
|
|
|
// Then sort based on offset, with larger offsets coming earlier.
|
|
return lhs.Offset > rhs.Offset;
|
|
});
|
|
|
|
// Remove duplicate edits.
|
|
allEdits.erase(
|
|
std::unique(allEdits.begin(), allEdits.end(),
|
|
[&](const SourceEdits::Edit &lhs, const SourceEdits::Edit &rhs) {
|
|
return lhs.Path == rhs.Path && lhs.Text == rhs.Text &&
|
|
lhs.Offset == rhs.Offset && lhs.Length == rhs.Length;
|
|
}),
|
|
allEdits.end());
|
|
|
|
OS << "[";
|
|
bool first = true;
|
|
for (auto &Edit : allEdits) {
|
|
if (first) {
|
|
first = false;
|
|
} else {
|
|
OS << ",";
|
|
}
|
|
OS << "\n";
|
|
OS << " {\n";
|
|
OS << " \"file\": \"";
|
|
OS.write_escaped(Edit.Path) << "\",\n";
|
|
OS << " \"offset\": " << Edit.Offset << ",\n";
|
|
if (Edit.Length != 0)
|
|
OS << " \"remove\": " << Edit.Length << ",\n";
|
|
if (!Edit.Text.empty()) {
|
|
OS << " \"text\": \"";
|
|
OS.write_escaped(Edit.Text) << "\"\n";
|
|
}
|
|
OS << " }";
|
|
}
|
|
OS << "\n]\n";
|
|
}
|