src/main/java/com/l2jserver/gameserver/scripting/ScriptEngineManager.java - develop - tofu/l2interlude

src/main/java/com/l2jserver/gameserver/scripting/ScriptEngineManager.java
221 строка7 KB
Zoey76
01 янв 2025, 07:57
01 янв 2025, 07:5716130a5
/*
 * Copyright © 2004-2025 L2J Server
 * 
 * This file is part of L2J Server.
 * 
 * L2J Server is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * L2J Server 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
 * General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package com.l2jserver.gameserver.scripting;

import static com.l2jserver.gameserver.config.Configuration.general;
import static com.l2jserver.gameserver.config.Configuration.server;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.LineNumberReader;
import java.io.Reader;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;

import javax.script.ScriptException;

import org.mdkt.compiler.InMemoryJavaCompiler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Script engine manager.
 * @author KenM
 * @author Zoey76
 */
public final class ScriptEngineManager {
	private static final Logger LOG = LoggerFactory.getLogger(ScriptEngineManager.class);
	
	private static final String CLASS_PATH = server().getScriptRoot().getAbsolutePath() + System.getProperty("path.separator") + System.getProperty("java.class.path");
	
	private static final String MAIN = "main";
	
	private static final Object[] MAIN_METHOD_ARGS = new Object[] {
		new String[0]
	};
	
	private static final Class<?>[] ARG_MAIN = new Class[] {
		String[].class
	};
	
	private InMemoryJavaCompiler compiler() {
		return InMemoryJavaCompiler.newInstance() //
			.useOptions("-classpath", CLASS_PATH, "-g") //
			.ignoreWarnings();
	}
	
	public void executeScriptList(File list) throws Exception {
		if (general().noQuests()) {
			return;
		}
		
		if (!list.isFile()) {
			throw new IllegalArgumentException("Argument must be an file containing a list of scripts to be loaded");
		}
		
		final var compiler = compiler();
		try (var fis = new FileInputStream(list);
			var isr = new InputStreamReader(fis);
			var lnr = new LineNumberReader(isr)) {
			String line;
			while ((line = lnr.readLine()) != null) {
				final var parts = line.trim().split("#");
				if ((parts.length <= 0) || parts[0].trim().isEmpty() || (parts[0].charAt(0) == '#')) {
					continue;
				}
				
				line = parts[0].trim();
				if (line.endsWith("/**")) {
					line = line.substring(0, line.length() - 3);
				} else if (line.endsWith("/*")) {
					line = line.substring(0, line.length() - 2);
				}
				
				final var file = new File(server().getScriptRoot(), line);
				if (file.isDirectory() && parts[0].endsWith("/**")) {
					executeAllScriptsInDirectory(compiler, file, true);
				} else if (file.isDirectory() && parts[0].endsWith("/*")) {
					executeAllScriptsInDirectory(compiler, file, false);
				} else if (file.isFile()) {
					addSource(compiler, file);
				} else {
					LOG.warn("Failed loading: ({}) @ {}:{} - Reason: doesnt exists or is not a file.", file.getCanonicalPath(), list.getName(), lnr.getLineNumber());
				}
			}
		}
		
		compiler.compileAll().forEach((k, v) -> runMain(v));
	}
	
	private void executeAllScriptsInDirectory(InMemoryJavaCompiler compiler, File dir, boolean recurseDown) {
		if (!dir.isDirectory()) {
			throw new IllegalArgumentException("The argument directory either doesnt exists or is not an directory.");
		}
		
		final var files = dir.listFiles();
		if (files == null) {
			return;
		}
		
		for (var file : files) {
			if (file.isDirectory() && recurseDown) {
				if (general().debug()) {
					LOG.info("Entering folder: {}", file.getName());
				}
				executeAllScriptsInDirectory(compiler, file, recurseDown);
			} else if (file.isFile()) {
				addSource(compiler, file);
			}
		}
	}
	
	public Class<?> compileScript(File file) {
		try (var fis = new FileInputStream(file);
			var isr = new InputStreamReader(fis);
			var reader = new BufferedReader(isr)) {
			return compiler().compile(getClassForFile(file), readerToString(reader));
		} catch (Exception ex) {
			LOG.warn("Error executing script!", ex);
		}
		return null;
	}
	
	public void executeScript(File file) {
		final Class<?> clazz = compileScript(file);
		runMain(clazz);
	}
	
	public void executeScript(String file) {
		executeScript(new File(server().getScriptRoot(), file));
	}
	
	public void addSource(InMemoryJavaCompiler compiler, File file) {
		if (general().debug()) {
			LOG.info("Loading Script: {}", file.getAbsolutePath());
		}
		
		try (var fis = new FileInputStream(file);
			var isr = new InputStreamReader(fis);
			var reader = new BufferedReader(isr)) {
			compiler.addSource(getClassForFile(file), readerToString(reader));
		} catch (Exception ex) {
			LOG.warn("Error executing script!", ex);
		}
	}
	
	private static String getClassForFile(File script) {
		final String path = script.getAbsolutePath();
		final String scpPath = server().getScriptRoot().getAbsolutePath();
		if (path.startsWith(scpPath)) {
			final int idx = path.lastIndexOf('.');
			return path.substring(scpPath.length() + 1, idx).replace('/', '.').replace('\\', '.');
		}
		return null;
	}
	
	private static void runMain(Class<?> clazz) {
		final var mainMethod = findMethod(clazz, MAIN, ARG_MAIN);
		if (mainMethod == null) {
			LOG.warn("Unable to find main method in class {}!", clazz);
			return;
		}
		
		try {
			mainMethod.invoke(null, MAIN_METHOD_ARGS);
		} catch (Exception ex) {
			LOG.error("Error loading script {}!", clazz);
		}
	}
	
	private static String readerToString(Reader reader) throws ScriptException {
		try (var in = new BufferedReader(reader)) {
			final var result = new StringBuilder();
			String line;
			while ((line = in.readLine()) != null) {
				result.append(line).append(System.lineSeparator());
			}
			return result.toString();
		} catch (IOException ex) {
			throw new ScriptException(ex);
		}
	}
	
	private static Method findMethod(Class<?> clazz, String methodName, Class<?>[] args) {
		try {
			final var mainMethod = clazz.getMethod(methodName, args);
			final int modifiers = mainMethod.getModifiers();
			if (Modifier.isPublic(modifiers) && Modifier.isStatic(modifiers)) {
				return mainMethod;
			}
		} catch (NoSuchMethodException ignored) {
		}
		return null;
	}
	
	public static ScriptEngineManager getInstance() {
		return SingletonHolder.INSTANCE;
	}
	
	private static class SingletonHolder {
		protected static final ScriptEngineManager INSTANCE = new ScriptEngineManager();
	}
}