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
|
//===--- MigrationState.cpp -----------------------------------------------===//
//
// 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 "swift/Basic/SourceManager.h"
#include "swift/Migrator/MigrationState.h"
#include "llvm/Support/Path.h"
#include "llvm/ADT/SmallString.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Support/raw_ostream.h"
using llvm::StringRef;
using namespace swift;
using namespace swift::migrator;
#pragma mark - MigrationState
std::string MigrationState::getInputText() const {
return SrcMgr.getEntireTextForBuffer(InputBufferID).str();
}
std::string MigrationState::getOutputText() const {
return SrcMgr.getEntireTextForBuffer(OutputBufferID).str();
}
static bool quickDumpText(StringRef OutFilename, StringRef Text) {
std::error_code Error;
llvm::raw_fd_ostream FileOS(OutFilename,
Error, llvm::sys::fs::OF_Text);
if (FileOS.has_error()) {
return true;
}
FileOS << Text;
FileOS.flush();
return FileOS.has_error();
}
bool MigrationState::print(size_t StateNumber, StringRef OutDir) const {
auto Failed = false;
SmallString<256> InputFileStatePath;
llvm::sys::path::append(InputFileStatePath, OutDir);
{
SmallString<256> InputStateFilenameScratch;
llvm::raw_svector_ostream InputStateFilenameOS(InputStateFilenameScratch);
InputStateFilenameOS << StateNumber << '-' << "FixitMigrationState";
InputStateFilenameOS << '-' << "Input";
llvm::sys::path::append(InputFileStatePath, InputStateFilenameOS.str());
llvm::sys::path::replace_extension(InputFileStatePath, ".swift");
}
Failed |= quickDumpText(InputFileStatePath, getInputText());
SmallString<256> OutputFileStatePath;
llvm::sys::path::append(OutputFileStatePath, OutDir);
{
SmallString<256> OutputStateFilenameScratch;
llvm::raw_svector_ostream OutputStateFilenameOS(OutputStateFilenameScratch);
OutputStateFilenameOS << StateNumber << '-' << "FixitMigrationState";
OutputStateFilenameOS << '-' << "Output";
llvm::sys::path::append(OutputFileStatePath, OutputStateFilenameOS.str());
llvm::sys::path::replace_extension(OutputFileStatePath, ".swift");
}
Failed |= quickDumpText(OutputFileStatePath, getOutputText());
return Failed;
}
|
