diff --git a/install-station b/install-station index b715f8b..ad1710d 100755 --- a/install-station +++ b/install-station @@ -1,19 +1,204 @@ #!/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/gbinstall/" +installer = "/usr/local/lib/gbi/" sys.path.append(installer) -from gbiWindow import MainWindow +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 +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 -def main(): - Gtk.main() - return 0 +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_page(self, widget, notebook): + """Go to the next window.""" + page = self.notebook.get_current_page() + if page == 0: + create_disk_partition_db() + 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, 1) + self.window.show_all() + self.notebook.next_page() + 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, 1) + self.window.show_all() + self.notebook.next_page() + 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, 1) + self.window.show_all() + self.notebook.next_page() + self.button3.set_sensitive(False) + elif page == 1: + 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, 2) + self.button3.set_label("Install") + self.window.show_all() + self.notebook.next_page() + self.button3.set_sensitive(True) + elif page == 2: + self.adduser.save_selection() + 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, 3) + self.nbButton.next_page() + self.window.show_all() + + def back_page(self, widget): + """Go back to the previous window.""" + current_page = self.notebook.get_current_page() + if current_page == 1: + self.button1.set_sensitive(False) + elif current_page == 2: + 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("GhostBSD Installer") + 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) + 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, 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.button1.show() + self.button1.set_sensitive(False) + self.button2 = Gtk.Button(label='Cancel') + self.button2.connect("clicked", self.delete) + self.table.attach(self.button2, 4, 5, 0, 1) + self.button2.show() + self.button3 = Gtk.Button(label='Next') + self.button3.connect("clicked", self.next_page, self.notebook) + self.table.attach(self.button3, 5, 6, 0, 1) + self.button3.show() + self.table.set_col_spacings(5) + self.table.show() + # Create a new notebook + 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() -if __name__ == "__main__": - MainWindow() - main() +MainWindow() +Gtk.main()