ChocolateBird/widgets/RomSelect.gd

196 lines
7.1 KiB
GDScript

extends PanelContainer
#warning-ignore-all:return_value_discarded
const FOLDER_ICON := preload('res://theme/icons/file_folder.tres')
const ALLOWED_EXTS := PoolStringArray(['bin', 'iso', 'sfc', 'srm', 'gba'])
const CD_EXTS := PoolStringArray(['bin', 'iso']) # If you have a weird disc image format, you can mount it yourself, leave me out of it
var EXT_ICONS := { # Dicts can't be const
'bin': preload('res://theme/icons/file_binary.tres'),
'iso': preload('res://theme/icons/file_disc.tres'),
'sfc': preload('res://theme/icons/file_cart.tres'),
'gba': preload('res://theme/icons/file_cart.tres'),
}
var TYPE_DESCS := { # Dicts can't be const
'bin': 'Binary',
'iso': 'CD-ROM Image',
'sfc': 'SNES ROM',
'gba': 'GBA ROM',
'srm': 'SNES Savefile'
}
var cached_cd_bin_paths := {}
var dir := Directory.new()
var home_path := ''
onready var itemlist: ItemList = $VBoxContainer/HBoxContainer/PanelContainer/ItemList
onready var vscroller: VScrollBar = itemlist.get_child(0)
onready var folder_buttons_scroller: ScrollContainer = $VBoxContainer/ScrollContainer
onready var folder_buttons: HBoxContainer = $VBoxContainer/ScrollContainer/folder_buttons
onready var btn_ok: Button = $VBoxContainer/HBoxContainer/VBoxContainer/btn_ok
onready var lbl_filename_header: Label = $VBoxContainer/HBoxContainer/VBoxContainer/lbl_filename
onready var lbl_filename_content: Label = $VBoxContainer/HBoxContainer/VBoxContainer/mc/filename
onready var lbl_filetype_header: Label = $VBoxContainer/HBoxContainer/VBoxContainer/lbl_type
onready var lbl_filetype_content: Label = $VBoxContainer/HBoxContainer/VBoxContainer/mc2/filetype
onready var lbl_filesize_header: Label = $VBoxContainer/HBoxContainer/VBoxContainer/lbl_size
onready var lbl_filesize_content: Label = $VBoxContainer/HBoxContainer/VBoxContainer/mc3/filesize
onready var lbl_fileinfo_header: Label = $VBoxContainer/HBoxContainer/VBoxContainer/lbl_info
onready var lbl_fileinfo_content: Label = $VBoxContainer/HBoxContainer/VBoxContainer/mc4/fileinfo
const inactive_modulate_color := Color(0.5, 0.5, 0.5)
const active_modulate_color := Color.white
var last_dir := ''
static func get_human_size(n: int) -> String:
if n > 0x100000000:
return '%.2f GiB' % (n/float(0x40000000))
if n > 0x400000:
return '%.1f MiB' % (n/float(0x100000))
if n > 0x1000:
return '%.1f KiB' % (n/float(0x400))
return '%d B' % n
static func tokenize_path(path: String) -> PoolStringArray:
return path.rstrip('/').split('/', true) # Allow empty means '/' will return ['']
static func join_path(tokens) -> String:
var path = '/'.join(tokens)
return path if path else '/'
func update_view():
var error = dir.list_dir_begin(true, true)
if error != OK:
print_debug(error)
return
for child in folder_buttons.get_children():
child.queue_free()
var path = ProjectSettings.globalize_path(dir.get_current_dir())
var cur_path: Array = tokenize_path(path)
var l := len(cur_path)
for i in l:
var btn := Button.new()
btn.connect('pressed', self, 'activate_entry', [join_path(cur_path.slice(0, i))])
btn.text = cur_path[i]
folder_buttons.add_child(btn)
if i == 0:
btn.text += '/'
if i == l-1:
btn.icon = FOLDER_ICON
if last_dir.begins_with(path): # i.e. we jumped up a level or more
var last_path: Array = tokenize_path(last_dir)
var l2 := len(last_path)
for i in range(l, l2):
var btn := Button.new()
btn.connect('pressed', self, 'activate_entry', [join_path(last_path.slice(0, i))])
btn.text = last_path[i]
btn.modulate = inactive_modulate_color
folder_buttons.add_child(btn)
# Doesn't work thanks to frame delay
# folder_buttons_scroller.set_h_scroll(9999)
# folder_buttons_scroller.ensure_control_visible(folder_buttons_scroller.get_h_scrollbar())
# folder_buttons_scroller.get_h_scrollbar().update()
itemlist.clear()
var directories := PoolStringArray()
var current_dir := dir.get_current_dir()
var files := []
while true:
var entry := dir.get_next()
if len(entry) == 0:
break
var filename := current_dir + '/' + entry
if filename in cached_cd_bin_paths:
files.append([entry, EXT_ICONS.iso])
elif dir.current_is_dir():
directories.append(entry)
elif '.' in entry:
var extension = entry.rsplit('.', true, 1)[1]
if extension in ALLOWED_EXTS:
files.append([entry, EXT_ICONS.get(extension)])
directories.sort()
files.sort()
for entry in directories:
itemlist.add_item(entry, FOLDER_ICON)
for entry in files:
itemlist.add_item(entry[0], entry[1])
func activate_entry(entry: String, _index: int = -1):
var curr_dir := dir.get_current_dir()
if not self.last_dir.begins_with(curr_dir):
self.last_dir = curr_dir
if dir.dir_exists(entry):
var error := dir.change_dir(entry)
if error == OK:
update_view()
else:
print_debug(error)
elif dir.file_exists(entry):
pass # Load the file
func view_entry(entry: String, index: int = -1):
init_labels()
lbl_filename_header.modulate = active_modulate_color
lbl_filename_content.text = entry
lbl_filetype_header.modulate = active_modulate_color
if dir.dir_exists(entry):
lbl_filetype_content.text = 'Folder'
elif dir.file_exists(entry):
var file := File.new()
var filename := dir.get_current_dir() + '/' + entry
var error := file.open(filename, File.READ)
if error != OK:
print_debug(error)
return
var size = file.get_len()
var human_size = get_human_size(size)
var ext = entry.rsplit('.', true, 1)[1].to_lower()
var type = TYPE_DESCS.get(ext, 'Unknown')
var prodcode := ''
if ext in CD_EXTS:
var cd := RomLoader.loader_cd_image.new(file)
for key in cd.directory:
var s = key.trim_prefix('./')
var re_match := RomLoader.psx_productcode_regex.search(s)
if re_match:
prodcode = '%s\n%s\n%s' % [cd.pvd.system_identifier.strip_edges(), cd.pvd.volume_identifier.strip_edges(), re_match.get_string(1)]
type = 'PSX CD-ROM Image'
cached_cd_bin_paths[filename] = prodcode
if index >= 0:
itemlist.set_item_icon(index, EXT_ICONS.iso)
break
var info = ('Info:\n %s' % prodcode) if prodcode else ''
lbl_filetype_content.text = type
lbl_filesize_header.modulate = active_modulate_color
lbl_filesize_content.text = human_size
lbl_fileinfo_header.modulate = active_modulate_color if prodcode else inactive_modulate_color
lbl_fileinfo_content.text = prodcode
func init_labels():
lbl_filename_header.modulate = inactive_modulate_color
lbl_filename_content.text = ''
lbl_filetype_header.modulate = inactive_modulate_color
lbl_filetype_content.text = ''
lbl_filesize_header.modulate = inactive_modulate_color
lbl_filesize_content.text = ''
lbl_fileinfo_header.modulate = inactive_modulate_color
lbl_fileinfo_content.text = ''
func _ready() -> void:
init_labels()
if OS.has_feature('windows'):
home_path = OS.get_environment('USERPROFILE')
else:
home_path = OS.get_environment('HOME')
# dir.open(OS.get_system_dir(OS.SYSTEM_DIR_DOWNLOADS))
var error = dir.open(home_path)
if error == OK:
update_view()
print(ProjectSettings.globalize_path('user://'))
print(ProjectSettings.globalize_path(dir.get_current_dir()))
func _on_ItemList_item_activated(index: int) -> void:
activate_entry(itemlist.get_item_text(index), index)
func _on_ItemList_item_selected(index: int) -> void:
view_entry(itemlist.get_item_text(index), index)