from fabric.api import run,env
from fabric.operations import get
import os
import yaml
import re

def _all_hosts():
   return filter(lambda fn: '.' in fn and not fn.startswith('.') and os.path.isdir(fn),os.listdir("."))

def _roledefs():
   rules = dict()

   rules_file = "cosmos-rules.yaml";
   if os.path.exists(rules_file):
      with open(rules_file) as fd:
         rules.update(yaml.load(fd))

   roles = dict() 
   for node_name in _all_hosts():
      for reg,cls in rules.iteritems():
         if re.search(reg,node_name):
            for cls_name in cls.keys():
               h = roles.get(cls_name,[])
               h.append(node_name)
               roles[cls_name] = h
   return roles

env.user = 'root'
env.timeout = 30
env.connection_attempts = 3
env.warn_only = True
env.skip_bad_hosts = True
env.roledefs = _roledefs()

#print repr(env.roledefs)

def all():
    env.hosts = _all_hosts()

def cosmos():
    run("cosmos update && cosmos apply");

def upgrade():
    run("apt-get -qq update && apt-get -y -q dist-upgrade");

def facts():
    get("/var/run/facts.yaml",local_path="facts/%(host)s.yaml")

def chassis():
    run("ipmi-chassis --get-chassis-status")

def newvm(fqdn,ip,domain):
    run("vmbuilder kvm ubuntu --domain %s --dest /var/lib/libvirt/images/%s.img --arch x86_64 --hostname %s --mem 512 --ip %s --addpkg openssh-server" % (domain,fqdn,fqdn,ip))