#!/usr/bin/env python

# This program is free software. It comes without any warranty, to
# the extent permitted by applicable law. You can redistribute it
# and/or modify it under the terms of the Do What The Fuck You Want
# To Public License, Version 2, as published by Sam Hocevar. See
# http://sam.zoy.org/wtfpl/COPYING for more details.

import time
import os
import sys
import urllib
import ftplib
import smtplib
import yaml
import subprocess
import tempfile
from termcolor import colored, cprint

if len(sys.argv) < 2 or not os.path.exists(sys.argv[1]):
    fh = open(os.path.join(os.getenv('HOME'), '.simpleton.yaml'))

services = yaml.load(fh)
tests, passed, started = (0, 0, time.time())

for k, v in services.items():
    for u in v:
        tests += 1
        msg = '* testing %s: %s' % (k, u)
        out = colored(msg, 'green')
        try:
            if k == 'HTTP':
                urllib.urlopen(u)
            if k == 'FTP':
                ftplib.FTP(u)
            if k == 'SMTP':
                smtplib.SMTP(u)
            if k == 'AFP':
                tmp_path = tempfile.mkdtemp()
                subprocess.check_call(['/sbin/mount_afp', '-o', 'nobrowse', u, tmp_path])
                subprocess.check_call(['/sbin/umount', tmp_path])
                os.rmdir(tmp_path)

            passed += 1
        except Exception, e:
            print e
            out = colored(msg, 'red')

        print out

print '%d of %d tests passed in %d seconds' % (passed, tests, int(time.time() - started))