~teejee2008/timeshift/trunk

/*

 * MainWindow.vala

 *

 * Copyright 2013 Tony George <teejee2008@gmail.com>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA 02110-1301, USA.

 *

 *

 */

using Gtk;

using Gee;

using TeeJee.Logging;

using TeeJee.FileSystem;

using TeeJee.Devices;

using TeeJee.JSON;

using TeeJee.ProcessManagement;

using TeeJee.GtkHelper;

using TeeJee.Multimedia;

using TeeJee.System;

using TeeJee.Misc;

public class RestoreWindow : Gtk.Dialog{

	private Box vbox_main;

	private Box hbox_action;

	private Notebook notebook;

    //target device

    private Label lbl_header_partitions;

    private RadioButton radio_sys;

    private RadioButton radio_other;

    private TreeView tv_partitions;

	private ScrolledWindow sw_partitions;

	private TreeViewColumn col_device_target;

	private TreeViewColumn col_mount;

	private TreeViewColumn col_fs;

	private TreeViewColumn col_size;

	private TreeViewColumn col_dist;

	private CellRendererCombo cell_mount;

	//bootloader

	private Label lbl_header_bootloader;

	private ComboBox cmb_boot_device;

	private CheckButton chk_skip_grub_install;

	//apps

	private Label lbl_app;

	private Box vbox_app;

	private Label lbl_app_message;

	private TreeView tv_app;

	private ScrolledWindow sw_app;

	private TreeViewColumn col_app;

	//exclude

	private Label lbl_exclude;

	private Box vbox_exclude;

	private LinkButton lnk_default_list;

	private TreeView tv_exclude;

	private ScrolledWindow sw_exclude;

	private TreeViewColumn col_exclude;

	private Toolbar toolbar_exclude;

	private ToolButton btn_remove;

	private ToolButton btn_warning;

	private ToolButton btn_reset_exclude_list;

	private MenuToolButton btn_exclude;

	private Gtk.Menu menu_exclude;

	private ImageMenuItem menu_exclude_add_file;

	private ImageMenuItem menu_exclude_add_folder;

	private ImageMenuItem menu_exclude_add_folder_contents;

	private MenuToolButton btn_include;

	private Gtk.Menu menu_include;

	private ImageMenuItem menu_include_add_file;

	private ImageMenuItem menu_include_add_folder;

	private Gee.ArrayList<string> temp_exclude_list;

	//actions

	private Button btn_cancel;

	private Button btn_restore;

	private Device selected_target = null;

	public RestoreWindow () {

		this.title = _("Restore");

        this.window_position = WindowPosition.CENTER_ON_PARENT;

        this.set_destroy_with_parent (true);

		this.set_modal (true);

        this.set_default_size (550, 500);

		this.skip_taskbar_hint = true;

		this.icon = get_app_icon(16);

	    //vbox_main

        vbox_main = get_content_area ();

		//notebook

		notebook = new Notebook ();

		notebook.margin = 6;

		vbox_main.pack_start (notebook, true, true, 0);

		//target device tab -------------------------------------------------

		//lbl_exclude

		lbl_exclude = new Label (_("Target"));

        //vbox_target

        Box vbox_target = new Box (Orientation.VERTICAL, 6);

        vbox_target.margin = 6;

        notebook.append_page (vbox_target, lbl_exclude);

        //hbox_device

        Box hbox_device = new Box (Orientation.HORIZONTAL, 6);

        //hbox_device.margin = 6;

		vbox_target.add(hbox_device);

		//lbl_header_partitions

		lbl_header_partitions = new Gtk.Label((App.mirror_system ? _("Device for Cloning System") : _("Device for Restoring Snapshot")) + ":");

		lbl_header_partitions.xalign = (float) 0.0;

		lbl_header_partitions.set_use_markup(true);

		hbox_device.add(lbl_header_partitions);

		radio_sys = new RadioButton(null);

		hbox_device.add(radio_sys);

		radio_sys.label = "Current System";

		radio_other = new RadioButton.from_widget(radio_sys);

		hbox_device.add(radio_other);

		radio_other.label = "Other Device";

		if (App.live_system() || App.mirror_system){

			radio_other.active = true;

			radio_sys.sensitive = false;

		}

		else{

			radio_sys.sensitive = true;

			radio_sys.active = true;

		}

		radio_sys.toggled.connect(() => {

			sw_partitions.sensitive = radio_other.active;

			refresh_tv_partitions();

			if (radio_sys.active){

				App.restore_target = App.root_device;

			}

			else{

				init_mounts();

			}

			//tv_partitions_select_target();

			cmb_boot_device_select_default();

		});

		//tv_partitions

		tv_partitions = new TreeView();

		tv_partitions.get_selection().mode = SelectionMode.SINGLE;

		tv_partitions.set_rules_hint (true);

		tv_partitions.button_release_event.connect(tv_partitions_button_press_event);

		//sw_partitions

		sw_partitions = new ScrolledWindow(null, null);

		sw_partitions.set_shadow_type (ShadowType.ETCHED_IN);

		sw_partitions.add (tv_partitions);

		sw_partitions.expand = true;

		vbox_target.add(sw_partitions);

		//col_device

		col_device_target = new TreeViewColumn();

		col_device_target.title = _("Device");

		col_device_target.spacing = 1;

		tv_partitions.append_column(col_device_target);

		CellRendererPixbuf cell_device_icon = new CellRendererPixbuf();

		cell_device_icon.xpad = 1;

		col_device_target.pack_start (cell_device_icon, false);

		col_device_target.set_attributes(cell_device_icon, "pixbuf", 3);

		CellRendererText cell_device_target = new CellRendererText ();

		col_device_target.pack_start (cell_device_target, false);

		col_device_target.set_cell_data_func (cell_device_target, cell_device_target_render);

		//col_fs

		col_fs = new TreeViewColumn();

		col_fs.title = _("Type");

		CellRendererText cell_fs = new CellRendererText ();

		cell_fs.xalign = (float) 0.5;

		col_fs.pack_start (cell_fs, false);

		col_fs.set_cell_data_func (cell_fs, cell_fs_render);

		tv_partitions.append_column(col_fs);

		//col_mount

		col_mount = new TreeViewColumn();

		col_mount.title = _("Mount");

		cell_mount = new CellRendererCombo();

		cell_mount.xalign = (float) 0.0;

		cell_mount.editable = true;

		cell_mount.width = 70;

		col_mount.pack_start (cell_mount, false);

		col_mount.set_cell_data_func (cell_mount, cell_mount_render);

		tv_partitions.append_column(col_mount);

		cell_mount.set_property ("text-column", 0);

		col_mount.add_attribute (cell_mount, "text", 1);

		//populate combo

		var model = new Gtk.ListStore(1, typeof(string));

		cell_mount.model = model;

		TreeIter iter;

		model.append(out iter);

		model.set (iter, 0, "/");

		model.append(out iter);

		model.set (iter, 0, "/home");

		model.append(out iter);

		model.set (iter, 0, "/boot");

		cell_mount.changed.connect((path, iter_new) => {

			string val;

			cell_mount.model.get (iter_new, 0, out val);

			model = (Gtk.ListStore) tv_partitions.model;

			model.get_iter_from_string (out iter, path);

			model.set (iter, 1, val);

		});

		cell_mount.edited.connect((path, new_text) => {

			model = (Gtk.ListStore) tv_partitions.model;

			model.get_iter_from_string (out iter, path);

			model.set (iter, 1, new_text);

		});

		//col_size

		col_size = new TreeViewColumn();

		col_size.title = _("Size");

		CellRendererText cell_size = new CellRendererText ();

		cell_size.xalign = (float) 1.0;

		col_size.pack_start (cell_size, false);

		col_size.set_cell_data_func (cell_size, cell_size_render);

		tv_partitions.append_column(col_size);

		//col_dist

		col_dist = new TreeViewColumn();

		col_dist.title = _("System");

		CellRendererText cell_dist = new CellRendererText ();

		col_dist.pack_start (cell_dist, false);

		col_dist.set_cell_data_func (cell_dist, cell_dist_render);

		tv_partitions.append_column(col_dist);

		tv_partitions.set_tooltip_column(2);

		//bootloader options -------------------------------------------

		//lbl_header_bootloader

		lbl_header_bootloader = new Gtk.Label(_("Device for Bootloader Installation") + ":");

		lbl_header_bootloader.set_use_markup(true);

		lbl_header_bootloader.xalign = (float) 0.0;

		vbox_target.add(lbl_header_bootloader);

		//hbox_grub

		Box hbox_grub = new Box (Orientation.HORIZONTAL, 6);

		hbox_grub.margin_right = 6;

        //hbox_grub.margin_bottom = 6;

        vbox_target.add (hbox_grub);

		//cmb_boot_device

		cmb_boot_device = new ComboBox ();

		cmb_boot_device.hexpand = true;

		hbox_grub.add(cmb_boot_device);

		CellRendererText cell_dev_margin = new CellRendererText ();

		cell_dev_margin.text = "";

		cmb_boot_device.pack_start (cell_dev_margin, false);

		CellRendererPixbuf cell_dev_icon = new CellRendererPixbuf ();

		cell_dev_icon.stock_id = "gtk-harddisk";

		cmb_boot_device.pack_start (cell_dev_icon, false);

		CellRendererText cell_device_grub = new CellRendererText();

        cmb_boot_device.pack_start(cell_device_grub, false );

        cmb_boot_device.set_cell_data_func (cell_device_grub, cell_device_grub_render);

		string tt = "<b>" + _("** Advanced Users **") + "</b>\n\n"+ _("Skips bootloader (re)installation on target device.\nFiles in /boot directory on target partition will remain untouched.\n\nIf you are restoring a system that was bootable previously then it should boot successfully.\nOtherwise the system may fail to boot.");

		//chk_skip_grub_install

		chk_skip_grub_install = new CheckButton.with_label(_("Skip bootloader installation (not recommended)"));

		chk_skip_grub_install.active = false;

		chk_skip_grub_install.set_tooltip_markup(tt);

		vbox_target.add (chk_skip_grub_install);

		chk_skip_grub_install.toggled.connect(()=>{

			cmb_boot_device.sensitive = !chk_skip_grub_install.active;

		});

        //Exclude Apps tab ---------------------------------------------

		//lbl_apps

		lbl_app = new Label (_("Exclude"));

        //vbox_apps

        vbox_app = new Box(Gtk.Orientation.VERTICAL, 6);

        vbox_app.margin = 6;

        notebook.append_page (vbox_app, lbl_app);

		//lbl_app_message

		string msg = _("Select the applications for which current settings should be kept.") + "\n";

		msg += _("For all other applications, settings will be restored from selected snapshot.");

		lbl_app_message = new Label (msg);

		lbl_app_message.xalign = (float) 0.0;

		vbox_app.add(lbl_app_message);

		//tv_app-----------------------------------------------

		//tv_app

		tv_app = new TreeView();

		tv_app.get_selection().mode = SelectionMode.MULTIPLE;

		tv_app.headers_visible = false;

		tv_app.set_rules_hint (true);

		//sw_app

		sw_app = new ScrolledWindow(null, null);

		sw_app.set_shadow_type (ShadowType.ETCHED_IN);

		sw_app.add (tv_app);

		sw_app.expand = true;

		vbox_app.add(sw_app);

        //col_app

		col_app = new TreeViewColumn();

		col_app.title = _("Application");

		col_app.expand = true;

		tv_app.append_column(col_app);

		CellRendererText cell_app_margin = new CellRendererText ();

		cell_app_margin.text = "";

		col_app.pack_start (cell_app_margin, false);

		CellRendererToggle cell_app_enabled = new CellRendererToggle ();

		cell_app_enabled.radio = false;

		cell_app_enabled.activatable = true;

		col_app.pack_start (cell_app_enabled, false);

		col_app.set_cell_data_func (cell_app_enabled, cell_app_enabled_render);

		cell_app_enabled.toggled.connect (cell_app_enabled_toggled);

		CellRendererText cell_app_text = new CellRendererText ();

		col_app.pack_start (cell_app_text, false);

		col_app.set_cell_data_func (cell_app_text, cell_app_text_render);

        //Advanced tab ---------------------------------------------

		//lbl_exclude

		lbl_exclude = new Label (_("Advanced"));

        //vbox_exclude

        vbox_exclude = new Box(Gtk.Orientation.VERTICAL, 6);

        vbox_exclude.margin = 6;

        notebook.append_page (vbox_exclude, lbl_exclude);

		//toolbar_exclude ---------------------------------------------------

        //toolbar_exclude

		toolbar_exclude = new Gtk.Toolbar ();

		toolbar_exclude.toolbar_style = ToolbarStyle.BOTH_HORIZ;

		vbox_exclude.add(toolbar_exclude);

		string png_exclude = App.share_folder + "/timeshift/images/item-gray.png";

		string png_include = App.share_folder + "/timeshift/images/item-blue.png";

		//btn_exclude

		btn_exclude = new Gtk.MenuToolButton(null,"");

		toolbar_exclude.add(btn_exclude);

		btn_exclude.is_important = true;

		btn_exclude.label = _("Exclude");

		btn_exclude.set_tooltip_text (_("Exclude"));

		btn_exclude.set_icon_widget(new Gtk.Image.from_file (png_exclude));

		//btn_include

		btn_include = new Gtk.MenuToolButton(null,"");

		toolbar_exclude.add(btn_include);

		btn_include.is_important = true;

		btn_include.label = _("Include");

		btn_include.set_tooltip_text (_("Include"));

		btn_include.set_icon_widget(new Gtk.Image.from_file (png_include));

		//btn_remove

		btn_remove = new Gtk.ToolButton.from_stock("gtk-remove");

		toolbar_exclude.add(btn_remove);

		btn_remove.is_important = true;

		btn_remove.label = _("Remove");

		btn_remove.set_tooltip_text (_("Remove selected items"));

		btn_remove.clicked.connect (btn_remove_clicked);

		//btn_warning

		btn_warning = new Gtk.ToolButton.from_stock("gtk-dialog-warning");

		toolbar_exclude.add(btn_warning);

		btn_warning.is_important = true;

		btn_warning.label = _("Warning");

		btn_warning.set_tooltip_text (_("Warning"));

		btn_warning.clicked.connect (btn_warning_clicked);

		//separator

		var separator = new Gtk.SeparatorToolItem();

		separator.set_draw (false);

		separator.set_expand (true);

		toolbar_exclude.add(separator);

		//btn_reset_exclude_list

		btn_reset_exclude_list = new Gtk.ToolButton.from_stock("gtk-refresh");

		toolbar_exclude.add(btn_reset_exclude_list);

		btn_reset_exclude_list.is_important = false;

		btn_reset_exclude_list.label = _("Reset");

		btn_reset_exclude_list.set_tooltip_text (_("Reset this list to default state"));

		btn_reset_exclude_list.clicked.connect (btn_reset_exclude_list_clicked);

		//menu --------------------------------------------------

        //menu_exclude

		menu_exclude = new Gtk.Menu();

		btn_exclude.set_menu(menu_exclude);

		//menu_exclude_add_file

		menu_exclude_add_file = new ImageMenuItem.with_label ("");

		menu_exclude_add_file.label = _("Exclude File(s)");

		menu_exclude_add_file.set_image(new Gtk.Image.from_file (png_exclude));

		menu_exclude.append(menu_exclude_add_file);

		menu_exclude_add_file.activate.connect (menu_exclude_add_files_clicked);

		//menu_exclude_add_folder

		menu_exclude_add_folder = new ImageMenuItem.with_label ("");

		menu_exclude_add_folder.label = _("Exclude Directory");

		menu_exclude_add_folder.set_image(new Gtk.Image.from_file (png_exclude));

		menu_exclude.append(menu_exclude_add_folder);

		menu_exclude_add_folder.activate.connect (menu_exclude_add_folder_clicked);

		//menu_exclude_add_folder_contents

		menu_exclude_add_folder_contents = new ImageMenuItem.with_label ("");

		menu_exclude_add_folder_contents.label = _("Exclude Directory Contents");

		menu_exclude_add_folder_contents.set_image(new Gtk.Image.from_file (png_exclude));

		menu_exclude.append(menu_exclude_add_folder_contents);

		menu_exclude_add_folder_contents.activate.connect (menu_exclude_add_folder_contents_clicked);

		//menu_include

		menu_include = new Gtk.Menu();

		btn_include.set_menu(menu_include);

		//menu_include_add_file

		menu_include_add_file = new ImageMenuItem.with_label ("");

		menu_include_add_file.label = _("Include File(s)");

		menu_include_add_file.set_image(new Gtk.Image.from_file (png_include));

		menu_include.append(menu_include_add_file);

		menu_include_add_file.activate.connect (menu_include_add_files_clicked);

		//menu_include_add_folder

		menu_include_add_folder = new ImageMenuItem.with_label ("");

		menu_include_add_folder.label = _("Include Directory");

		menu_include_add_folder.set_image(new Gtk.Image.from_file (png_include));

		menu_include.append(menu_include_add_folder);

		menu_include_add_folder.activate.connect (menu_include_add_folder_clicked);

		menu_exclude.show_all();

		menu_include.show_all();

		//tv_exclude-----------------------------------------------

		//tv_exclude

		tv_exclude = new TreeView();

		tv_exclude.get_selection().mode = SelectionMode.MULTIPLE;

		tv_exclude.headers_visible = true;

		tv_exclude.set_rules_hint (true);

		//tv_exclude.row_activated.connect(tv_exclude_row_activated);

		//sw_exclude

		sw_exclude = new ScrolledWindow(null, null);

		sw_exclude.set_shadow_type (ShadowType.ETCHED_IN);

		sw_exclude.add (tv_exclude);

		sw_exclude.expand = true;

		vbox_exclude.add(sw_exclude);

        //col_exclude

		col_exclude = new TreeViewColumn();

		col_exclude.title = _("File Pattern");

		col_exclude.expand = true;

		CellRendererText cell_exclude_margin = new CellRendererText ();

		cell_exclude_margin.text = "";

		col_exclude.pack_start (cell_exclude_margin, false);

		CellRendererPixbuf cell_exclude_icon = new CellRendererPixbuf ();

		col_exclude.pack_start (cell_exclude_icon, false);

		col_exclude.set_attributes(cell_exclude_icon, "pixbuf", 1);

		CellRendererText cell_exclude_text = new CellRendererText ();

		col_exclude.pack_start (cell_exclude_text, false);

		col_exclude.set_cell_data_func (cell_exclude_text, cell_exclude_text_render);

		cell_exclude_text.editable = true;

		tv_exclude.append_column(col_exclude);

		cell_exclude_text.edited.connect (cell_exclude_text_edited);

		//lnk_default_list

		lnk_default_list = new LinkButton.with_label("",_("Some locations are excluded by default"));

		lnk_default_list.xalign = (float) 0.0;

		lnk_default_list.activate_link.connect(lnk_default_list_activate);

		vbox_exclude.add(lnk_default_list);

		//Actions ----------------------------------------------

		//hbox_action

        hbox_action = (Box) get_action_area ();

        //btn_restore

        btn_restore = new Button();

        hbox_action.add(btn_restore);

        btn_restore.set_label (" " + _("Restore"));

        btn_restore.set_tooltip_text (_("Restore"));

        Gtk.Image img_restore = new Image.from_stock("gtk-go-forward", Gtk.IconSize.BUTTON);

		btn_restore.set_image(img_restore);

        btn_restore.clicked.connect (btn_restore_clicked);

        //btn_cancel

        btn_cancel = new Button();

        hbox_action.add(btn_cancel);

        btn_cancel.set_label (" " + _("Cancel"));

        btn_cancel.set_tooltip_text (_("Cancel"));

        Gtk.Image img_cancel = new Image.from_stock("gtk-cancel", Gtk.IconSize.BUTTON);

		btn_cancel.set_image(img_cancel);

        btn_cancel.clicked.connect (btn_cancel_clicked);

		//initialize -----------------------------------------

		btn_reset_exclude_list_clicked();

		refresh_tv_partitions();

		refresh_cmb_boot_device();

		//refresh_tv_exclude(); //called by btn_reset_exclude_list_clicked()

		refresh_tv_apps();

		sw_partitions.sensitive = radio_other.active;

		notebook.switch_page.connect_after((page, new_page_index) => {

			if (new_page_index == 1){

				bool ok = check_and_mount_devices();

				if (!ok){

					notebook.set_current_page(0);

					gtk_do_events();

					return;

				}

				//save current app selections

				Gee.ArrayList<string> selected_app_list = new Gee.ArrayList<string>();

				foreach(AppExcludeEntry entry in App.exclude_list_apps){

					if (entry.enabled){

						selected_app_list.add(entry.relpath);

					}

				}

				//refresh the list

				App.add_app_exclude_entries();

				//restore app selections

				foreach(AppExcludeEntry entry in App.exclude_list_apps){

					if (selected_app_list.contains(entry.relpath)){

						entry.enabled = true;

					}

				}

				//refresh treeview

				refresh_tv_apps();

			}

		});

		set_app_page_state();

		init_mounts();

	}

	private void init_mounts(){

		TreeIter iter;

		Gtk.ListStore store;

		App.init_mount_list();

		if (App.mirror_system){

			//default all mount points to root device except /boot

			for(int i = App.mount_list.size - 1; i >= 0; i--){

				MountEntry mnt = App.mount_list[i];

				if (mnt.mount_point != "/boot"){

					App.mount_list.remove_at(i);

				}

			}

			/* Note:

			 * While cloning the system, /boot is the only mount point that we will leave unchanged (to avoid encrypted systems from breaking)

			 * All other mounts like /home will be defaulted to target device (to prevent the "cloned" system from using the original device)

			 * */

		}

		//find the root mount point set by user

		store = (Gtk.ListStore) tv_partitions.model;

		for (bool next = store.get_iter_first (out iter); next; next = store.iter_next (ref iter)) {

			Device pi;

			string mount_point;

			store.get(iter, 0, out pi);

			store.get(iter, 1, out mount_point);

			foreach(MountEntry mnt in App.mount_list){

				if (mnt.device.device == pi.device){