#!/usr/bin/env python3

import sys, os, pwd
from subprocess import Popen, PIPE

from get import *
from set import *

args     = sys.argv[1:]
home     = os.environ['HOME']     if 'HOME'     in os.environ else pwd.getpwuid(os.getuid()).pw_dir
session_ = os.environ['SESSION_'] if 'SESSION_' in os.environ else ''
hostname = os.uname().nodename.lower()

t = lambda lopt, sopt : any(arg in args for arg in (lopt, '-' + lopt, '--' + lopt, sopt, '-' + sopt))

mirror  = t('mirror',  'm')
swap    = t('swap',    's')
tv      = t('tv',      't')
wide    = t('wide',    'w')
crt     = t('crt',     'c')
large   = t('large',   'l')
single  = t('single',  '1')
pretend = t('pretend', 'P')

[screen] = get_setup()
ok = False

if pretend:
    def apply_setup(display):
        print(repr(display.to_xrandr()))
        return True



### Configurations


if hostname == 'zenith':
    prime           = screen['DisplayPort-2' if not swap else 'DisplayPort-1']
    sec             = screen['DisplayPort-1' if not swap else 'DisplayPort-2']
    embed           = None

    prime_alt       = None
    sec_alt         = None

    prime.want_mode = '1920x1200'
    sec.want_mode   = '1920x1200'


else:
    print('%s: no configurations found for this machine' % sys.argv[0], file = sys.stderr)
    sys.exit(1)

if not prime.connected and prime_alt is not None:
    prime, prime_alt = prime_alt, prime
if not sec.connected and sec_alt is not None:
    sec, sec_alt = sec_alt, sec


if large:
    prime.want_mode = '1792x1344'
    sec.want_mode   = '1792x1344'

if crt:
    prime.want_mode = '800x600'
    sec.want_mode   = '800x600'

if wide:
    prime.want_mode = '1920x1080'
    sec.want_mode   = '1920x1080'

if tv:
    sec.want_mode   = '1920x1080'


if prime is not None:
    prime.want_rate = prime.best_rate(prime.want_mode)
if sec is not None:
    sec.want_rate = sec.best_rate(sec.want_mode)
if embed is not None:
    embed.want_rate = embed.best_rate(embed.want_mode)




if '+prime' in args:
    prime.connected = True
if '+sec' in args:
    sec.connected = True
if '+embed' in args:
    embed.connected = True

if '-prime' in args:
    prime.connected = False
if '-sec' in args:
    sec.connected = False
if '-embed' in args:
    embed.connected = False




### Apply

if prime.connected and sec.connected and not single:
    display = Display()
    ok = True

    output = Output(prime.name)
    display.outputs.append(output)
    output.mode = prime.want_mode
    output.rate = prime.want_rate
    output.primary = True
    output.relpos  = None
    output.relto   = None

    output = Output(sec.name)
    display.outputs.append(output)
    output.mode = sec.want_mode
    output.rate = sec.want_rate
    output.primary = False
    output.relpos  = 'left-of' if not mirror else 'same-as'
    output.relto   = prime.name

    if embed is not None:
        output = Output(embed.name)
        display.outputs.append(output)
        output.off = True


elif prime.connected:
    display = Display()
    ok = True

    output = Output(prime.name)
    display.outputs.append(output)
    output.mode = prime.want_mode
    output.rate = prime.want_rate
    output.primary = True

    output = Output(sec.name)
    display.outputs.append(output)
    output.off = True

    if embed is not None:
        output = Output(embed.name)
        display.outputs.append(output)
        output.off = True


elif sec.connected:
    display = Display()
    ok = True

    output = Output(sec.name)
    display.outputs.append(output)
    output.mode = sec.want_mode
    output.rate = sec.want_rate
    output.primary = True

    output = Output(prime.name)
    display.outputs.append(output)
    output.off = True

    if embed is not None:
        output = Output(embed.name)
        display.outputs.append(output)
        output.off = True


elif embed is None or not embed.connected:
    print('%s: don\'t know how to configure' % sys.argv[0], file = sys.stderr)


else:
    display = Display()
    ok = True

    output = Output(embed.name)
    display.outputs.append(output)
    output.mode = embed.want_mode
    output.rate = embed.want_rate
    output.primary = True

    output = Output(prime.name)
    display.outputs.append(output)
    output.off = True

    output = Output(sec.name)
    display.outputs.append(output)
    output.off = True

if ok:
    if prime_alt is not None:
        output = Output(prime_alt.name)
        display.outputs.append(output)
        output.off = True

    if sec_alt is not None:
        output = Output(sec_alt.name)
        display.outputs.append(output)
        output.off = True

    ok = apply_setup(display)



### Epilogue

if not ok:
    sys.exit(1)
if pretend:
    sys.exit(0)

[screen] = get_setup()
prime = screen[True]
if prime.position[0] > 0:
    print(prime.position[0], flush = True)

for file in ('%s/.config/background.%s' % (home, session_), '%s/.config/background' % home):
    if os.path.exists(file):
        try:
            os.execlp('xwallpaper', 'xwallpaper', '--zoom', file)
        except:
            pass