Subversion Repositories Kolibri OS

Rev

Rev 9376 | Rev 9381 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | Download | RSS feed

  1. import os
  2. import sys
  3. import shlex
  4. import shutil
  5. import urllib.request
  6. import subprocess
  7.  
  8. path_to_tools_workspace = os.path.dirname(os.path.abspath(__file__))
  9. path_to_tools = os.path.dirname(path_to_tools_workspace)
  10. sys.path.append(path_to_tools)
  11.  
  12. from workspace.build import build
  13.  
  14. from lib.makeflop import Floppy
  15. from lib.platform import is_win32, path
  16. from lib.logging import log
  17. from lib.network import download
  18.  
  19. # TODO: Move into _tools/lib
  20. def get_file_directory(path):
  21.     path = path.replace("\\", "/")
  22.     if "/" in path:
  23.         folder = "/".join(path.split("/")[:-1])
  24.         if folder == "":
  25.             return "/" # It was a file in the root folder
  26.         return folder
  27.     else:
  28.         return "." # Just a filename, let's return current folder
  29.  
  30. # TODO: Move into _tools/lib
  31. def run_qemu(start_dir = "workspace"):
  32.     qemu_command = f"qemu-system-i386"
  33.     flags = ""
  34.     flags += "-L . " # IDK why it does not work without this
  35.     flags += "-m 128 "
  36.     flags += f"-drive format=raw,file={start_dir}/kolibri.img,index=0,if=floppy -boot a "
  37.     flags += "-vga vmware "
  38.     flags += "-net nic,model=rtl8139 -net user "
  39.     flags += "-soundhw ac97 "
  40.     if is_win32():
  41.         qemu_full_path = shutil.which(qemu_command)
  42.         qemu_directory = get_file_directory(qemu_full_path)
  43.         flags += f"-L {qemu_directory} "
  44.     s = f"{qemu_command} {flags}"
  45.     qemu_stdout = open(f"{start_dir}/qemu_stdout.log", "w")
  46.     qemu_stderr = open(f"{start_dir}/qemu_stderr.log", "w")
  47.     if is_win32():
  48.         return subprocess.Popen(s, bufsize = 0, stdout = qemu_stdout, stderr = qemu_stderr, stdin = subprocess.DEVNULL, shell = True, start_new_session = True)
  49.     else:
  50.         a = shlex.split(s)
  51.         return subprocess.Popen(a, bufsize = 0, stdout = qemu_stdout, stderr = qemu_stderr, stdin = subprocess.DEVNULL, start_new_session = True)
  52.  
  53. if __name__ == "__main__":
  54.     program_name = build()
  55.  
  56.     os.makedirs("workspace", exist_ok = True)
  57.  
  58.     if not os.path.exists("workspace/kolibri.unmodified.img"):
  59.         img_url = "http://builds.kolibrios.org/eng/data/data/kolibri.img"
  60.         download(img_url, "workspace/kolibri.unmodified.img")
  61.  
  62.     # Create a copy of IMG
  63.     shutil.copyfile("workspace/kolibri.unmodified.img", "workspace/kolibri.img")
  64.  
  65.     # Open the IMG
  66.     with open("workspace/kolibri.img", "rb") as img:
  67.         img_data = img.read()
  68.     img = Floppy(img_data)
  69.  
  70.     # Remove unuseful folders
  71.     img.delete_path("GAMES")
  72.     img.delete_path("DEMOS")
  73.     img.delete_path("3D")
  74.  
  75.     log("Moving program into kolibri image... ", end = "")
  76.     with open(program_name, "rb") as file:
  77.         file_data = file.read()
  78.     if not img.add_file_path(program_name.upper(), file_data):
  79.         print(f"Coudn't move {program_name} into IMG")
  80.     log("Done")
  81.  
  82.     # TODO: Figure out which of compiled files is a program executable and only run it
  83.     log("Adding program to autorun.dat... ", end = "")
  84.     lines_to_add = bytes(f"\r\n/SYS/{program_name.upper()}\t\t""\t0\t# Your program", "ascii")
  85.     autorun_dat = img.extract_file_path("SETTINGS\AUTORUN.DAT")
  86.     place_for_new_lines = autorun_dat.index(b"\r\n/SYS/@TASKBAR")# b"\r\n### Hello, ASM World! ###")
  87.     autorun_dat = autorun_dat[:place_for_new_lines] + lines_to_add + autorun_dat[place_for_new_lines:]
  88.     img.delete_path("SETTINGS\AUTORUN.DAT")
  89.     img.add_file_path("SETTINGS\AUTORUN.DAT", autorun_dat)
  90.     log("Done")
  91.  
  92.     img.save("workspace/kolibri.img")
  93.  
  94.     run_qemu()
  95.