assign client updater methods as base lambdas

This commit is contained in:
onyx-and-iris 2023-07-26 10:51:38 +01:00
parent 18d291c6eb
commit 9940fbbf9f
3 changed files with 22 additions and 22 deletions

View File

@ -2,7 +2,6 @@ require "socket"
require "websocket/driver" require "websocket/driver"
require "digest/sha2" require "digest/sha2"
require "json" require "json"
require "observer"
require "waitutil" require "waitutil"
require_relative "mixin" require_relative "mixin"
@ -23,10 +22,10 @@ module OBSWS
end end
class Base class Base
include Observable
include Mixin::OPCodes include Mixin::OPCodes
attr_reader :id, :driver, :closed attr_reader :closed
attr_writer :updater
def initialize(**kwargs) def initialize(**kwargs)
host = kwargs[:host] || "localhost" host = kwargs[:host] || "localhost"
@ -56,7 +55,7 @@ module OBSWS
) { @identified } ) { @identified }
end end
def start_driver private def start_driver
Thread.new do Thread.new do
@driver.start @driver.start
@ -68,6 +67,12 @@ module OBSWS
end end
end end
public def stop_driver
@driver.close
end
private
def auth_token(salt:, challenge:) def auth_token(salt:, challenge:)
Digest::SHA256.base64digest( Digest::SHA256.base64digest(
Digest::SHA256.base64digest(@password + salt) + challenge Digest::SHA256.base64digest(@password + salt) + challenge
@ -99,12 +104,11 @@ module OBSWS
when Mixin::OPCodes::IDENTIFIED when Mixin::OPCodes::IDENTIFIED
@identified = true @identified = true
when Mixin::OPCodes::EVENT, Mixin::OPCodes::REQUESTRESPONSE when Mixin::OPCodes::EVENT, Mixin::OPCodes::REQUESTRESPONSE
changed @updater.call(data[:op], data[:d])
notify_observers(data[:op], data[:d])
end end
end end
def req(id, type_, data = nil) public def req(id, type_, data = nil)
payload = { payload = {
op: Mixin::OPCodes::REQUEST, op: Mixin::OPCodes::REQUEST,
d: { d: {

View File

@ -70,20 +70,18 @@ module OBSWS
kwargs[:subs] ||= SUBS::LOW_VOLUME kwargs[:subs] ||= SUBS::LOW_VOLUME
@base_client = Base.new(**kwargs) @base_client = Base.new(**kwargs)
LOGGER.info("#{self} succesfully identified with server") LOGGER.info("#{self} succesfully identified with server")
@base_client.add_observer(self) @base_client.updater = ->(op_code, data) {
if op_code == Mixin::OPCodes::EVENT
event = data[:eventType]
data = data.fetch(:eventData, {})
notify_observers(event, Mixin::Data.new(data, data.keys))
end
}
end end
def to_s def to_s
self.class.name.split("::").last(2).join("::") self.class.name.split("::").last(2).join("::")
end end
def update(op_code, data)
if op_code == Mixin::OPCodes::EVENT
event = data[:eventType]
data = data.key?(:eventData) ? data[:eventData] : {}
notify_observers(event, Mixin::Data.new(data, data.keys))
end
end
end end
end end
end end

View File

@ -15,7 +15,9 @@ module OBSWS
def initialize(**kwargs) def initialize(**kwargs)
@base_client = Base.new(**kwargs) @base_client = Base.new(**kwargs)
LOGGER.info("#{self} succesfully identified with server") LOGGER.info("#{self} succesfully identified with server")
@base_client.add_observer(self) @base_client.updater = ->(op_code, data) {
@response = data if op_code == Mixin::OPCodes::REQUESTRESPONSE
}
@response = {requestId: 0} @response = {requestId: 0}
end end
@ -26,7 +28,7 @@ module OBSWS
def run def run
yield(self) yield(self)
ensure ensure
close stop_driver
WaitUtil.wait_for_condition( WaitUtil.wait_for_condition(
"driver to close", "driver to close",
delay_sec: 0.01, delay_sec: 0.01,
@ -34,10 +36,6 @@ module OBSWS
) { @base_client.closed } ) { @base_client.closed }
end end
def update(op_code, data)
@response = data if op_code == Mixin::OPCodes::REQUESTRESPONSE
end
def call(req, data = nil) def call(req, data = nil)
id = rand(1..1000) id = rand(1..1000)
@base_client.req(id, req, data) @base_client.req(id, req, data)