Re-commit files

Author: YorkshireDev

.gitignore

@@ -250,3 +250,5 @@ dmypy.json
 # Cython debug symbols
 cython_debug/
 
+!/DLSS/
+!/Game_List.TXT

.idea/modules.xml

@@ -0,0 +1,21 @@
+import zipfile
+from sys import argv
+
+
+def main(dir_list_str: str) -> str:
+
+    dir_list_split: list = dir_list_str.split(">")
+
+    game_dlss_dir: str = dir_list_split[0]
+    backup_file_name: str = dir_list_split[1] + dir_list_split[2] + ".zip"
+
+    with zipfile.ZipFile(backup_file_name, "w", compression=zipfile.ZIP_LZMA, compresslevel=9) as backup_zip:
+        backup_zip.write(game_dlss_dir, arcname="nvngx_dlss.dll")
+
+    return str(True)
+
+
+if __name__ == "__main__":
+
+    result: str = main(argv[1])
+    print(result)

Python/ModelPythonBackup.py

@@ -0,0 +1,29 @@
+import zipfile
+from sys import argv
+from os import remove
+
+
+def main(game_dlss_dir: str, replace_dlss_file_dir: str) -> str:
+
+    remove(game_dlss_dir)
+
+    game_dlss_dir_split: list = game_dlss_dir.split("\\")
+    game_dlss_dir = ""
+
+    for i in range(len(game_dlss_dir_split) - 1):
+
+        game_dlss_dir += game_dlss_dir_split[i]
+
+        if i < len(game_dlss_dir_split) - 1:
+            game_dlss_dir += "\\"
+
+    with zipfile.ZipFile(replace_dlss_file_dir, "r") as dlss_zip_file:
+        dlss_zip_file.extractall(game_dlss_dir)
+
+    return str(True)
+
+
+if __name__ == "__main__":
+
+    result: str = main(argv[1], argv[2])
+    print(result)

Python/ModelPythonReplace.py

@@ -20,4 +20,3 @@ def main(dlss_dir: str) -> str:
 
     result: str = main(argv[1])
     print(result)
-

Python/ModelPythonSearch.py

@@ -1,11 +1,18 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <module type="JAVA_MODULE" version="4">
+  <component name="FacetManager">
+    <facet type="Python" name="Python">
+      <configuration sdkName="Python 3.10 (Supercede)" />
+    </facet>
+  </component>
   <component name="NewModuleRootManager" inherit-compiler-output="true">
     <exclude-output />
     <content url="file://$MODULE_DIR$">
       <sourceFolder url="file://$MODULE_DIR$/src" isTestSource="false" />
+      <excludeFolder url="file://$MODULE_DIR$/venv" />
     </content>
     <orderEntry type="inheritedJdk" />
     <orderEntry type="sourceFolder" forTests="false" />
+    <orderEntry type="library" name="Python 3.10 (Supercede) interpreter library" level="application" />
   </component>
 </module>

Supersede.iml

@@ -1,2 +1,166 @@
-package GUI;public class ViewUserInterface {
+package GUI;
+
+import History.ControllerBackup;
+import Replace.ControllerReplace;
+import Search.ControllerSearch;
+
+import javax.swing.*;
+import java.awt.*;
+import java.io.File;
+import java.io.IOException;
+import java.net.URI;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.Objects;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+
+public class ViewUserInterface extends JFrame {
+
+    private JPanel panelMain;
+    private JLabel labelRequiredAction;
+    private JButton buttonDLSSWebsite;
+    private JButton buttonAddGameDirectory;
+    private JLabel labelRestoreBackup;
+    private JButton buttonRestoreBackup;
+    private JLabel labelGitHubLink;
+    private JButton buttonReplaceDLSS;
+    private JLabel labelProgress;
+    private JLabel labelProgressUpdate;
+
+    private ExecutorService controllerService;
+    public static boolean isControllerRunning;
+
+    public ViewUserInterface() {
+
+        SwingUtilities.invokeLater(this::initUserInterface);
+        SwingUtilities.invokeLater(this::initNonUserInterface);
+
+        buttonDLSSWebsite.addActionListener(e -> {
+
+            try {
+                Desktop desktop = Desktop.getDesktop();
+                desktop.browse(URI.create("https://www.techpowerup.com/download/nvidia-dlss-dll/"));
+            } catch (IOException ex) { throw new RuntimeException(ex); }
+
+        });
+
+        buttonAddGameDirectory.addActionListener(e -> {
+
+            if (isControllerRunning) return;
+
+            File gameRootDir = null;
+
+            JFileChooser dirChooser = new JFileChooser();
+            dirChooser.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
+            dirChooser.setDialogTitle("Select Game Root Directory");
+
+            int dirChoice = dirChooser.showDialog(this, "Select");
+
+            if (dirChoice == JFileChooser.APPROVE_OPTION) gameRootDir = dirChooser.getSelectedFile();
+
+            if (gameRootDir != null) responseButtonAddGameDirectory(gameRootDir);
+
+        });
+
+        buttonReplaceDLSS.addActionListener(e -> {
+
+            if (isControllerRunning) return;
+
+            File[] DLSSDir = new File("DLSS").listFiles();
+
+            if (DLSSDir == null || DLSSDir.length == 0) {
+
+                showMessage("DLSS directory is empty!");
+                return;
+
+            }
+
+            ControllerBackup controllerBackup = new ControllerBackup(this);
+            ControllerReplace controllerReplace = new ControllerReplace(this, false);
+            this.controllerService.submit(controllerBackup);
+            this.controllerService.submit(controllerReplace);
+
+        });
+
+        buttonRestoreBackup.addActionListener(e -> {
+
+            if (isControllerRunning) return;
+
+            File[] backupDir = new File("Backup").listFiles();
+
+            if (backupDir == null || backupDir.length == 0) {
+
+                showMessage("Backup directory is empty!");
+                return;
+
+            }
+
+            ControllerReplace controllerReplace = new ControllerReplace(this, true);
+            this.controllerService.submit(controllerReplace);
+
+        });
+
+    }
+
+    private void responseButtonAddGameDirectory(File gameRootDir) {
+
+        ControllerSearch controllerSearch = new ControllerSearch(this, gameRootDir);
+        this.controllerService.submit(controllerSearch);
+
+    }
+
+    private void showMessage(String text) {
+
+        SwingUtilities.invokeLater(() ->
+                JOptionPane.showMessageDialog(this, text, "Warning", JOptionPane.WARNING_MESSAGE));
+    }
+
+    public void updateLabelProgressUpdate(String progressText) {
+
+        SwingUtilities.invokeLater(() -> labelProgressUpdate.setText(progressText));
+
+    }
+
+    private void initUserInterface() {
+
+        this.labelRequiredAction.setText("Required User Action");
+        this.buttonDLSSWebsite.setText("Download DLSS");
+        this.buttonAddGameDirectory.setText("Add Game Directory");
+
+        this.labelRestoreBackup.setText("Restore");
+        this.buttonRestoreBackup.setText("Restore Backup");
+
+        this.labelGitHubLink.setText("https://github.com/YorkshireDev");
+        this.buttonReplaceDLSS.setText("Replace DLSS");
+        this.labelProgress.setText("Progress");
+        this.labelProgressUpdate.setText("N/A");
+
+        this.setContentPane(panelMain);
+        this.setTitle("Supersede");
+        this.setPreferredSize(new Dimension(480, 240));
+        this.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
+        this.setLocationRelativeTo(null);
+        this.pack();
+        this.setVisible(true);
+
+    }
+
+    private void initNonUserInterface() {
+
+        Path dlssFolderPath = Path.of("DLSS");
+        Path dlssValidGameFilePath = Path.of("Game_List.TXT");
+
+        try {
+
+            if (! Files.exists(dlssFolderPath)) Files.createDirectory(dlssFolderPath);
+            if (! Files.exists(dlssValidGameFilePath)) Files.createFile(dlssValidGameFilePath);
+
+        } catch (IOException e) { throw new RuntimeException(e); }
+
+        this.controllerService = Executors.newSingleThreadExecutor();
+        isControllerRunning = false;
+
+    }
+
 }

src/GUI/ViewUserInterface.form

@@ -1,2 +1,75 @@
-package History;public class ControllerBackup {
+package History;
+
+import GUI.ViewUserInterface;
+
+import java.io.BufferedReader;
+import java.io.FileReader;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+
+public class ControllerBackup implements Runnable {
+
+    private final ViewUserInterface viewUserInterface;
+
+    private final ExecutorService modelBackupService;
+    static CountDownLatch modelBackupLatch;
+
+    public ControllerBackup(ViewUserInterface viewUserInterface) {
+
+        this.viewUserInterface = viewUserInterface;
+        this.modelBackupService = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());
+
+    }
+
+    @Override
+    public void run() {
+
+        ViewUserInterface.isControllerRunning = true;
+
+        try {
+
+            Path backupPath = Path.of("Backup");
+            if (! Files.exists(backupPath)) Files.createDirectory(backupPath);
+
+            List<ModelBackup> modelBackupList = new ArrayList<>();
+
+            BufferedReader gameListReader = new BufferedReader(new FileReader("Game_List.TXT"));
+            String gameLine;
+
+            while ((gameLine = gameListReader.readLine()) != null) modelBackupList.add(new ModelBackup(gameLine));
+
+            Collections.shuffle(modelBackupList);
+
+            int gameAmount = modelBackupList.size();
+            modelBackupLatch = new CountDownLatch(gameAmount);
+
+            for (ModelBackup modelBackup : modelBackupList) this.modelBackupService.submit(modelBackup);
+
+            while (modelBackupLatch.getCount() > 0) {
+
+                long searchedDirs = gameAmount - modelBackupLatch.getCount();
+                String progressText = searchedDirs + "/" + gameAmount;
+                viewUserInterface.updateLabelProgressUpdate(progressText);
+
+                try {
+                    Thread.sleep(1L);
+                } catch (InterruptedException e) { throw new RuntimeException(e); }
+
+            }
+
+            viewUserInterface.updateLabelProgressUpdate("Done!");
+
+        } catch (IOException e) { throw new RuntimeException(e); }
+
+        ViewUserInterface.isControllerRunning = false;
+
+    }
+
 }