# -*- coding: utf-8 -*-
#
# Take desktop/mobile shots of all themes from publik-base-theme.
#
# Usage:
#  --directory DIRECTORY    -- directory to store shots, defaults to shots/
#  --no-headless            -- run in an actual browser window
#  --no-memcached           -- do not restart memcached after theme changes
#  --overlay OVERLAY        -- limit to given overlay ("none" to exclude overlays)
#  --reshot                 -- retake existing shots
#  --timeout TIMEOUT        -- timeout between shots (time for hobo deploy)
#  --cell-timeout TIMEOUT   -- timeout after url loading (time for ajax cells loading)
#  --theme THEME            -- theme to shot (substring match)
#  URL                      -- specific URL to shot
#
# Example:
#  hobo-manage tenant_command runscript shot_them_all.py -d hobo.fred.local.0d.be
#

import argparse
import os
import time
from selenium import webdriver
from hobo.theme.utils import get_selected_theme, get_themes, set_theme
from hobo.deploy.signals import notify_agents
import sys

parser = argparse.ArgumentParser()
parser.add_argument('--directory', dest='directory', type=str, default='shots')
parser.add_argument('--no-headless', dest='no_headless', action='store_true')
parser.add_argument('--no-memcached', dest='no_memcached', action='store_true')
parser.add_argument('--overlay', dest='overlay', type=str, default='none')
parser.add_argument('--reshot', dest='reshot', action='store_true')
parser.add_argument('--timeout', dest='timeout', type=int, default=20)  # seconds
parser.add_argument('--cell-timeout', dest='cell_timeout', type=int, default=0)  # seconds
parser.add_argument('--theme', dest='theme', type=str)
parser.add_argument('url', metavar='URL', type=str, nargs='?',
                    default='https://auquo.fred.local.0d.be/contactez-nous/inscription-sur-les-listes/')
args = parser.parse_args()

if not os.path.exists(args.directory):
    os.mkdir(args.directory)

options = webdriver.ChromeOptions()
if not args.no_headless:
    options.add_argument('headless')
browser = webdriver.Chrome(chrome_options=options)

for theme in get_themes():
    theme_id = theme['id']
    if args.overlay == 'all':
        pass
    elif args.overlay == 'none' and not theme.get('overlay'):
        pass
    elif args.overlay == theme.get('overlay'):
        pass
    else:
        continue
    if args.theme and args.theme not in theme['id']:
        continue
    sys.stderr.write("%-25s" % theme_id)
    shots = [
        [1400, 1000, 'desktop'],
        [360, 740, 'mobile'],  # Samsung Galaxy S8
        [740, 360, 'mobile-horizontal'],  # ditto, horizonal
    ]
    if not args.reshot:
        if all([os.path.exists(os.path.join(args.directory, '%s-%s.png' % (theme_id, x[2]))) for x in shots]):
            sys.stderr.write(u' 📷 \n')
            continue
    if get_selected_theme() != theme_id:
        set_theme(theme_id)
        notify_agents(None)
        for i in range(args.timeout):
            sys.stderr.write('.')
            time.sleep(1)
    if not args.no_memcached:
        os.system('sudo service memcached restart')
    browser.get(args.url)
    for i in range(args.cell_timeout):
        sys.stderr.write('.')
        time.sleep(1)

    def size(value):
        return browser.execute_script('return document.body.parentNode.scroll' + value)

    for shot in shots:
        browser.set_window_size(shot[0], shot[1])
        browser.set_window_size(shot[0], size('Height'))
        browser.save_screenshot(os.path.join(args.directory, '%s-%s.png' % (theme_id, shot[2])))

    sys.stderr.write(u' 📸 \n')

browser.close()