initial commit

src/de/extio/lm_launcher/Data.java

@@ -0,0 +1,115 @@
+package de.extio.lm_launcher;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Objects;
+import java.util.Properties;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+public class Data {
+	
+	static final Properties props;
+	
+	static ModelData modelData;
+	
+	static AppData appData;
+	
+	static {
+		props = new Properties();
+		try {
+			props.load(new FileInputStream("lm-launcher.properties"));
+		}
+		catch (final IOException e) {
+			e.printStackTrace();
+		}
+		
+		final ObjectMapper objectMapper = new ObjectMapper();
+		
+		try {
+			modelData = objectMapper.readValue(new File("models.json"), ModelData.class);
+		}
+		catch (final IOException e) {
+			e.printStackTrace();
+		}
+		if (modelData == null || modelData.models == null) {
+			modelData = new ModelData(new ArrayList<>());
+		}
+		// Migration
+		for (int i = 0; i < modelData.models().size(); i++) {
+			final Model model = modelData.models.get(i);
+			if (model.maxContextSize() < model.contextSize() || model.promptTemplate() == null) {
+				modelData.models.set(i, new Model(model.path(), model.contextSize(), model.contextSize(), model.gpuLayers(), model.threads(), Objects.requireNonNullElse(model.promptTemplate(), "")));
+			}
+		}
+		
+		try {
+			appData = objectMapper.readValue(new File("apps.json"), AppData.class);
+		}
+		catch (final IOException e) {
+			e.printStackTrace();
+		}
+		if (appData == null || appData.apps == null) {
+			appData = new AppData(new ArrayList<>());
+		}
+		// Migration
+		for (int i = 0; i < appData.apps.size(); i++) {
+			final App app = appData.apps.get(i);
+			if (app.appArguments() == null) {
+				final List<AppArgument> arguments = new ArrayList<>(app.arguments().size());
+				app.arguments().stream().map(argument -> new AppArgument(argument, false, true)).forEach(arguments::add);
+				appData.apps.set(i, new App(app.path(), app.interpreter(), new ArrayList<>(), arguments));
+			}
+		}
+	}
+	
+	static void saveModels() {
+		final ObjectMapper objectMapper = new ObjectMapper();
+		try {
+			objectMapper.writeValue(new File("models.json"), modelData);
+		}
+		catch (final IOException e) {
+			e.printStackTrace();
+		}
+	}
+	
+	static void saveApps() {
+		final ObjectMapper objectMapper = new ObjectMapper();
+		try {
+			objectMapper.writeValue(new File("apps.json"), appData);
+		}
+		catch (final IOException e) {
+			e.printStackTrace();
+		}
+	}
+	
+	static List<String> scanModels() {
+		final List<String> result = new ArrayList<>();
+		try {
+			Files
+					.list(Path.of(props.getProperty("models")))
+					.filter(p -> p.getFileName().toString().endsWith(".gguf"))
+					.map(Path::toAbsolutePath)
+					.map(Path::toString)
+					.sorted()
+					.forEach(result::add);
+		}
+		catch (final IOException e) {
+			e.printStackTrace();
+		}
+		return result;
+	}
+	
+	static record ModelData(List<Model> models) {
+		
+	}
+	
+	static record AppData(List<App> apps) {
+		
+	}
+}