1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
//===--- 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";
}
|
