wurstscript / WurstScript / 228

Committed 29 Nov 2023 05:00PM UTC coverage: 62.48% (-0.09%) from 62.574%

Build # 228

Build Type

push

circleci

Committed by

web-flow

Commit Message

Show dialog for choosing game path, cleanup (#1083)

* show dialog for choosing game path

* cleanup code

* remove logs and refactor

* remove confusing mpq error, make some mpq loads readonly

Run Details

17295 of 27681 relevant lines covered (62.48%)

0.62 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

de.peeeq.wurstscript/src/main/java/de/peeeq/wurstio/Main.java

package de.peeeq.wurstio;

import com.google.common.base.Charsets;
import com.google.common.io.Files;
import config.WurstProjectConfig;
import config.WurstProjectConfigData;
import de.peeeq.wurstio.compilationserver.WurstServer;
import de.peeeq.wurstio.gui.AboutDialog;
import de.peeeq.wurstio.gui.WurstGuiImpl;
import de.peeeq.wurstio.hotdoc.HotdocGenerator;
import de.peeeq.wurstio.languageserver.LanguageServerStarter;
import de.peeeq.wurstio.languageserver.ProjectConfigBuilder;
import de.peeeq.wurstio.languageserver.WFile;
import de.peeeq.wurstio.map.importer.ImportFile;
import de.peeeq.wurstio.mpq.MpqEditor;
import de.peeeq.wurstio.mpq.MpqEditorFactory;
import de.peeeq.wurstio.utils.W3InstallationData;
import de.peeeq.wurstscript.CompileTimeInfo;
import de.peeeq.wurstscript.ErrorReporting;
import de.peeeq.wurstscript.RunArgs;
import de.peeeq.wurstscript.WLogger;
import de.peeeq.wurstscript.attributes.CompileError;
import de.peeeq.wurstscript.attributes.prettyPrint.PrettyUtils;
import de.peeeq.wurstscript.gui.WurstGui;
import de.peeeq.wurstscript.gui.WurstGuiCliImpl;
import de.peeeq.wurstscript.utils.Utils;
import org.eclipse.jdt.annotation.Nullable;

import javax.swing.*;
import java.io.File;
import java.lang.management.ManagementFactory;
import java.lang.management.RuntimeMXBean;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.StandardCopyOption;
import java.util.LinkedList;
import java.util.List;
import java.util.Optional;

import static de.peeeq.wurstio.languageserver.ProjectConfigBuilder.FILE_NAME;
import static de.peeeq.wurstio.languageserver.WurstCommands.getCompileArgs;
import static java.util.Arrays.asList;

public class Main {

    /**
     * @param args
     */
    public static void main(String[] args) {
        if (args.length == 0) {
            // If no args are passed, display help
            new RunArgs("-help");
            return;
        }

        WurstGui gui = null;
        RunArgs runArgs = new RunArgs(args);
        try {
            if (runArgs.isLanguageServer()) {
                WLogger.setLogger("languageServer");
            }
            logStartup(args);

            if (runArgs.showHelp()) {
                return;
            }

            if (runArgs.isShowVersion()) {
                System.out.println(CompileTimeInfo.version);
                return;
            }

            if (runArgs.showAbout()) {
                new AboutDialog(null, false).setVisible(true);
                return;
            }

            if (runArgs.isStartServer()) {
                WurstServer.startServer();
                return;
            }

            if (runArgs.isLanguageServer()) {
                LanguageServerStarter.start();
                return;
            }

            if (runArgs.isPrettyPrint()) {
                PrettyUtils.pretty(runArgs.getFiles());
                return;
            }

            WLogger.info("runArgs.isExtractImports() = " + runArgs.isExtractImports());
            if (runArgs.isExtractImports()) {
                File mapFile = new File(runArgs.getMapFile());
                ImportFile.extractImportsFromMap(mapFile, runArgs);
                return;
            }

            if (runArgs.createHotDoc()) {
                HotdocGenerator hg = new HotdocGenerator(runArgs.getFiles());
                hg.generateDoc();
                return;
            }

            if (runArgs.isGui()) {
                gui = new WurstGuiImpl();
                // use the error reporting with GUI
                ErrorReporting.instance = new ErrorReportingIO();
            } else {
                gui = new WurstGuiCliImpl();
            }

            if (runArgs.showLastErrors()) {
                JOptionPane.showMessageDialog(null, "not implemented");
                return;
            }

            try {
                WurstProjectConfigData projectConfig = null;
                Path buildDir = null;
                Optional<Path> target = Optional.empty();
                String workspaceroot = runArgs.getWorkspaceroot();
                if (runArgs.isBuild() && runArgs.getInputmap() != null && workspaceroot != null) {
                    Path root = Paths.get(workspaceroot);
                    Path inputMap = root.resolve(runArgs.getInputmap());
                    projectConfig = WurstProjectConfig.INSTANCE.loadProject(root.resolve(FILE_NAME));

                    if (java.nio.file.Files.exists(inputMap) && projectConfig != null) {
                        buildDir = root.resolve("_build");
                        java.nio.file.Files.createDirectories(buildDir);
                        target = Optional.of(buildDir.resolve(projectConfig.getBuildMapData().getFileName() + ".w3x"));
                        java.nio.file.Files.copy(inputMap, target.get(), StandardCopyOption.REPLACE_EXISTING);
                        runArgs.setMapFile(target.get().toAbsolutePath().toString());
                    }
                }

                String mapFilePath = runArgs.getMapFile();

                RunArgs compileArgs = runArgs;
                if (workspaceroot != null) {
                    WLogger.info("workspaceroot: " + workspaceroot);
                    List<String> argList = new LinkedList<>(asList(args));
                    List<String> argsList = getCompileArgs(WFile.create(workspaceroot));
                    WLogger.info("workspaceroot: " + (argsList == null));
                    argList.addAll(argsList);
                    compileArgs = new RunArgs(argList);
                }
                CompilationProcess compilationProcess = new CompilationProcess(gui, compileArgs);
                @Nullable CharSequence compiledScript;

                if (mapFilePath != null && workspaceroot != null) {
                    try (MpqEditor mpqEditor = MpqEditorFactory.getEditor(Optional.of(new File(mapFilePath)))) {
                        File projectFolder = Paths.get(workspaceroot).toFile();
                        compiledScript = compilationProcess.doCompilation(mpqEditor, projectFolder, true);
                        if (compiledScript != null) {
                            gui.sendProgress("Writing to map");
                            mpqEditor.deleteFile("war3map.j");
                            byte[] war3map = compiledScript.toString().getBytes(Charsets.UTF_8);
                            mpqEditor.insertFile("war3map.j", war3map);
                        }
                        ImportFile.importFilesFromImports(projectFolder, mpqEditor);
                    }
                } else {
                    compiledScript = compilationProcess.doCompilation(null, true);
                }

                if (compiledScript != null) {
                    File scriptFile = new File("compiled.j.txt");
                    Files.write(compiledScript.toString().getBytes(Charsets.UTF_8), scriptFile);

                    if (projectConfig != null && target.isPresent()) {
                        ProjectConfigBuilder.apply(projectConfig, target.get().toFile(), scriptFile, buildDir.toFile(),
                            runArgs, new W3InstallationData(null, Paths.get(workspaceroot).toFile()));

                        WLogger.info("map build success");
                        System.out.println("Build succeeded. Output file: <" + target.get().toAbsolutePath() + ">");
                    }
                }

                gui.sendProgress("Finished!");
            } catch (AbortCompilationException e) {
                gui.showInfoMessage(e.getMessage());
            }
        } catch (Throwable t) {
            String source = "";
            // TODO add additional information to source
            ErrorReporting.instance.handleSevere(t, source);
            if (!runArgs.isGui()) {
                System.exit(2);
            }
        } finally {
            if (gui != null) {
                gui.sendFinished();
                if (!runArgs.isGui()) {
                    if (gui.getErrorCount() > 0) {
                        //         print error messages
                        for (CompileError err : gui.getErrorList()) {
                            System.out.println(err);
                        }
                        // signal that there was an error when compiling
                        System.exit(1);
                    }
                    // print warnings:
                    for (CompileError err : gui.getWarningList()) {
                        System.out.println(err);
                    }
                }
            }
        }
    }

    private static void logStartup(String[] args) {
        // VM Arguments
        RuntimeMXBean runtimeMxBean = ManagementFactory.getRuntimeMXBean();
        List<String> arguments = runtimeMxBean.getInputArguments();

        WLogger.info("### Started wurst version: (" + AboutDialog.version + ")");
        WLogger.info("### With wurst-args " + Utils.printSep(", ", args));
        if (arguments != null && arguments.size() > 0) {
            WLogger.info("### With vm-args " + Utils.printSep(", ", arguments.toArray(new String[0])));
        }
        try {
            WLogger.info("### compiler path1: " + Main.class.getProtectionDomain().getCodeSource().getLocation());
            WLogger.info("### compiler path2: " + ClassLoader.getSystemClassLoader().getResource(".").getPath());
        } catch (Throwable ignored) {
        }
        WLogger.info("### ============================================");
    }

}

1	package de.peeeq.wurstio;
2
3	import com.google.common.base.Charsets;
4	import com.google.common.io.Files;
5	import config.WurstProjectConfig;
6	import config.WurstProjectConfigData;
7	import de.peeeq.wurstio.compilationserver.WurstServer;
8	import de.peeeq.wurstio.gui.AboutDialog;
9	import de.peeeq.wurstio.gui.WurstGuiImpl;
10	import de.peeeq.wurstio.hotdoc.HotdocGenerator;
11	import de.peeeq.wurstio.languageserver.LanguageServerStarter;
12	import de.peeeq.wurstio.languageserver.ProjectConfigBuilder;
13	import de.peeeq.wurstio.languageserver.WFile;
14	import de.peeeq.wurstio.map.importer.ImportFile;
15	import de.peeeq.wurstio.mpq.MpqEditor;
16	import de.peeeq.wurstio.mpq.MpqEditorFactory;
17	import de.peeeq.wurstio.utils.W3InstallationData;
18	import de.peeeq.wurstscript.CompileTimeInfo;
19	import de.peeeq.wurstscript.ErrorReporting;
20	import de.peeeq.wurstscript.RunArgs;
21	import de.peeeq.wurstscript.WLogger;
22	import de.peeeq.wurstscript.attributes.CompileError;
23	import de.peeeq.wurstscript.attributes.prettyPrint.PrettyUtils;
24	import de.peeeq.wurstscript.gui.WurstGui;
25	import de.peeeq.wurstscript.gui.WurstGuiCliImpl;
26	import de.peeeq.wurstscript.utils.Utils;
27	import org.eclipse.jdt.annotation.Nullable;
28
29	import javax.swing.*;
30	import java.io.File;
31	import java.lang.management.ManagementFactory;
32	import java.lang.management.RuntimeMXBean;
33	import java.nio.file.Path;
34	import java.nio.file.Paths;
35	import java.nio.file.StandardCopyOption;
36	import java.util.LinkedList;
37	import java.util.List;
38	import java.util.Optional;
39
40	import static de.peeeq.wurstio.languageserver.ProjectConfigBuilder.FILE_NAME;
41	import static de.peeeq.wurstio.languageserver.WurstCommands.getCompileArgs;
42	import static java.util.Arrays.asList;
43
44	public class Main {	×
45
46	/**
47	* @param args
48	*/
49	public static void main(String[] args) {
50	if (args.length == 0) {	×
51	// If no args are passed, display help
52	new RunArgs("-help");	×
53	return;	×
54	}
55
56	WurstGui gui = null;	×
57	RunArgs runArgs = new RunArgs(args);	×
58	try {
59	if (runArgs.isLanguageServer()) {	×
60	WLogger.setLogger("languageServer");	×
61	}
62	logStartup(args);	×
63
64	if (runArgs.showHelp()) {	×
65	return;	×
66	}
67
68	if (runArgs.isShowVersion()) {	×
69	System.out.println(CompileTimeInfo.version);	×
70	return;	×
71	}
72
73	if (runArgs.showAbout()) {	×
74	new AboutDialog(null, false).setVisible(true);	×
75	return;	×
76	}
77
78	if (runArgs.isStartServer()) {	×
79	WurstServer.startServer();	×
80	return;	×
81	}
82
83	if (runArgs.isLanguageServer()) {	×
84	LanguageServerStarter.start();	×
85	return;	×
86	}
87
88	if (runArgs.isPrettyPrint()) {	×
89	PrettyUtils.pretty(runArgs.getFiles());	×
90	return;	×
91	}
92
93	WLogger.info("runArgs.isExtractImports() = " + runArgs.isExtractImports());	×
94	if (runArgs.isExtractImports()) {	×
95	File mapFile = new File(runArgs.getMapFile());	×
96	ImportFile.extractImportsFromMap(mapFile, runArgs);	×
97	return;	×
98	}
99
100	if (runArgs.createHotDoc()) {	×
101	HotdocGenerator hg = new HotdocGenerator(runArgs.getFiles());	×
102	hg.generateDoc();	×
103	return;	×
104	}
105
106	if (runArgs.isGui()) {	×
107	gui = new WurstGuiImpl();	×
108	// use the error reporting with GUI
109	ErrorReporting.instance = new ErrorReportingIO();	×
110	} else {
111	gui = new WurstGuiCliImpl();	×
112	}
113
114	if (runArgs.showLastErrors()) {	×
115	JOptionPane.showMessageDialog(null, "not implemented");	×
116	return;	×
117	}
118
119	try {
120	WurstProjectConfigData projectConfig = null;	×
121	Path buildDir = null;	×
122	Optional<Path> target = Optional.empty();	×
123	String workspaceroot = runArgs.getWorkspaceroot();	×
124	if (runArgs.isBuild() && runArgs.getInputmap() != null && workspaceroot != null) {	×
125	Path root = Paths.get(workspaceroot);	×
126	Path inputMap = root.resolve(runArgs.getInputmap());	×
127	projectConfig = WurstProjectConfig.INSTANCE.loadProject(root.resolve(FILE_NAME));	×
128
129	if (java.nio.file.Files.exists(inputMap) && projectConfig != null) {	×
130	buildDir = root.resolve("_build");	×
131	java.nio.file.Files.createDirectories(buildDir);	×
132	target = Optional.of(buildDir.resolve(projectConfig.getBuildMapData().getFileName() + ".w3x"));	×
133	java.nio.file.Files.copy(inputMap, target.get(), StandardCopyOption.REPLACE_EXISTING);	×
134	runArgs.setMapFile(target.get().toAbsolutePath().toString());	×
135	}
136	}
137
138	String mapFilePath = runArgs.getMapFile();	×
139
140	RunArgs compileArgs = runArgs;	×
141	if (workspaceroot != null) {	×
142	WLogger.info("workspaceroot: " + workspaceroot);	×
143	List<String> argList = new LinkedList<>(asList(args));	×
144	List<String> argsList = getCompileArgs(WFile.create(workspaceroot));	×
145	WLogger.info("workspaceroot: " + (argsList == null));	×
146	argList.addAll(argsList);	×
147	compileArgs = new RunArgs(argList);	×
148	}
149	CompilationProcess compilationProcess = new CompilationProcess(gui, compileArgs);	×
150	@Nullable CharSequence compiledScript;
151
152	if (mapFilePath != null && workspaceroot != null) {	×
153	try (MpqEditor mpqEditor = MpqEditorFactory.getEditor(Optional.of(new File(mapFilePath)))) {	×
154	File projectFolder = Paths.get(workspaceroot).toFile();	×
155	compiledScript = compilationProcess.doCompilation(mpqEditor, projectFolder, true);	×
156	if (compiledScript != null) {	×
157	gui.sendProgress("Writing to map");	×
158	mpqEditor.deleteFile("war3map.j");	×
159	byte[] war3map = compiledScript.toString().getBytes(Charsets.UTF_8);	×
160	mpqEditor.insertFile("war3map.j", war3map);	×
161	}
162	ImportFile.importFilesFromImports(projectFolder, mpqEditor);	×
163	}
164	} else {
165	compiledScript = compilationProcess.doCompilation(null, true);	×
166	}
167
168	if (compiledScript != null) {	×
169	File scriptFile = new File("compiled.j.txt");	×
170	Files.write(compiledScript.toString().getBytes(Charsets.UTF_8), scriptFile);	×
171
172	if (projectConfig != null && target.isPresent()) {	×
173	ProjectConfigBuilder.apply(projectConfig, target.get().toFile(), scriptFile, buildDir.toFile(),	×
174	runArgs, new W3InstallationData(null, Paths.get(workspaceroot).toFile()));	×
175
176	WLogger.info("map build success");	×
177	System.out.println("Build succeeded. Output file: <" + target.get().toAbsolutePath() + ">");	×
178	}
179	}
180
181	gui.sendProgress("Finished!");	×
182	} catch (AbortCompilationException e) {	×
183	gui.showInfoMessage(e.getMessage());	×
184	}	×
185	} catch (Throwable t) {	×
186	String source = "";	×
187	// TODO add additional information to source
188	ErrorReporting.instance.handleSevere(t, source);	×
189	if (!runArgs.isGui()) {	×
190	System.exit(2);	×
191	}
192	} finally {
193	if (gui != null) {	×
194	gui.sendFinished();	×
195	if (!runArgs.isGui()) {	×
196	if (gui.getErrorCount() > 0) {	×
197	// print error messages
198	for (CompileError err : gui.getErrorList()) {	×
199	System.out.println(err);	×
200	}	×
201	// signal that there was an error when compiling
202	System.exit(1);	×
203	}
204	// print warnings:
205	for (CompileError err : gui.getWarningList()) {	×
206	System.out.println(err);	×
207	}	×
208	}
209	}
210	}
211	}	×
212
213	private static void logStartup(String[] args) {
214	// VM Arguments
215	RuntimeMXBean runtimeMxBean = ManagementFactory.getRuntimeMXBean();	×
216	List<String> arguments = runtimeMxBean.getInputArguments();	×
217
218	WLogger.info("### Started wurst version: (" + AboutDialog.version + ")");	×
219	WLogger.info("### With wurst-args " + Utils.printSep(", ", args));	×
220	if (arguments != null && arguments.size() > 0) {	×
221	WLogger.info("### With vm-args " + Utils.printSep(", ", arguments.toArray(new String[0])));	×
222	}
223	try {
224	WLogger.info("### compiler path1: " + Main.class.getProtectionDomain().getCodeSource().getLocation());	×
225	WLogger.info("### compiler path2: " + ClassLoader.getSystemClassLoader().getResource(".").getPath());	×
226	} catch (Throwable ignored) {	×
227	}	×
228	WLogger.info("### ============================================");	×
229	}	×
230
231	}

wurstscript / WurstScript / 228

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous