WIP Ractor Task stuff

This commit is contained in:
2026-01-29 21:44:54 -06:00
parent 4a8457e233
commit 6e79c4639d
5 changed files with 218 additions and 50 deletions

View File

@@ -3,6 +3,9 @@ class W3DHub
class Task
LOG_TAG = "W3DHub::ApplicationManager::Task".freeze
class AbortTaskExecutionError < StandardError
end
# Task failed
EVENT_FAILURE = -1
# Task started, show application taskbar
@@ -10,17 +13,15 @@ class W3DHub
# Task completed successfully
EVENT_SUCCESS = 1
# Task progress
EVENT_PROGRESS = 2
# List of packages this task will be working over
EVENT_PACKAGE_LIST = 3
# Update a package's status: verifying, downloading, unpacking, patching
EVENT_PACKAGE_STATUS = 4
EVENT_STATUS = 2
# Subtask progress
EVENT_STATUS_OPERATION = 3
Context = Data.define(
:task_id,
:app_type,
:app_id,
:channel_id,
:application,
:channel,
:version,
:target_path,
:temp_path
@@ -34,6 +35,7 @@ class W3DHub
)
attr_reader :context, :state
attr_accessor :status
def initialize(context:)
@context = context
@@ -43,6 +45,8 @@ class W3DHub
# remember all case insensitive file paths
@cache_file_paths = {}
@status = Status.new(application: context.application, channel: context.channel)
@failure = false
@failure_reason = ""
end
@@ -51,14 +55,13 @@ class W3DHub
end
def fail!(reason: "")
@state = :failed
@failure = true
@failure_reason = reason
Ractor.current.send(
MessageEvent.new(
context.task_id,
send_task_result
)
)
raise AbortTaskExecutionError, reason
end
def failed?
@@ -72,12 +75,20 @@ class W3DHub
end
def send_task_result
Ractor.current.send(
internal_send_message_event(
failed? ? EVENT_FAILURE : EVENT_SUCCESS,
nil,
failed? ? { type: :failure, title: "Task Failed", message: @failure_reason } : nil
)
end
def internal_send_message_event(type, subtype = nil, data = nil)
Ractor.yield(
MessageEvent.new(
context.task_id,
@failure ? EVENT_FAILURE : EVENT_SUCCESS,
nil,
@failure_reason
type,
subtype,
data
)
)
end
@@ -87,9 +98,27 @@ class W3DHub
end
def start
@state = :running
# only mark task as running then return unless we're NOT running on the
# main ractor. Task is deep copied when past to the ractor.
return if Ractor.main?
internal_send_message_event(EVENT_START)
sleep 1
execute_task
sleep 1
@state = failed? ? :failed : :complete
send_task_result
sleep 1
rescue StandardError => e
fail!(reason: "Fatal Error\n#{e}") unless e.is_a?(AbortTaskExecutionError)
end
# returns true on success and false on failure
@@ -102,14 +131,52 @@ class W3DHub
# Quick checks before network and computational work starts
def fail_fast!
# can read/write to destination
# is wine present?
if W3DHub.unix?
wine_present = W3DHub.command("which #{Store.settings[:wine_command]}")
fail!("FAIL FAST: Insufficient disk space available.") unless disk_space_available?
unless wine_present
fail!(reason: "FAIL FAST: `which #{Store.settings[:wine_command]}` command failed, wine is not installed.\n\n"\
"Will be unable to launch game.\n\n"\
"Check wine options in launcher's settings.")
end
end
# can read/write to destination
# TODO
# have enough disk space
fail!(reason: "FAIL FAST: Insufficient disk space available.") unless disk_space_available?
end
def fetch_manifests
result = CyberarmEngine::Result.new
def fetch_manifests(version)
manifests = []
result = Result.new
while (package_result = fetch_package(category, subcategory, version, "manifest.xml"))
break unless package_result.okay?
path = package_file_path(category, subcategory, version, "manifest.xml")
unless File.exist?(path) && !File.directory?(path)
result.error = RuntimeError.new("File missing: #{path}")
return result
end
manifest = LegacyManifest.new(path)
manifests << manifest
break unless manifest.patch?
version = manifest.base_version
end
# return in oldest to newest order
result.data = manifests.reverse
result
rescue StandardError => e # Async derives its errors from StandardError
result.error = e
result
end
@@ -180,8 +247,63 @@ class W3DHub
true
end
# returns JSON hash on success, false or nil on failure
def fetch_package_details(packages)
endpoint = "/apis/launcher/1/get-package-details"
result = Result.new
hash = {
packages: packages.map do |h|
{ category: h[:category], subcategory: h[:subcategory], name: h[:name], version: h[:version] }
end
}
body = URI.encode_www_form("data": JSON.dump(hash))
Sync do
Async::HTTP::Internet.post("#{UPSTREAM_ENDPOINT}#{endpoint}", CLIENT_FORM_ENCODED_HEADERS, body) do |response|
if response.success?
result.data = JSON.parse(response.read)
else
result.error = RuntimeError.new(response) # FIXME: have better error
end
rescue StandardError => e
result.error = e
end
end
result
end
def fetch_package(version, name)
result = CyberarmEngine::Result.new
endpoint = "/apis/launcher/1/get-package"
result = Result.new
path = package_file_path(category, subcategory, version, name)
headers = [
["user-agent", USER_AGENT],
["content-type", "application/x-www-form-urlencoded"],
["authorization", "Bearer #{FAKE_BEARER_TOKEN}"]
].freeze
body = URI.encode_www_form("data": JSON.dump({ category: category, subcategory: subcategory, name: name, version: version }))
Sync do
Async::HTTP::Internet.post("#{UPSTREAM_ENDPOINT}#{endpoint}", headers, body) do |response|
if response.success?
create_directories(path)
File.open(path, "wb") do |file|
response.each do |chunk|
file.write(chunk)
end
end
result.data = true
end
rescue StandardError => e
result.error = e
end
end
result
end
@@ -216,6 +338,14 @@ class W3DHub
result
end
def package_file_path(category, subcategory, version, name)
"#{PACKAGE_CACHE}/"\
"#{category}/"\
"#{subcategory.to_s.empty? ? "" : "#{subcategory}/"}"\
"#{version.to_s.empty? ? "" : "#{version}/"}"\
"#{name}"
end
end
end
end