Files
carbon-lang/migrate_cpp/migrate_cpp.py
T
Jon Meow 2344a52e24 Start handling variable declarations (#571)
TODOs in the test for known issues.

I may switch the approach to getting the variable type (on examination, this isn't working quite as well as I'd thought) but for now I think it's okay. I had an earlier approach though that may work better overall -- I'd been thinking this would work better, but as you can see in the null check for type information, I think I missed a key point.

Anyways, what'd really been vexing me was `int i, j` which I think I handle passably well now. There's obviously room for improvement, but given I've been going at this for a couple days now, I thought it best to checkpoint where I was.

This also includes some related framework changes to fix bumps I was running into. Overall the tool should operate a bit more smoothly with these changes. There are still issues with overlapping replacements, but I think it's primarily with range-based for loops which I just need to take some time to fix.
2021-06-23 10:58:32 -07:00

112 lines
3.6 KiB
Python

"""Migrates C++ code to Carbon."""
__copyright__ = """
Part of the Carbon Language project, under the Apache License v2.0 with LLVM
Exceptions. See /LICENSE for license information.
SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
"""
import argparse
import glob
import os
import subprocess
import sys
_CPP_REFACTORING = "./cpp_refactoring/cpp_refactoring"
_H_EXTS = {".h", ".hpp"}
_CPP_EXTS = {".c", ".cc", ".cpp", ".cxx"}
class _Workflow(object):
def __init__(self):
"""Parses command-line arguments and flags."""
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument(
"dir",
type=str,
help="A directory containing C++ files to migrate to Carbon.",
)
parsed_args = parser.parse_args()
self._parsed_args = parsed_args
self._data_dir = os.path.dirname(sys.argv[0])
# Validate arguments.
if not os.path.isdir(parsed_args.dir):
sys.exit("%r must point to a directory." % parsed_args.dir)
def run(self):
"""Runs the migration workflow."""
try:
self._gather_files()
self._clang_tidy()
self._cpp_refactoring()
self._rename_files()
self._print_header("Done!")
except subprocess.CalledProcessError as e:
# Discard the stack for subprocess errors.
sys.exit(e)
def _data_file(self, relative_path):
"""Returns the path to a data file."""
return os.path.join(self._data_dir, relative_path)
@staticmethod
def _print_header(header):
print("*" * 79)
print("* %-75s *" % header)
print("*" * 79)
def _gather_files(self):
"""Returns the list of C++ files to convert."""
self._print_header("Gathering C++ files...")
all_files = glob.glob(
os.path.join(self._parsed_args.dir, "**/*.*"), recursive=True
)
exts = _CPP_EXTS.union(_H_EXTS)
cpp_files = [f for f in all_files if os.path.splitext(f)[1] in exts]
if not cpp_files:
sys.exit(
"%r doesn't contain any C++ files to convert."
% self._parsed_args.dir
)
self._cpp_files = sorted(cpp_files)
print("%d files found." % len(self._cpp_files))
def _clang_tidy(self):
"""Runs clang-tidy to fix C++ files in a directory."""
self._print_header("Running clang-tidy...")
with open(self._data_file("clang_tidy.yaml")) as f:
config = f.read()
subprocess.run(
["clang-tidy", "--fix", "--config", config] + self._cpp_files,
check=True,
)
def _cpp_refactoring(self):
"""Runs cpp_refactoring to migrate C++ files towards Carbon syntax."""
self._print_header("Running cpp_refactoring...")
cpp_refactoring = self._data_file(_CPP_REFACTORING)
subprocess.run([cpp_refactoring] + self._cpp_files, check=True)
def _rename_files(self):
"""Renames C++ files to the destination Carbon filenames."""
api_renames = 0
impl_renames = 0
for f in self._cpp_files:
parts = os.path.splitext(f)
if parts[1] in _H_EXTS:
os.rename(f, parts[0] + ".carbon")
api_renames += 1
else:
os.rename(f, parts[0] + ".impl.carbon")
impl_renames += 1
print(
"Renaming resulted in %d API files and %d impl files."
% (api_renames, impl_renames)
)
if __name__ == "__main__":
_Workflow().run()