ericbsd
239 lines
8.9 KiB
Python
Executable File
239 lines
8.9 KiB
Python
Executable File
#!/usr/local/bin/python
|
|
|
|
import sys
|
|
import os
|
|
import shutil
|
|
import gi
|
|
gi.require_version('Gtk', '3.0')
|
|
from gi.repository import Gtk
|
|
installer = "/usr/local/lib/gbi/"
|
|
sys.path.append(installer)
|
|
from welcome_live import Welcome
|
|
from installType import Types
|
|
from use_ufs import use_ufs
|
|
from partition import Partitions
|
|
from use_zfs import ZFS
|
|
from boot_manager import bootManager
|
|
from partition_handler import create_disk_partition_db
|
|
from install import installProgress, installSlide
|
|
from network_setup import network_setup
|
|
|
|
logo = "/usr/local/lib/gbi/logo.png"
|
|
tmp = "/tmp/.gbi/"
|
|
if not os.path.exists(tmp):
|
|
os.makedirs(tmp)
|
|
disk = '%sdisk' % tmp
|
|
dslice = '%sslice' % tmp
|
|
disk_schem = '%sscheme' % tmp
|
|
zfs_config = '%szfs_config' % tmp
|
|
ufs_config = '%sufs_config' % tmp
|
|
partitiondb = "%spartitiondb/" % tmp
|
|
|
|
|
|
class MainWindow():
|
|
"""Main window class."""
|
|
|
|
def delete(self, widget, event=None):
|
|
"""Close the main window."""
|
|
if os.path.exists('/tmp/.gbi'):
|
|
shutil.rmtree('/tmp/.gbi')
|
|
Gtk.main_quit()
|
|
return False
|
|
|
|
def next_button(self, widget):
|
|
if self.welcome.get_what() == "install":
|
|
self.next_install_page()
|
|
else:
|
|
self.next_setup_page()
|
|
|
|
def next_install_page(self):
|
|
"""Go to the next window."""
|
|
page = self.notebook.get_current_page()
|
|
if page == 0:
|
|
typebox = Gtk.VBox(False, 0)
|
|
typebox.show()
|
|
self.types = Types()
|
|
get_types = self.types.get_model()
|
|
typebox.pack_start(get_types, True, True, 0)
|
|
label = Gtk.Label("Types")
|
|
self.notebook.insert_page(typebox, label, 1)
|
|
self.notebook.next_page()
|
|
self.notebook.show_all()
|
|
self.button2.show()
|
|
self.button3.show()
|
|
elif page == 1:
|
|
create_disk_partition_db()
|
|
self.button1.show()
|
|
if self.types.get_type() == "ufs":
|
|
udbox = Gtk.VBox(False, 0)
|
|
udbox.show()
|
|
self.partition = use_ufs(self.button3)
|
|
get_ud = self.partition.get_model()
|
|
udbox.pack_start(get_ud, True, True, 0)
|
|
label = Gtk.Label("UFS Disk Configuration")
|
|
self.notebook.insert_page(udbox, label, 2)
|
|
self.notebook.next_page()
|
|
self.notebook.show_all()
|
|
self.button3.set_sensitive(False)
|
|
elif self.types.get_type() == "custom":
|
|
Pbox = Gtk.VBox(False, 0)
|
|
Pbox.show()
|
|
self.partition = Partitions(self.button3)
|
|
get_part = self.partition.get_model()
|
|
Pbox.pack_start(get_part, True, True, 0)
|
|
label = Gtk.Label("UFS Custom Configuration")
|
|
self.notebook.insert_page(Pbox, label, 2)
|
|
self.notebook.next_page()
|
|
self.notebook.show_all()
|
|
self.button3.set_sensitive(False)
|
|
elif self.types.get_type() == "zfs":
|
|
Zbox = Gtk.VBox(False, 0)
|
|
Zbox.show()
|
|
self.partition = ZFS(self.button3)
|
|
get_ZFS = self.partition.get_model()
|
|
Zbox.pack_start(get_ZFS, True, True, 0)
|
|
label = Gtk.Label("ZFS Configuration")
|
|
self.notebook.insert_page(Zbox, label, 2)
|
|
self.notebook.next_page()
|
|
self.notebook.show_all()
|
|
self.button3.set_sensitive(False)
|
|
elif page == 2:
|
|
self.partition.save_selection()
|
|
Mbox = Gtk.VBox(False, 0)
|
|
Mbox.show()
|
|
self.bootmanager = bootManager()
|
|
get_root = self.bootmanager.get_model()
|
|
Mbox.pack_start(get_root, True, True, 0)
|
|
label = Gtk.Label("Boot Option")
|
|
self.notebook.insert_page(Mbox, label, 3)
|
|
self.button3.set_label("Install")
|
|
self.notebook.next_page()
|
|
self.notebook.show_all()
|
|
self.button3.set_sensitive(True)
|
|
elif page == 3:
|
|
Ibox = Gtk.VBox(False, 0)
|
|
Ibox.show()
|
|
install = installSlide()
|
|
get_install = install.get_model()
|
|
Ibox.pack_start(get_install, True, True, 0)
|
|
label = Gtk.Label("Installation")
|
|
self.notebook.insert_page(Ibox, label, 8)
|
|
self.notebook.next_page()
|
|
instpro = installProgress()
|
|
progressBar = instpro.getProgressBar()
|
|
box1 = Gtk.VBox(False, 0)
|
|
box1.show()
|
|
label = Gtk.Label("Progress Bar")
|
|
box1.pack_end(progressBar, False, False, 0)
|
|
self.nbButton.insert_page(box1, label, 4)
|
|
self.nbButton.next_page()
|
|
self.window.show_all()
|
|
|
|
def next_setup_page(self):
|
|
page = self.notebook.get_current_page()
|
|
if page == 0:
|
|
self.window.set_title("Setup GhostBSD")
|
|
net_setup_box = Gtk.VBox(False, 0)
|
|
net_setup_box.show()
|
|
self.net_setup = network_setup()
|
|
model = self.net_setup.get_model()
|
|
net_setup_box.pack_start(model, True, True, 0)
|
|
label = Gtk.Label("Network Setup")
|
|
self.notebook.insert_page(net_setup_box, label, 1)
|
|
self.notebook.next_page()
|
|
self.notebook.show_all()
|
|
self.button3.show()
|
|
|
|
def back_page(self, widget):
|
|
"""Go back to the previous window."""
|
|
current_page = self.notebook.get_current_page()
|
|
if current_page == 2:
|
|
self.button1.hide()
|
|
elif current_page == 3:
|
|
self.button3.set_label("Next")
|
|
self.notebook.prev_page()
|
|
new_page = self.notebook.get_current_page()
|
|
if current_page == 2 and new_page == 1:
|
|
if os.path.exists(partitiondb):
|
|
shutil.rmtree(partitiondb)
|
|
if os.path.exists(tmp + 'create'):
|
|
os.remove(tmp + 'create')
|
|
if os.path.exists(tmp + 'delete'):
|
|
os.remove(tmp + 'delete')
|
|
if os.path.exists(tmp + 'destroy'):
|
|
os.remove(tmp + 'destroy')
|
|
if os.path.exists(tmp + 'partlabel'):
|
|
os.remove(tmp + 'partlabel')
|
|
if os.path.exists(zfs_config):
|
|
os.remove(zfs_config)
|
|
if os.path.exists(ufs_config):
|
|
os.remove(ufs_config)
|
|
if os.path.exists(disk):
|
|
os.remove(disk)
|
|
if os.path.exists(dslice):
|
|
os.remove(dslice)
|
|
if os.path.exists(disk_schem):
|
|
os.remove(disk_schem)
|
|
self.button3.set_sensitive(True)
|
|
|
|
def __init__(self):
|
|
"""Were the Main window start."""
|
|
self.window = Gtk.Window()
|
|
self.window.connect("delete_event", self.delete)
|
|
self.window.set_border_width(0)
|
|
self.window.set_default_size(800, 500)
|
|
self.window.set_size_request(800, 500)
|
|
self.window.set_title("Install GhostBSD")
|
|
self.window.set_border_width(0)
|
|
self.window.set_icon_from_file(logo)
|
|
mainHBox = Gtk.HBox(False, 0)
|
|
mainHBox.show()
|
|
mainVbox = Gtk.VBox(False, 0)
|
|
mainVbox.show()
|
|
self.window.add(mainHBox)
|
|
mainHBox.pack_start(mainVbox, True, True, 0)
|
|
# Create a new self.notebook
|
|
self.notebook = Gtk.Notebook()
|
|
mainVbox.pack_start(self.notebook, True, True, 0)
|
|
self.notebook.show()
|
|
self.notebook.set_show_tabs(False)
|
|
self.notebook.set_show_border(False)
|
|
welcome_box = Gtk.VBox(False, 0)
|
|
welcome_box.show()
|
|
self.welcome = Welcome(self.next_install_page, self.next_setup_page)
|
|
get_types = self.welcome.get_model()
|
|
welcome_box.pack_start(get_types, True, True, 0)
|
|
label = Gtk.Label("Welcome")
|
|
self.notebook.insert_page(welcome_box, label, 0)
|
|
|
|
# Set what page to start at type of installation
|
|
self.notebook.set_current_page(0)
|
|
# Create buttons
|
|
self.table = Gtk.Table(1, 6, True)
|
|
self.button1 = Gtk.Button(label='Back')
|
|
self.button1.connect("clicked", self.back_page)
|
|
self.table.attach(self.button1, 3, 4, 0, 1)
|
|
self.button2 = Gtk.Button(label='Cancel')
|
|
self.button2.connect("clicked", self.delete)
|
|
self.table.attach(self.button2, 4, 5, 0, 1)
|
|
self.button3 = Gtk.Button(label='Next')
|
|
self.button3.connect("clicked", self.next_button)
|
|
self.table.attach(self.button3, 5, 6, 0, 1)
|
|
self.table.set_col_spacings(5)
|
|
self.table.show()
|
|
self.nbButton = Gtk.Notebook()
|
|
mainVbox.pack_end(self.nbButton, False, False, 5)
|
|
self.nbButton.show()
|
|
self.nbButton.set_show_tabs(False)
|
|
self.nbButton.set_show_border(False)
|
|
label = Gtk.Label("Button")
|
|
self.nbButton.insert_page(self.table, label, 0)
|
|
self.window.show_all()
|
|
self.button1.hide()
|
|
self.button2.hide()
|
|
self.button3.hide()
|
|
|
|
|
|
MainWindow()
|
|
Gtk.main()
|