import logging
import contextlib

from ..orchestra import run

log = logging.getLogger(__name__)

@contextlib.contextmanager
def task(ctx, config):
    """
    Sync or skew clock

    This will initially sync the clocks.  Eventually it should let us also
    skew by some number of seconds.

    example:

    tasks:
    - clock:
    - ceph:
    - interactive:

    to sync.

    """

    log.info('Syncing clocks and checking initial clock skew...')
    for rem in ctx.cluster.remotes.iterkeys():
        rem.run(
            args=[
                'sudo',
                'service', 'ntp', 'stop',
                run.Raw(';'),
                'sudo',
                'ntpdate',
                'clock1.dreamhost.com',
                'clock2.dreamhost.com',
                'clock3.dreamhost.com',
                run.Raw(';'),
                'sudo',
                'service', 'ntp', 'start',
                run.Raw(';'),
                'ntpdc', '-p',
                ],
            logger=log.getChild(rem.name),
        )

    try:
        yield

    finally:
        log.info('Checking final clock skew...')
        for rem in ctx.cluster.remotes.iterkeys():
            rem.run(
                args=[
                    'ntpdc', '-p',
                    ],
                logger=log.getChild(rem.name),
                )