mirror of
https://gitlab.gnome.org/jwestman/blueprint-compiler.git
synced 2025-05-04 15:59:08 -04:00
289 lines
9.7 KiB
Python
289 lines
9.7 KiB
Python
# main.py
|
|
#
|
|
# Copyright 2021 James Westman <james@jwestman.net>
|
|
#
|
|
# This file is free software; you can redistribute it and/or modify it
|
|
# under the terms of the GNU Lesser General Public License as
|
|
# published by the Free Software Foundation; either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This file is distributed in the hope that it will be useful, but
|
|
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public
|
|
# License along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
# SPDX-License-Identifier: LGPL-3.0-or-later
|
|
|
|
|
|
import argparse
|
|
import difflib
|
|
import os
|
|
import sys
|
|
import typing as T
|
|
|
|
from . import interactive_port, parser, tokenizer
|
|
from .errors import CompileError, CompilerBugError, PrintableError, report_bug
|
|
from .formatter import Format
|
|
from .gir import add_typelib_search_path
|
|
from .lsp import LanguageServer
|
|
from .outputs import XmlOutput
|
|
from .utils import Colors
|
|
|
|
VERSION = "uninstalled"
|
|
LIBDIR = None
|
|
|
|
|
|
class BlueprintApp:
|
|
def main(self):
|
|
self.parser = argparse.ArgumentParser()
|
|
self.subparsers = self.parser.add_subparsers(metavar="command")
|
|
self.parser.set_defaults(func=self.cmd_help)
|
|
|
|
compile = self.add_subcommand(
|
|
"compile", "Compile blueprint files", self.cmd_compile
|
|
)
|
|
compile.add_argument("--output", dest="output", default="-")
|
|
compile.add_argument("--typelib-path", nargs="?", action="append")
|
|
compile.add_argument(
|
|
"input", metavar="filename", default=sys.stdin, type=argparse.FileType("r")
|
|
)
|
|
|
|
batch_compile = self.add_subcommand(
|
|
"batch-compile",
|
|
"Compile many blueprint files at once",
|
|
self.cmd_batch_compile,
|
|
)
|
|
batch_compile.add_argument("output_dir", metavar="output-dir")
|
|
batch_compile.add_argument("input_dir", metavar="input-dir")
|
|
batch_compile.add_argument("--typelib-path", nargs="?", action="append")
|
|
batch_compile.add_argument(
|
|
"inputs",
|
|
nargs="+",
|
|
metavar="filenames",
|
|
default=sys.stdin,
|
|
type=argparse.FileType("r"),
|
|
)
|
|
|
|
format = self.add_subcommand(
|
|
"format", "Format given blueprint files", self.cmd_format
|
|
)
|
|
format.add_argument(
|
|
"-f",
|
|
"--fix",
|
|
help="Format the files in place, don't output the diff",
|
|
default=False,
|
|
action="store_true",
|
|
)
|
|
format.add_argument(
|
|
"-t",
|
|
"--tabs",
|
|
help="Use tabs instead of spaces",
|
|
default=False,
|
|
action="store_true",
|
|
)
|
|
format.add_argument(
|
|
"-s",
|
|
"--spaces",
|
|
help="How many spaces should be used per indent",
|
|
default=2,
|
|
type=int,
|
|
)
|
|
format.add_argument(
|
|
"inputs",
|
|
nargs="+",
|
|
metavar="filenames",
|
|
)
|
|
|
|
port = self.add_subcommand("port", "Interactive porting tool", self.cmd_port)
|
|
|
|
lsp = self.add_subcommand(
|
|
"lsp", "Run the language server (for internal use by IDEs)", self.cmd_lsp
|
|
)
|
|
|
|
self.add_subcommand("help", "Show this message", self.cmd_help)
|
|
|
|
self.parser.add_argument("--version", action="version", version=VERSION)
|
|
|
|
try:
|
|
opts = self.parser.parse_args()
|
|
opts.func(opts)
|
|
except SystemExit as e:
|
|
raise e
|
|
except KeyboardInterrupt:
|
|
print(f"\n\n{Colors.RED}{Colors.BOLD}Interrupted.{Colors.CLEAR}")
|
|
except EOFError:
|
|
print(f"\n\n{Colors.RED}{Colors.BOLD}Interrupted.{Colors.CLEAR}")
|
|
except:
|
|
report_bug()
|
|
|
|
def add_subcommand(self, name: str, help: str, func):
|
|
parser = self.subparsers.add_parser(name, help=help)
|
|
parser.set_defaults(func=func)
|
|
return parser
|
|
|
|
def cmd_help(self, opts):
|
|
self.parser.print_help()
|
|
|
|
def cmd_compile(self, opts):
|
|
if opts.typelib_path != None:
|
|
for typelib_path in opts.typelib_path:
|
|
add_typelib_search_path(typelib_path)
|
|
|
|
data = opts.input.read()
|
|
try:
|
|
xml, warnings = self._compile(data)
|
|
|
|
for warning in warnings:
|
|
warning.pretty_print(opts.input.name, data, stream=sys.stderr)
|
|
|
|
if opts.output == "-":
|
|
print(xml)
|
|
else:
|
|
with open(opts.output, "w") as file:
|
|
file.write(xml)
|
|
except PrintableError as e:
|
|
e.pretty_print(opts.input.name, data, stream=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
def cmd_batch_compile(self, opts):
|
|
if opts.typelib_path != None:
|
|
for typelib_path in opts.typelib_path:
|
|
add_typelib_search_path(typelib_path)
|
|
|
|
for file in opts.inputs:
|
|
data = file.read()
|
|
file_abs = os.path.abspath(file.name)
|
|
input_dir_abs = os.path.abspath(opts.input_dir)
|
|
|
|
try:
|
|
if not os.path.commonpath([file_abs, input_dir_abs]):
|
|
print(
|
|
f"{Colors.RED}{Colors.BOLD}error: input file '{file.name}' is not in input directory '{opts.input_dir}'{Colors.CLEAR}"
|
|
)
|
|
sys.exit(1)
|
|
|
|
xml, warnings = self._compile(data)
|
|
|
|
for warning in warnings:
|
|
warning.pretty_print(file.name, data, stream=sys.stderr)
|
|
|
|
path = os.path.join(
|
|
opts.output_dir,
|
|
os.path.relpath(
|
|
os.path.splitext(file.name)[0] + ".ui", opts.input_dir
|
|
),
|
|
)
|
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
|
with open(path, "w") as file:
|
|
file.write(xml)
|
|
except PrintableError as e:
|
|
e.pretty_print(file.name, data)
|
|
sys.exit(1)
|
|
|
|
def cmd_format(self, opts):
|
|
input_files = []
|
|
|
|
for path in opts.inputs:
|
|
if os.path.isfile(path):
|
|
input_files.append(path)
|
|
elif os.path.isdir(path):
|
|
for root, subfolders, files in os.walk(path):
|
|
for file in files:
|
|
if file.endswith(".blp"):
|
|
input_files.append(os.path.join(root, file))
|
|
else:
|
|
print(
|
|
f"{Colors.RED}{Colors.BOLD}Could not find file: {path}{Colors.CLEAR}"
|
|
)
|
|
|
|
formatted_files = 0
|
|
|
|
for file in input_files:
|
|
with open(file, "r+") as file:
|
|
data = file.read()
|
|
|
|
try:
|
|
xml, warnings = self._compile(data)
|
|
|
|
for warning in warnings:
|
|
warning.pretty_print(file.name, data, stream=sys.stderr)
|
|
|
|
formatted_str = Format.format(data, opts.spaces, not opts.tabs)
|
|
|
|
if data != formatted_str:
|
|
happened = "Would reformat"
|
|
|
|
if opts.fix:
|
|
file.seek(0)
|
|
file.truncate()
|
|
file.write(formatted_str)
|
|
happened = "Reformatted"
|
|
|
|
a_lines = data.splitlines(keepends=True)
|
|
b_lines = formatted_str.splitlines(keepends=True)
|
|
diff_lines = []
|
|
|
|
for line in difflib.unified_diff(
|
|
a_lines, b_lines, fromfile=file.name, tofile=file.name, n=5
|
|
):
|
|
# Work around https://bugs.python.org/issue2142
|
|
# See:
|
|
# https://www.gnu.org/software/diffutils/manual/html_node/Incomplete-Lines.html
|
|
if line[-1] == "\n":
|
|
diff_lines.append(line)
|
|
else:
|
|
diff_lines.append(line + "\n")
|
|
diff_lines.append("\\ No newline at end of file\n")
|
|
|
|
print(
|
|
f"{''.join(diff_lines)}\n{Colors.BOLD}{happened} {file.name}{Colors.CLEAR}\n"
|
|
)
|
|
|
|
formatted_files += 1
|
|
|
|
except PrintableError as e:
|
|
e.pretty_print(file.name, data, stream=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
left_files = len(input_files) - formatted_files
|
|
|
|
if formatted_files == 0:
|
|
print(f"{Colors.GREEN}{Colors.BOLD}Nothing to do.{Colors.CLEAR}")
|
|
elif opts.fix:
|
|
print(
|
|
f"{Colors.RED}{Colors.BOLD}Reformatted {formatted_files} files, {left_files} were left unchanged.{Colors.CLEAR}"
|
|
)
|
|
else:
|
|
print(
|
|
f"{Colors.RED}{Colors.BOLD}{formatted_files} files would be reformatted, {left_files} would be left unchanged.{Colors.CLEAR}"
|
|
)
|
|
sys.exit(1)
|
|
|
|
def cmd_lsp(self, opts):
|
|
langserv = LanguageServer()
|
|
langserv.run()
|
|
|
|
def cmd_port(self, opts):
|
|
interactive_port.run(opts)
|
|
|
|
def _compile(self, data: str) -> T.Tuple[str, T.List[CompileError]]:
|
|
tokens = tokenizer.tokenize(data)
|
|
ast, errors, warnings = parser.parse(tokens)
|
|
|
|
if errors:
|
|
raise errors
|
|
if ast is None:
|
|
raise CompilerBugError()
|
|
|
|
formatter = XmlOutput()
|
|
|
|
return formatter.emit(ast), warnings
|
|
|
|
|
|
def main(version, libdir):
|
|
global VERSION, LIBDIR
|
|
VERSION, LIBDIR = version, libdir
|
|
BlueprintApp().main()
|