#!/usr/bin/env python
# -*- coding: utf-8 -*-


import sys
import shutil
import Apt
import Interface
from gi.repository import Gtk


def main():

	run_gui=True
	
	if (len(sys.argv)>1):
		if sys.argv[1]=="--help":
			run_gui=False
			print ("Halp!")
		
		if sys.argv[1]=="--list":
			run_gui=False
			apt = Apt.Apt()
			print ("[Available Profiles]:")
			for name,file in apt.skel:
				print name
			#print (apt.skel)
			
			print("")
			print("[Optional Sources]:")
			for name,file in apt.optional:
				print name
			#print(apt.optional)

		if sys.argv[1]=="--add-ubuntu":
			run_gui=False
			apt=Apt.Apt()
			for name,file in apt.optional:
				print "Copying "+file+ " to /etc/apt/sources.list.d"
				shutil.copy(file,"/etc/apt/sources.list.d/")

			
		if sys.argv[1]=="--configure":
			run_gui=False
			sources_list=[]
			apt =Apt.Apt()
			lst = sys.argv[2:]
			
			if len(lst)>0:
				print ("Looking up: " + lst[0])
				for name,file in apt.skel:
					if name==lst[0]:
						print ("Including: " + file)
						sources_list.append(file)
						break
					
				if len(lst)>1:
					for opt in lst[1:]:
						print ("* Looking up: " + opt)
						for name,file in apt.optional:
							if name==opt:
								print ("Including: "+file)
								sources_list.append(file)
								break 
								
				if len(sources_list)>0:
					apt.create_source_list_file(sources_list)
					
					
			else:
				print ("No profile specified")
			
	if run_gui:	
		interface = Interface.Interface()
		Gtk.main()



if __name__=="__main__":
	main()