#!/usr/bin/env python2
#Print commands to delete load balancers, NICs, and IPs not attached to any VM
#Plenty of bugs here.  Assumes a stock setup with nothing fancy.
import json
import subprocess

def query(command):
  return json.load(subprocess.Popen(command, stdout=subprocess.PIPE).stdout)

lb = query(["az", "network", "lb", "list", "-o", "json"])
vmss = query(["az", "vmss", "list", "-o", "json"])

in_use_lbs = []
for m in vmss:
  balancer = m["virtualMachineProfile"]["networkProfile"]["networkInterfaceConfigurations"][0]["ipConfigurations"][0]["loadBalancerBackendAddressPools"]
  if balancer:
    in_use_lbs.append('/'.join(balancer[0]["id"].split('/')[0:9]))
 
allocated_lbs = ['/'.join(l["frontendIpConfigurations"][0]["id"].split('/')[0:9]) for l in lb]

unused_lbs = set(allocated_lbs) - set(in_use_lbs)
for l in unused_lbs:
  split = l.split('/')
  print "az network lb delete -g " +  split[4] +  " -n " + split[8]

#TODO: exclude the lbs that are to be deleted
ip_used_by_lb = [l["frontendIpConfigurations"][0]["publicIpAddress"]["id"] for l in lb]


vm = query(["az", "vm", "list", "-o", "json"])
in_use_nics = [v["networkProfile"]["networkInterfaces"][0]["id"] for v in vm]
nics = query(["az", "network", "nic", "list", "-o", "json"])
allocated_nics = [n["id"] for n in nics]
for n in set(allocated_nics) - set(in_use_nics):
  split = n.split('/')
  print "az network nic delete -g " + split[4] + " -n " + split[8]

#TODO: exclude the nics that are to be deleted
ip_used_by_nic = [n["ipConfigurations"][0]["publicIpAddress"]["id"] for n in nics]
in_use_ips = ip_used_by_lb + ip_used_by_nic

allocated_ips = [i["id"] for i in query(["az", "network", "public-ip", "list", "-o", "json"])]
for i in set(allocated_ips) - set(in_use_ips):
  split = i.split('/')
  print "az network public-ip delete -g " + split[4] + " -n " + split[8]

nsg_all = [n['id'] for n in query(["az", "network", "nsg", "list", "-o", "json"])]
for i in set(nsg_all) - set([n['networkSecurityGroup']['id'] for n in nics]):
  split = i.split('/')
  print "az network nsg delete -g " + split[4] + " -n " + split[8]