def insert_multiple_floppies(self, insert_paths):
paths = []
for path in insert_paths:
embedded_files = []
if path.endswith(".zip"):
archive = Archive(path)
files = archive.list_files()
for file in files:
name, ext = os.path.splitext(file)
# FIXME: get list of floppy extensions from a central
# place
if ext in [".adf", ".ipf"]:
embedded_files.append(file)
if len(embedded_files) > 0:
embedded_files.sort()
print("found embedded floppy images:")
print(embedded_files)
for file in embedded_files:
paths.append(file)
else:
paths.append(path)
default_dir = FSGSDirectories.get_floppies_dir()
checksum_tool = ChecksumTool()
for i, path in enumerate(paths):
sha1 = checksum_tool.checksum(path)
path = Paths.contract_path(path, default_dir)
if i < 4:
self.set_config(
[
("floppy_drive_{0}".format(i), path),
("x_floppy_drive_{0}_sha1".format(i), sha1),
]
)
self.set_config(
[
("floppy_image_{0}".format(i), path),
("x_floppy_image_{0}_sha1".format(i), sha1),
]
)
# blank the rest of the drives
for i in range(len(paths), 4):
self.set_config(
[
("floppy_drive_{0}".format(i), ""),
("x_floppy_drive_{0}_sha1".format(i), ""),
]
)
# blank the rest of the image list
for i in range(len(paths), 20):
self.set_config(
[
("floppy_image_{0}".format(i), ""),
("x_floppy_image_{0}_sha1".format(i), ""),
]
)
def multi_select(cls, parent=None):
default_dir = FSGSDirectories.get_floppies_dir()
dialog = LauncherFilePicker(parent,
gettext("Select Multiple Floppies"),
"floppy",
multiple=True)
if not dialog.show_modal():
return
original_paths = dialog.get_paths()
original_paths.sort()
paths = []
for path in original_paths:
path = Paths.get_real_case(path)
embedded_files = []
if path.endswith(".zip"):
archive = Archive(path)
files = archive.list_files()
for file in files:
name, ext = os.path.splitext(file)
# FIXME: get list of floppy extensions from a central
# place
if ext in [".adf", ".ipf"]:
embedded_files.append(file)
if len(embedded_files) > 0:
embedded_files.sort()
print("found embedded floppy images:")
print(embedded_files)
for file in embedded_files:
paths.append(file)
else:
paths.append(path)
checksum_tool = ChecksumTool(parent)
for i, path in enumerate(paths):
sha1 = checksum_tool.checksum(path)
path = Paths.contract_path(path,
default_dir,
force_real_case=False)
if i < 4:
LauncherConfig.set_multiple([
("floppy_drive_{0}".format(i), path),
("x_floppy_drive_{0}_sha1".format(i), sha1)
])
LauncherConfig.set_multiple([("floppy_image_{0}".format(i), path),
("x_floppy_image_{0}_sha1".format(i),
sha1)])
# blank the rest of the drives
for i in range(len(paths), 4):
LauncherConfig.set_multiple([("floppy_drive_{0}".format(i), ""),
("x_floppy_drive_{0}_sha1".format(i),
"")])
# blank the rest of the image list
for i in range(len(paths), 20):
LauncherConfig.set_multiple([("floppy_image_{0}".format(i), ""),
("x_floppy_image_{0}_sha1".format(i),
"")])
def insert_multiple_floppies(self, insert_paths):
paths = []
for path in insert_paths:
embedded_files = []
if path.endswith(".zip"):
archive = Archive(path)
files = archive.list_files()
for file in files:
name, ext = os.path.splitext(file)
# FIXME: get list of floppy extensions from a central
# place
if ext in [".adf", ".ipf"]:
embedded_files.append(file)
if len(embedded_files) > 0:
embedded_files.sort()
print("found embedded floppy images:")
print(embedded_files)
for file in embedded_files:
paths.append(file)
else:
paths.append(path)
default_dir = FSGSDirectories.get_floppies_dir()
checksum_tool = ChecksumTool()
for i, path in enumerate(paths):
sha1 = checksum_tool.checksum(path)
path = Paths.contract_path(path, default_dir)
if i < 4:
self.set_config([
("floppy_drive_{0}".format(i), path),
("x_floppy_drive_{0}_sha1".format(i), sha1),
])
self.set_config([
("floppy_image_{0}".format(i), path),
("x_floppy_image_{0}_sha1".format(i), sha1),
])
# blank the rest of the drives
for i in range(len(paths), 4):
self.set_config([
("floppy_drive_{0}".format(i), ""),
("x_floppy_drive_{0}_sha1".format(i), ""),
])
# blank the rest of the image list
for i in range(len(paths), 20):
self.set_config([
("floppy_image_{0}".format(i), ""),
("x_floppy_image_{0}_sha1".format(i), ""),
])
def scan_file(self, file_database, path):
name = os.path.basename(path)
# path = os.path.normcase(os.path.normpath(path))
self.scan_count += 1
self.set_status(
gettext("Scanning files ({count} scanned)").format(
count=self.scan_count),
name,
)
try:
st = os.stat(path)
except:
print("[FILES] WARNING: Error stat-ing file", repr(path))
return
size = st.st_size
mtime = int(st.st_mtime)
result = file_database.find_file(path=path)
if result["path"]:
if size == result["size"] and mtime == result["mtime"]:
# We've already got this file indexed.
self.database_file_ids.remove(result["id"])
return
archive = Archive(path)
file_id = self.scan_archive_stream(file_database, archive, path, name,
size, mtime)
for p in archive.list_files():
if p.endswith("/"):
# don't index archive directory entries
continue
# print(p)
if self.stop_check():
return
# n = p.replace("\\", "/").replace("|", "/").split("/")[-1]
n = os.path.basename(p)
self.scan_count += 1
self.scan_archive_stream(file_database,
archive,
p,
n,
size,
mtime,
parent=file_id)
if self.stop_check():
return
def scan_file(self, file_database, path):
name = os.path.basename(path)
# path = os.path.normcase(os.path.normpath(path))
self.scan_count += 1
self.set_status(
gettext("Scanning files ({count} scanned)").format(
count=self.scan_count
),
name,
)
try:
st = os.stat(path)
except:
print("[FILES] WARNING: Error stat-ing file", repr(path))
return
size = st.st_size
mtime = int(st.st_mtime)
result = file_database.find_file(path=path)
if result["path"]:
if size == result["size"] and mtime == result["mtime"]:
# We've already got this file indexed.
self.database_file_ids.remove(result["id"])
return
archive = Archive(path)
file_id = self.scan_archive_stream(
file_database, archive, path, name, size, mtime
)
for p in archive.list_files():
if p.endswith("/"):
# don't index archive directory entries
continue
# print(p)
if self.stop_check():
return
# n = p.replace("\\", "/").replace("|", "/").split("/")[-1]
n = os.path.basename(p)
self.scan_count += 1
self.scan_archive_stream(
file_database, archive, p, n, size, mtime, parent=file_id
)
if self.stop_check():
return
def calculate_whdload_args(archive_path: str) -> str:
"""
This function, as it is currently written, only works if there
is an .info with the same name as the .slave file. In theory, they
could be different since the .info file contains a slave=... tool type.
"""
archive = Archive(archive_path)
slave_args = {}
lower_to_name = {}
for path in archive.list_files():
path_lower = path.lower()
if path_lower.rsplit("#/", 1)[1] == "s/startup-sequence":
logging.info("[WHDLOAD] Found Startup-Sequence, assuming "
"non-WHDLoad archive")
return ""
lower_to_name[path_lower] = path
for path in lower_to_name.values():
name = os.path.basename(path)
name_lower = name.lower()
if name_lower.endswith(".info"):
try:
args = read_whdload_args_from_info_stream(archive.open(path))
args = strip_whdload_slave_prefix(args)
except Exception as e:
logging.warning("[WHDLOAD] WARNING: Error reading args: %s",
repr(e))
else:
if args:
archive_name = path.rsplit("#/", 1)[1]
logging.debug("[WHDLOAD] %s => %s".format(
archive_name, " ".join(args)))
slave_args[archive_name] = args
if len(slave_args) == 0:
return ""
if len(slave_args) > 1:
logging.debug("[WHDLOAD] Multiple WHDLoad icons found")
for icon, args in slave_args.items():
if icon.lower() in primary_icons:
logging.debug("[WHDLOAD] Choosing %s as primary icon", icon)
return fix_whdload_args(args)
raise Exception("Multiple slaves found")
return fix_whdload_args(slave_args.popitem()[1])
def download_game_file_archive(self, url):
print("\ndownload_game_file_archive", url)
archive_path = Downloader.cache_file_from_url(url)
archive = Archive(archive_path)
archive_files = archive.list_files()
print(archive_files)
for name in archive_files:
print(name)
ifs = archive.open(name)
data = ifs.read()
Downloader.cache_data(data)
if len(archive_files) == 0:
# might not be an archive then
with open(archive_path, "rb") as f:
data = f.read()
Downloader.cache_data(data)
# the downloaded archive is no longer needed, now that we have
# extracted all the files
os.remove(archive_path)
print("\n")
def download_game_file_archive(self, url):
print("\ndownload_game_file_archive", url)
archive_path = Downloader.cache_file_from_url(url)
archive = Archive(archive_path)
archive_files = archive.list_files()
print(archive_files)
for name in archive_files:
print(name)
ifs = archive.open(name)
data = ifs.read()
Downloader.cache_data(data)
if len(archive_files) == 0:
# might not be an archive then
with open(archive_path, "rb") as f:
data = f.read()
Downloader.cache_data(data)
# the downloaded archive is no longer needed, now that we have
# extracted all the files
os.remove(archive_path)
print("\n")
def multi_select(cls, parent=None):
default_dir = FSGSDirectories.get_floppies_dir()
dialog = LauncherFilePicker(
parent,
gettext("Select Multiple Floppies"),
"floppy",
multiple=True,
)
if not dialog.show_modal():
return
original_paths = dialog.get_paths()
original_paths.sort()
paths = []
for path in original_paths:
path = Paths.get_real_case(path)
embedded_files = []
if path.endswith(".zip"):
archive = Archive(path)
files = archive.list_files()
for file in files:
name, ext = os.path.splitext(file)
# FIXME: get list of floppy extensions from a central
# place
if ext in [".adf", ".ipf"]:
embedded_files.append(file)
if len(embedded_files) > 0:
embedded_files.sort()
print("found embedded floppy images:")
print(embedded_files)
for file in embedded_files:
paths.append(file)
else:
paths.append(path)
checksum_tool = ChecksumTool(parent)
for i, path in enumerate(paths):
sha1 = checksum_tool.checksum(path)
path = Paths.contract_path(
path, default_dir, force_real_case=False
)
if i < 4:
LauncherConfig.set_multiple(
[
("floppy_drive_{0}".format(i), path),
("x_floppy_drive_{0}_sha1".format(i), sha1),
]
)
LauncherConfig.set_multiple(
[
("floppy_image_{0}".format(i), path),
("x_floppy_image_{0}_sha1".format(i), sha1),
]
)
# blank the rest of the drives
for i in range(len(paths), 4):
LauncherConfig.set_multiple(
[
("floppy_drive_{0}".format(i), ""),
("x_floppy_drive_{0}_sha1".format(i), ""),
]
)
# blank the rest of the image list
for i in range(len(paths), 20):
LauncherConfig.set_multiple(
[
("floppy_image_{0}".format(i), ""),
("x_floppy_image_{0}_sha1".format(i), ""),
]
)
def run_config_or_game(cls):
config_path = None
archive_path = None
floppy_image_paths = []
cdrom_image_paths = []
config_uuid = None
floppy_extensions = (".adf", ".ipf", ".dms", ".adz")
cdrom_extensions = (".cue", ".iso")
archive_extensions = (".zip", ".lha")
# FIXME: replace argument "parsing" with use of argparse module
# at some point
last_arg = sys.argv[-1]
file_ext = os.path.splitext(last_arg)[-1].lower()
if file_ext == ".fs-uae":
config_path = last_arg
elif file_ext in archive_extensions:
archive_path = last_arg
# elif file_ext in floppy_extensions:
# floppy_image_paths = [last_arg]
elif is_uuid(last_arg):
config_uuid = last_arg.lower()
for arg in sys.argv[1:]:
if not arg.startswith("--"):
_, ext = os.path.splitext(arg)
if ext in floppy_extensions:
floppy_image_paths.append(arg)
elif ext in cdrom_extensions:
cdrom_image_paths.append(arg)
if config_path:
print("[STARTUP] Config path given:", config_path)
if not os.path.exists(config_path):
print("[STARTUP] Config path does not exist", file=sys.stderr)
return True
LauncherConfig.load_file(config_path)
fsgs.config.add_from_argv()
return cls.run_config_directly()
if archive_path:
print("[STARTUP] Archive path given:", archive_path)
if not os.path.exists(archive_path):
print("[STARTUP] Archive path does not exist", file=sys.stderr)
return True
archive = Archive(os.path.realpath(archive_path))
archive_name = os.path.basename(archive_path)
# We want to exclude pure directory entries when checking for
# archives with only floppies.
arc_files = [
x for x in archive.list_files() if not x.endswith("/")
]
if all(
map(lambda f: f.lower().endswith(floppy_extensions),
arc_files)):
print("[STARTUP] Archive contains floppy disk images only")
floppy_image_paths = arc_files
else:
if cls.auto_detect_game:
# FIXME: Could also do this for floppy file archives.
archive_util = ArchiveUtil(archive_path)
archive_uuid = archive_util.create_variant_uuid()
print("[STARTUP] Try auto-detecting variant, uuid =",
archive_uuid)
if fsgs.load_game_variant(archive_uuid):
print("[STARTUP] Auto-detected variant", archive_uuid)
print("[STARTUP] Adding archive files to file index")
for archive_file in archive.list_files():
stream = archive.open(archive_file)
data = stream.read()
size = len(data)
sha1 = hashlib.sha1(data).hexdigest()
FileDatabase.add_static_file(archive_file,
size=size,
sha1=sha1)
fsgs.config.add_from_argv()
fsgs.config.set("__config_name", archive_name)
LauncherConfig.post_load_values(fsgs.config)
return cls.run_config_directly()
values = whdload.generate_config_for_archive(archive_path)
values["hard_drive_0"] = archive_path
values.update(fsgs.config.config_from_argv())
# archive_name, archive_ext = os.path.splitext(archive_name)
values["__config_name"] = archive_name
return cls.run_config_directly_with_values(values)
if floppy_image_paths:
enum_paths = tuple(enumerate(floppy_image_paths))
values = {}
values.update(fsgs.config.config_from_argv())
max_drives = int(values.get("floppy_drive_count", "4"))
values.update({
"floppy_drive_{0}".format(k): v
for k, v in enum_paths[:max_drives]
})
values.update(
{"floppy_image_{0}".format(k): v
for k, v in enum_paths[:20]})
# FIXME: Generate a better config name for save dir?
values["__config_name"] = "Default"
return cls.run_config_directly_with_values(values)
if cdrom_image_paths:
enum_paths = tuple(enumerate(cdrom_image_paths))
values = {"amiga_model": "CD32"}
values.update(fsgs.config.config_from_argv())
max_drives = int(values.get("cdrom_drive_count", "1"))
values.update({
"cdrom_drive_{0}".format(k): v
for k, v in enum_paths[:max_drives]
})
values.update(
{"cdrom_image_{0}".format(k): v
for k, v in enum_paths[:20]})
# FIXME: Generate a better config name for save dir?
values["__config_name"] = "Default"
return cls.run_config_directly_with_values(values)
if config_uuid:
print("[STARTUP] Config uuid given:", config_uuid)
variant_uuid = config_uuid
# values = fsgs.game.set_from_variant_uuid(variant_uuid)
if fsgs.load_game_variant(variant_uuid):
print("[STARTUP] Loaded variant")
else:
print("[STARTUP] Could not load variant, try to load game")
game_uuid = config_uuid
variant_uuid = fsgs.find_preferred_game_variant(game_uuid)
print("[STARTUP] Preferred variant:", variant_uuid)
fsgs.load_game_variant(variant_uuid)
fsgs.config.add_from_argv()
LauncherConfig.post_load_values(fsgs.config)
return cls.run_config_directly()
def run_config_or_game(cls):
config_path = None
archive_path = None
floppy_image_paths = []
cdrom_image_paths = []
config_uuid = None
floppy_extensions = (".adf", ".ipf", ".dms", ".adz")
cdrom_extensions = (".cue", ".iso")
archive_extensions = (".zip", ".lha")
# FIXME: replace argument "parsing" with use of argparse module
# at some point
last_arg = sys.argv[-1]
file_ext = os.path.splitext(last_arg)[-1].lower()
if file_ext == ".fs-uae":
config_path = last_arg
elif file_ext in archive_extensions:
archive_path = last_arg
# elif file_ext in floppy_extensions:
# floppy_image_paths = [last_arg]
elif is_uuid(last_arg):
config_uuid = last_arg.lower()
for arg in sys.argv[1:]:
if not arg.startswith("--"):
_, ext = os.path.splitext(arg)
if ext in floppy_extensions:
floppy_image_paths.append(arg)
elif ext in cdrom_extensions:
cdrom_image_paths.append(arg)
if config_path:
print("[STARTUP] Config path given:", config_path)
if not os.path.exists(config_path):
print("[STARTUP] Config path does not exist", file=sys.stderr)
return True
LauncherConfig.load_file(config_path)
fsgs.config.add_from_argv()
return cls.run_config_directly()
if archive_path:
print("[STARTUP] Archive path given:", archive_path)
if not os.path.exists(archive_path):
print("[STARTUP] Archive path does not exist", file=sys.stderr)
return True
archive = Archive(os.path.realpath(archive_path))
archive_name = os.path.basename(archive_path)
# We want to exclude pure directory entries when checking for
# archives with only floppies.
arc_files = [
x for x in archive.list_files() if not x.endswith("/")
]
if all(
map(lambda f: f.lower().endswith(floppy_extensions), arc_files)
):
print("[STARTUP] Archive contains floppy disk images only")
floppy_image_paths = arc_files
else:
if cls.auto_detect_game:
# FIXME: Could also do this for floppy file archives.
archive_util = ArchiveUtil(archive_path)
archive_uuid = archive_util.create_variant_uuid()
print(
"[STARTUP] Try auto-detecting variant, uuid =",
archive_uuid,
)
if fsgs.load_game_variant(archive_uuid):
print("[STARTUP] Auto-detected variant", archive_uuid)
print("[STARTUP] Adding archive files to file index")
for archive_file in archive.list_files():
stream = archive.open(archive_file)
data = stream.read()
size = len(data)
sha1 = hashlib.sha1(data).hexdigest()
FileDatabase.add_static_file(
archive_file, size=size, sha1=sha1
)
fsgs.config.add_from_argv()
fsgs.config.set("__config_name", archive_name)
LauncherConfig.post_load_values(fsgs.config)
return cls.run_config_directly()
values = whdload.generate_config_for_archive(archive_path)
values["hard_drive_0"] = archive_path
values.update(fsgs.config.config_from_argv())
# archive_name, archive_ext = os.path.splitext(archive_name)
values["__config_name"] = archive_name
return cls.run_config_directly_with_values(values)
if floppy_image_paths:
enum_paths = tuple(enumerate(floppy_image_paths))
values = {}
values.update(fsgs.config.config_from_argv())
max_drives = int(values.get("floppy_drive_count", "4"))
values.update(
{
"floppy_drive_{0}".format(k): v
for k, v in enum_paths[:max_drives]
}
)
values.update(
{"floppy_image_{0}".format(k): v for k, v in enum_paths[:20]}
)
# FIXME: Generate a better config name for save dir?
values["__config_name"] = "Default"
return cls.run_config_directly_with_values(values)
if cdrom_image_paths:
enum_paths = tuple(enumerate(cdrom_image_paths))
values = {"amiga_model": "CD32"}
values.update(fsgs.config.config_from_argv())
max_drives = int(values.get("cdrom_drive_count", "1"))
values.update(
{
"cdrom_drive_{0}".format(k): v
for k, v in enum_paths[:max_drives]
}
)
values.update(
{"cdrom_image_{0}".format(k): v for k, v in enum_paths[:20]}
)
# FIXME: Generate a better config name for save dir?
values["__config_name"] = "Default"
return cls.run_config_directly_with_values(values)
if config_uuid:
print("[STARTUP] Config uuid given:", config_uuid)
variant_uuid = config_uuid
# values = fsgs.game.set_from_variant_uuid(variant_uuid)
if fsgs.load_game_variant(variant_uuid):
print("[STARTUP] Loaded variant")
else:
print("[STARTUP] Could not load variant, try to load game")
game_uuid = config_uuid
variant_uuid = fsgs.find_preferred_game_variant(game_uuid)
print("[STARTUP] Preferred variant:", variant_uuid)
fsgs.load_game_variant(variant_uuid)
fsgs.config.add_from_argv()
LauncherConfig.post_load_values(fsgs.config)
return cls.run_config_directly()
