mirror of https://github.com/Horhik/dotfiles.git
148 lines
4.3 KiB
Python
Executable File
148 lines
4.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import socket
|
|
import json
|
|
import argparse
|
|
|
|
def get_property_cmd(property, request_id=None):
|
|
cmd = {'command': ['get_property', property]}
|
|
if isinstance(request_id, int):
|
|
cmd['request_id'] = request_id
|
|
return (json.dumps(cmd) + '\n').encode('utf-8')
|
|
|
|
def update_property(property, get_new_value):
|
|
client.send(get_property_cmd(property))
|
|
msg = client.recv(BUFSIZE).decode('utf-8')
|
|
value = get_new_value(msg)
|
|
cmd = {'command': ['set_property', property, value]}
|
|
client.send((json.dumps(cmd) + '\n').encode('utf-8'))
|
|
|
|
MPV_SOCKET = '/tmp/mpvsocket'
|
|
BUFSIZE = 1024
|
|
METADATA_CMD = get_property_cmd('filtered-metadata', 1)
|
|
OBSERVE_METADATA_CMD = b'{ "command": ["observe_property", 1, "filtered-metadata"] }\n'
|
|
MEDIA_TITLE_CMD = get_property_cmd('media-title', 2)
|
|
PERCENT_POS_CMD = get_property_cmd('percent-pos', 3)
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-t', '--truncate', metavar='N', type=int, help='truncate output to N characters')
|
|
parser.add_argument('-c', '--color', default='#fff', metavar='C', help='set color of underline (progress bar)')
|
|
parser.add_argument('-p', '--property', choices=['pause', 'time-pos', 'playlist-pos'], help='update mpv property and exit')
|
|
parser.add_argument('args', nargs=argparse.REMAINDER)
|
|
args = parser.parse_args()
|
|
|
|
if args.property:
|
|
try:
|
|
client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
client.connect(MPV_SOCKET)
|
|
except (FileNotFoundError, ConnectionRefusedError):
|
|
raise RuntimeError('Failed connecting to mpv socket.')
|
|
|
|
if args.property == 'pause':
|
|
update_property('pause', lambda msg: not json.loads(msg)['data'])
|
|
elif args.property in ['time-pos', 'playlist-pos']:
|
|
update_property(args.property, lambda msg: json.loads(msg)['data'] + int(args.args[0]))
|
|
|
|
client.close()
|
|
exit()
|
|
|
|
import time
|
|
import sys
|
|
from threading import Thread
|
|
|
|
def first_value(data, keys):
|
|
for key in keys:
|
|
value = data.get(key)
|
|
if value:
|
|
return value
|
|
|
|
def maybe_truncate(text):
|
|
limit = args.truncate
|
|
return (text[:limit - 1] + '…') if limit and len(text) > limit else text
|
|
|
|
def puts(output='', pos=0):
|
|
if pos > 0:
|
|
output = f'%{{u{args.color}}}%{{+u}}{output[0:pos]}%{{-u}}{output[pos:len(output)]}'
|
|
sys.stdout.write(output + '\n')
|
|
sys.stdout.flush()
|
|
|
|
def create_client():
|
|
client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
while True:
|
|
try:
|
|
client.connect(MPV_SOCKET)
|
|
client.send(METADATA_CMD)
|
|
client.send(OBSERVE_METADATA_CMD)
|
|
Thread(target=percent_pos_sender, args=[client]).start()
|
|
break
|
|
except (FileNotFoundError, ConnectionRefusedError):
|
|
time.sleep(1)
|
|
return client
|
|
|
|
def parse_msg(msg):
|
|
messages = msg.decode('utf-8').split('\n')
|
|
not_empty = filter(lambda s: s, messages)
|
|
parsed = map(lambda m: json.loads(m), not_empty)
|
|
return list(parsed)
|
|
|
|
def format_data(data):
|
|
artist = first_value(data, ['Artist', 'Album_Artist']) or ''
|
|
title = first_value(data, ['Title', 'icy-title']) or ''
|
|
not_empty = filter(lambda d: d, [artist.strip(), title.strip()])
|
|
return ' - '.join(not_empty)
|
|
|
|
def percent_pos_sender(client):
|
|
try:
|
|
while True:
|
|
time.sleep(1)
|
|
client.send(PERCENT_POS_CMD)
|
|
except BrokenPipeError:
|
|
pass
|
|
|
|
client = create_client()
|
|
current_output = ''
|
|
current_pos = 0
|
|
while True:
|
|
msg = client.recv(BUFSIZE)
|
|
|
|
# If connection is broken
|
|
if len(msg) == 0:
|
|
current_output = ''
|
|
current_pos = 0
|
|
puts()
|
|
time.sleep(1)
|
|
client = create_client()
|
|
continue
|
|
|
|
for m in parse_msg(msg):
|
|
data = m.get('data')
|
|
|
|
if m.get('event') == 'seek':
|
|
client.send(PERCENT_POS_CMD)
|
|
|
|
# 'filtered-metadata' event or response
|
|
elif (m.get('id') == 1 or m.get('request_id') == 1) and isinstance(data, dict):
|
|
text = format_data(data)
|
|
if text:
|
|
output = maybe_truncate(text)
|
|
if output != current_output:
|
|
current_output = output
|
|
puts(output)
|
|
client.send(PERCENT_POS_CMD)
|
|
else:
|
|
client.send(MEDIA_TITLE_CMD)
|
|
|
|
# 'media-title' response
|
|
elif m.get('request_id') == 2:
|
|
current_output = maybe_truncate(data)
|
|
puts(current_output)
|
|
client.send(PERCENT_POS_CMD)
|
|
|
|
# 'percent-pos' response
|
|
elif m.get('request_id') == 3 and isinstance(data, float):
|
|
length = len(current_output)
|
|
pos = round(length / 100 * data)
|
|
if pos != current_pos:
|
|
current_pos = pos
|
|
puts(current_output, pos)
|