77 lines
3.1 KiB
Python
77 lines
3.1 KiB
Python
|
import json
|
||
|
|
import asyncio
|
||
|
|
import websockets
|
||
|
|
from obswebsocket import obsws, requests
|
||
|
|
|
||
|
|
def load_config(file_path):
|
||
|
|
try:
|
||
|
|
with open(file_path, 'r') as config_file:
|
||
|
|
config = json.load(config_file)
|
||
|
|
return config
|
||
|
|
except FileNotFoundError:
|
||
|
|
print(f"Config file not found at {file_path}.")
|
||
|
|
return None
|
||
|
|
|
||
|
|
async def listen_to_relay(your_pubkey, your_relay_url, obs_host, obs_port, obs_password):
|
||
|
|
uri = f"{your_relay_url}/ws"
|
||
|
|
|
||
|
|
# Create OBS connection
|
||
|
|
obs = obsws(obs_host, obs_port, obs_password)
|
||
|
|
obs.connect()
|
||
|
|
|
||
|
|
async with websockets.connect(uri) as websocket:
|
||
|
|
# Subscribe to all kind 3 events
|
||
|
|
subscription_id = "kind_3_events" # Replace with a unique identifier
|
||
|
|
await websocket.send(json.dumps(["REQ", subscription_id, {"kinds": [3]}]))
|
||
|
|
|
||
|
|
while True:
|
||
|
|
response = await websocket.recv()
|
||
|
|
await handle_event(response, your_pubkey, obs) # Pass the OBS connection to handle_event
|
||
|
|
|
||
|
|
async def handle_event(event_json, your_pubkey, obs):
|
||
|
|
event = json.loads(event_json)
|
||
|
|
if event[0] == "EVENT":
|
||
|
|
event_data = event[2]
|
||
|
|
if event_data["kind"] == 3: # Follow list event
|
||
|
|
await handle_follow_list(event_data, your_pubkey, obs) # Pass the OBS connection to handle_follow_list
|
||
|
|
|
||
|
|
|
||
|
|
async def handle_follow_list(follow_list_event, your_pubkey, obs):
|
||
|
|
if "tags" in follow_list_event:
|
||
|
|
# Extract the list of "p" tags from the follow list event
|
||
|
|
p_tags = [tag for tag in follow_list_event["tags"] if tag[0] == "p"]
|
||
|
|
|
||
|
|
# Check if your pubkey is the last entry in the "p" tags
|
||
|
|
if p_tags and p_tags[-1][1] == your_pubkey:
|
||
|
|
print(f"You have a new follower: {follow_list_event.get('pubkey', '')}")
|
||
|
|
|
||
|
|
# Toggle the specified source group in OBS
|
||
|
|
source_group_name = "heyguys" # Replace with the name of your source group
|
||
|
|
scene_name = "Escape From Tarkov" # Replace with the name of your scene
|
||
|
|
await toggle_source_group(obs, scene_name, source_group_name)
|
||
|
|
|
||
|
|
# Modify the toggle_source_group function to accept the scene name
|
||
|
|
async def toggle_source_group(obs, scene_name, source_group_name):
|
||
|
|
# Enable the source group in the specified scene
|
||
|
|
obs_response_enable = obs.call(requests.SetSceneItemProperties(item=source_group_name, scene=scene_name, visible=True))
|
||
|
|
print(f"Enable Response: {obs_response_enable}")
|
||
|
|
|
||
|
|
# Wait for a certain interval (e.g., 5 seconds)
|
||
|
|
await asyncio.sleep(5)
|
||
|
|
|
||
|
|
# Disable the source group in the specified scene
|
||
|
|
obs_response_disable = obs.call(requests.SetSceneItemProperties(item=source_group_name, scene=scene_name, visible=False))
|
||
|
|
print(f"Disable Response: {obs_response_disable}")
|
||
|
|
|
||
|
|
print(f"Toggled source group '{source_group_name}' in scene '{scene_name}'.")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
config_path = 'config.json'
|
||
|
|
config = load_config(config_path)
|
||
|
|
|
||
|
|
if config:
|
||
|
|
asyncio.get_event_loop().run_until_complete(listen_to_relay(
|
||
|
|
config["your_pubkey"], config["your_relay_url"],
|
||
|
|
config["obs_host"], config["obs_port"], config["obs_password"]
|
||
|
|
))
|