bcv-vf/src/main/java/the/bytecode/club/bytecodeviewer/util/BootCheck.java

package the.bytecode.club.bytecodeviewer.util;

import the.bytecode.club.bootloader.Boot;
import the.bytecode.club.bootloader.ILoader;
import the.bytecode.club.bootloader.resource.EmptyExternalResource;
import the.bytecode.club.bootloader.resource.ExternalResource;
import the.bytecode.club.bytecodeviewer.BytecodeViewer;
import the.bytecode.club.bytecodeviewer.CommandLineInput;

import javax.swing.*;
import java.io.File;
import java.util.ArrayList;
import java.util.List;

import static the.bytecode.club.bytecodeviewer.Constants.nl;

/**
 * Loads the libraries on boot. If booting failed for some reason, this kicks in as a fail safe.
 *
 * This broke with maven so now only FatJar builds will work.
 *
 * To get this system working again for smaller binaries/automatic updating libraries maven support will need to be added.
 *
 * @author Konloch
 * @author Bibl (don't ban me pls)
 * @since 6/21/2021
 */
public class BootCheck implements Runnable
{
	@Override
	public void run()
	{
		long start = System.currentTimeMillis();
		
		//7 second failsafe
		while (System.currentTimeMillis() - start < 7000)
		{
			try {
				Thread.sleep(100);
			} catch (InterruptedException ignored) { }
		}
		
		//if it's failed to boot and it's not downloading attempt to load the libraries
		failSafeLoadLibraries();
	}
	
	@SuppressWarnings({"rawtypes", "unchecked"})
	public void failSafeLoadLibraries()
	{
		if (!Boot.completedboot && !Boot.downloading)
		{
			File libsDir = Boot.libsDir();
			File[] listFiles = libsDir.listFiles();
			
			//first boot failed to download libraries
			if (listFiles == null || listFiles.length <= 0)
			{
				BytecodeViewer.showMessage(
						"Github is loading extremely slow, BCV needs to download libraries from github in order"
								+ nl + "to work, please try adjusting your network settings or manually downloading these libraries"
								+ nl + "if this error persists.");
				return;
			}
			
			Boot.setState("Bytecode Viewer Boot Screen (OFFLINE MODE) - Unable to connect to github, force booting...");
			System.out.println("Unable to connect to github, force booting...");
			
			List<String> libsFileList = new ArrayList<>();
			for (File f : listFiles)
				libsFileList.add(f.getAbsolutePath());
			
			ILoader<?> loader = Boot.findLoader();
			for (String s : libsFileList)
			{
				if (s.endsWith(".jar"))
				{
					File f = new File(s);
					if (f.exists())
					{
						Boot.setState("Bytecode Viewer Boot Screen (OFFLINE MODE) - Force Loading Library "
								+ f.getName());
						System.out.println("Force loading library " + f.getName());
						
						try
						{
							ExternalResource res = new EmptyExternalResource<>(f.toURI().toURL());
							loader.bind(res);
							System.out.println("Successfully loaded " + f.getName());
						}
						catch (Exception e)
						{
							e.printStackTrace();
							f.delete();
							JOptionPane.showMessageDialog(null,
									"Error, Library " + f.getName() + " is corrupt, please restart to re-download it.",
									"Error", JOptionPane.ERROR_MESSAGE);
						}
					}
				}
			}
			
			Boot.checkEnjarify();
			Boot.checkKrakatau();
			
			Boot.globalstop = false;
			Boot.hide();
			
			if (CommandLineInput.parseCommandLine(BytecodeViewer.launchArgs) == CommandLineInput.OPEN_FILE)
				BytecodeViewer.boot(false);
			else {
				BytecodeViewer.boot(true);
				CommandLineInput.executeCommandLine(BytecodeViewer.launchArgs);
			}
		}
	}
}