policy-bluetooth.lua: Adjust style

This commit is contained in:
Ashok Sidipotu 2022-07-20 08:22:17 +05:30 committed by Julian Bouzas
parent da2ac1b7a1
commit 1f77be36e5

View file

@ -36,8 +36,8 @@ local INVALID = -1
local timeout_source = nil
local restore_timeout_source = nil
local state = use_persistent_storage and State("policy-bluetooth") or nil
local headset_profiles = state and state:load() or {}
local state = use_persistent_storage and State ("policy-bluetooth") or nil
local headset_profiles = state and state:load () or {}
local last_profiles = {}
local active_streams = {}
@ -49,7 +49,7 @@ json = Json.Raw (apps_setting_str)
apps_setting = json:parse ()
for i = 1, #apps_setting do
applications[apps_setting[i]] = true
applications [apps_setting [i]] = true
end
metadata_om = ObjectManager {
@ -75,8 +75,8 @@ streams_om = ObjectManager {
}
}
local function parseParam(param_to_parse, id)
local param = param_to_parse:parse()
local function parseParam (param_to_parse, id)
local param = param_to_parse:parse ()
if param.pod_type == "Object" and param.object_id == id then
return param.properties
else
@ -84,70 +84,70 @@ local function parseParam(param_to_parse, id)
end
end
local function storeAfterTimeout()
local function storeAfterTimeout ()
if not use_persistent_storage then
return
end
if timeout_source then
timeout_source:destroy()
timeout_source:destroy ()
end
timeout_source = Core.timeout_add(1000, function ()
local saved, err = state:save(headset_profiles)
timeout_source = Core.timeout_add (1000, function ()
local saved, err = state:save (headset_profiles)
if not saved then
Log.warning(err)
Log.warning (err)
end
timeout_source = nil
end)
end
local function saveHeadsetProfile(device, profile_name)
local key = "saved-headset-profile:" .. device.properties["device.name"]
headset_profiles[key] = profile_name
storeAfterTimeout()
local function saveHeadsetProfile (device, profile_name)
local key = "saved-headset-profile:" .. device.properties ["device.name"]
headset_profiles [key] = profile_name
storeAfterTimeout ()
end
local function getSavedHeadsetProfile(device)
local key = "saved-headset-profile:" .. device.properties["device.name"]
return headset_profiles[key]
local function getSavedHeadsetProfile (device)
local key = "saved-headset-profile:" .. device.properties ["device.name"]
return headset_profiles [key]
end
local function saveLastProfile(device, profile_name)
last_profiles[device.properties["device.name"]] = profile_name
local function saveLastProfile (device, profile_name)
last_profiles [device.properties ["device.name"]] = profile_name
end
local function getSavedLastProfile(device)
return last_profiles[device.properties["device.name"]]
local function getSavedLastProfile (device)
return last_profiles [device.properties ["device.name"]]
end
local function isSwitched(device)
return getSavedLastProfile(device) ~= nil
local function isSwitched (device)
return getSavedLastProfile (device) ~= nil
end
local function isBluez5AudioSink(sink_name)
if sink_name and string.find(sink_name, "bluez_output.") ~= nil then
local function isBluez5AudioSink (sink_name)
if sink_name and string.find (sink_name, "bluez_output.") ~= nil then
return true
end
return false
end
local function isBluez5DefaultAudioSink()
local metadata = metadata_om:lookup()
local default_audio_sink = metadata:find(0, "default.audio.sink")
return isBluez5AudioSink(default_audio_sink)
local function isBluez5DefaultAudioSink ()
local metadata = metadata_om:lookup ()
local default_audio_sink = metadata:find (0, "default.audio.sink")
return isBluez5AudioSink (default_audio_sink)
end
local function findProfile(device, index, name)
for p in device:iterate_params("EnumProfile") do
local profile = parseParam(p, "EnumProfile")
local function findProfile (device, index, name)
for p in device:iterate_params ("EnumProfile") do
local profile = parseParam (p, "EnumProfile")
if not profile then
goto skip_enum_profile
end
Log.debug("Profile name: " .. profile.name .. ", priority: "
.. tostring(profile.priority) .. ", index: " .. tostring(profile.index))
Log.debug ("Profile name: " .. profile.name .. ", priority: "
.. tostring (profile.priority) .. ", index: " .. tostring (profile.index))
if (index ~= nil and profile.index == index) or
(name ~= nil and profile.name == name) then
(name ~= nil and profile.name == name) then
return profile.priority, profile.index, profile.name
end
@ -157,9 +157,9 @@ local function findProfile(device, index, name)
return INVALID, INVALID, nil
end
local function getCurrentProfile(device)
for p in device:iterate_params("Profile") do
local profile = parseParam(p, "Profile")
local function getCurrentProfile (device)
for p in device:iterate_params ("Profile") do
local profile = parseParam (p, "Profile")
if profile then
return profile.name
end
@ -168,13 +168,13 @@ local function getCurrentProfile(device)
return nil
end
local function highestPrioProfileWithInputRoute(device)
local function highestPrioProfileWithInputRoute (device)
local profile_priority = INVALID
local profile_index = INVALID
local profile_name = nil
for p in device:iterate_params("EnumRoute") do
local route = parseParam(p, "EnumRoute")
for p in device:iterate_params ("EnumRoute") do
local route = parseParam (p, "EnumRoute")
-- Parse pod
if not route then
goto skip_enum_route
@ -184,12 +184,12 @@ local function highestPrioProfileWithInputRoute(device)
goto skip_enum_route
end
Log.debug("Route with index: " .. tostring(route.index) .. ", direction: "
Log.debug ("Route with index: " .. tostring (route.index) .. ", direction: "
.. route.direction .. ", name: " .. route.name .. ", description: "
.. route.description .. ", priority: " .. route.priority)
if route.profiles then
for _, v in pairs(route.profiles) do
local priority, index, name = findProfile(device, v)
for _, v in pairs (route.profiles) do
local priority, index, name = findProfile (device, v)
if priority ~= INVALID then
if profile_priority < priority then
profile_priority = priority
@ -206,11 +206,11 @@ local function highestPrioProfileWithInputRoute(device)
return profile_priority, profile_index, profile_name
end
local function hasProfileInputRoute(device, profile_index)
for p in device:iterate_params("EnumRoute") do
local route = parseParam(p, "EnumRoute")
local function hasProfileInputRoute (device, profile_index)
for p in device:iterate_params ("EnumRoute") do
local route = parseParam (p, "EnumRoute")
if route and route.direction == "Input" and route.profiles then
for _, v in pairs(route.profiles) do
for _, v in pairs (route.profiles) do
if v == profile_index then
return true
end
@ -220,36 +220,36 @@ local function hasProfileInputRoute(device, profile_index)
return false
end
local function switchProfile()
local function switchProfile ()
local index
local name
if restore_timeout_source then
restore_timeout_source:destroy()
restore_timeout_source:destroy ()
restore_timeout_source = nil
end
for device in devices_om:iterate() do
if isSwitched(device) then
for device in devices_om:iterate () do
if isSwitched (device) then
goto skip_device
end
local cur_profile_name = getCurrentProfile(device)
saveLastProfile(device, cur_profile_name)
local cur_profile_name = getCurrentProfile (device)
saveLastProfile (device, cur_profile_name)
_, index, name = findProfile(device, nil, cur_profile_name)
if hasProfileInputRoute(device, index) then
Log.info("Current profile has input route, not switching")
_, index, name = findProfile (device, nil, cur_profile_name)
if hasProfileInputRoute (device, index) then
Log.info ("Current profile has input route, not switching")
goto skip_device
end
local saved_headset_profile = getSavedHeadsetProfile(device)
local saved_headset_profile = getSavedHeadsetProfile (device)
index = INVALID
if saved_headset_profile then
_, index, name = findProfile(device, nil, saved_headset_profile)
_, index, name = findProfile (device, nil, saved_headset_profile)
end
if index == INVALID then
_, index, name = highestPrioProfileWithInputRoute(device)
_, index, name = highestPrioProfileWithInputRoute (device)
end
if index ~= INVALID then
@ -258,34 +258,34 @@ local function switchProfile()
index = index
}
Log.info("Setting profile of '"
.. device.properties["device.description"]
Log.info ("Setting profile of '"
.. device.properties ["device.description"]
.. "' from: " .. cur_profile_name
.. " to: " .. name)
device:set_params("Profile", pod)
device:set_params ("Profile", pod)
else
Log.warning("Got invalid index when switching profile")
Log.warning ("Got invalid index when switching profile")
end
::skip_device::
end
end
local function restoreProfile()
for device in devices_om:iterate() do
if isSwitched(device) then
local profile_name = getSavedLastProfile(device)
local cur_profile_name = getCurrentProfile(device)
local function restoreProfile ()
for device in devices_om:iterate () do
if isSwitched (device) then
local profile_name = getSavedLastProfile (device)
local cur_profile_name = getCurrentProfile (device)
saveLastProfile(device, nil)
saveLastProfile (device, nil)
if cur_profile_name then
Log.info("Setting saved headset profile to: " .. cur_profile_name)
saveHeadsetProfile(device, cur_profile_name)
Log.info ("Setting saved headset profile to: " .. cur_profile_name)
saveHeadsetProfile (device, cur_profile_name)
end
if profile_name then
local _, index, name = findProfile(device, nil, profile_name)
local _, index, name = findProfile (device, nil, profile_name)
if index ~= INVALID then
local pod = Pod.Object {
@ -293,29 +293,29 @@ local function restoreProfile()
index = index
}
Log.info("Restoring profile of '"
.. device.properties["device.description"]
Log.info ("Restoring profile of '"
.. device.properties ["device.description"]
.. "' from: " .. cur_profile_name
.. " to: " .. name)
device:set_params("Profile", pod)
device:set_params ("Profile", pod)
else
Log.warning("Failed to restore profile")
Log.warning ("Failed to restore profile")
end
end
end
end
end
local function triggerRestoreProfile()
local function triggerRestoreProfile ()
if restore_timeout_source then
return
end
if next(active_streams) ~= nil then
if next (active_streams) ~= nil then
return
end
restore_timeout_source = Core.timeout_add(profile_restore_timeout_msec, function ()
restore_timeout_source = Core.timeout_add (profile_restore_timeout_msec, function ()
restore_timeout_source = nil
restoreProfile()
restoreProfile ()
end)
end
@ -323,83 +323,83 @@ end
-- media.role set to Communication in props or it is in our list of
-- applications as these applications do not set media.role correctly or at
-- all.
local function checkStreamStatus(stream)
local app_name = stream.properties["application.name"]
local stream_role = stream.properties["media.role"]
local function checkStreamStatus (stream)
local app_name = stream.properties ["application.name"]
local stream_role = stream.properties ["media.role"]
if not (stream_role == "Communication" or applications[app_name]) then
if not (stream_role == "Communication" or applications [app_name]) then
return false
end
if not isBluez5DefaultAudioSink() then
if not isBluez5DefaultAudioSink () then
return false
end
-- If a stream we previously saw stops running, we consider it
-- inactive, because some applications (Teams) just cork input
-- streams, but don't close them.
if previous_streams[stream["bound-id"]] and stream.state ~= "running" then
if previous_streams [stream ["bound-id"]] and stream.state ~= "running" then
return false
end
return true
end
local function handleStream(stream)
local function handleStream (stream)
if not use_headset_profile then
return
end
if checkStreamStatus(stream) then
active_streams[stream["bound-id"]] = true
previous_streams[stream["bound-id"]] = true
switchProfile()
if checkStreamStatus (stream) then
active_streams [stream ["bound-id"]] = true
previous_streams [stream ["bound-id"]] = true
switchProfile ()
else
active_streams[stream["bound-id"]] = nil
triggerRestoreProfile()
active_streams [stream ["bound-id"]] = nil
triggerRestoreProfile ()
end
end
local function handleAllStreams()
local function handleAllStreams ()
for stream in streams_om:iterate {
Constraint { "media.class", "matches", "Stream/Input/Audio", type = "pw-global" },
Constraint { "stream.monitor", "!", "true" }
} do
handleStream(stream)
handleStream (stream)
end
end
streams_om:connect("object-added", function (_, stream)
stream:connect("state-changed", function (stream, old_state, cur_state)
handleStream(stream)
streams_om:connect ("object-added", function (_, stream)
stream:connect ("state-changed", function (stream, old_state, cur_state)
handleStream (stream)
end)
stream:connect("params-changed", handleStream)
handleStream(stream)
stream:connect ("params-changed", handleStream)
handleStream (stream)
end)
streams_om:connect("object-removed", function (_, stream)
active_streams[stream["bound-id"]] = nil
previous_streams[stream["bound-id"]] = nil
triggerRestoreProfile()
streams_om:connect ("object-removed", function (_, stream)
active_streams [stream ["bound-id"]] = nil
previous_streams [stream ["bound-id"]] = nil
triggerRestoreProfile ()
end)
devices_om:connect("object-added", function (_, device)
devices_om:connect ("object-added", function (_, device)
-- Devices are unswitched initially
if isSwitched(device) then
saveLastProfile(device, nil)
if isSwitched (device) then
saveLastProfile (device, nil)
end
handleAllStreams()
handleAllStreams ()
end)
metadata_om:connect("object-added", function (_, metadata)
metadata:connect("changed", function (m, subject, key, t, value)
metadata_om:connect ("object-added", function (_, metadata)
metadata:connect ("changed", function (m, subject, key, t, value)
if (use_headset_profile and subject == 0 and key == "default.audio.sink"
and isBluez5AudioSink(value)) then
and isBluez5AudioSink (value)) then
-- If bluez sink is set as default, rescan for active input streams
handleAllStreams()
handleAllStreams ()
end
end)
end)
metadata_om:activate()
devices_om:activate()
streams_om:activate()
metadata_om:activate ()
devices_om:activate ()
streams_om:activate ()