+ (* Wire up the loosely-coupled external components of the filetree.
+ * See the note about signals in {!Filetree.tree} documentation.
+ *)
+ ignore (tree#op_checksum_file
+ ~callback:(Op_checksum_file.checksum_file tree));
+ ignore (tree#op_copy_regvalue
+ ~callback:(Op_copy_regvalue.copy_regvalue tree));
+ ignore (tree#op_disk_usage
+ ~callback:(Op_disk_usage.disk_usage tree));
+ ignore (tree#op_download_as_reg
+ ~callback:(Op_download_as_reg.download_as_reg tree));
+ ignore (tree#op_download_dir_find0
+ ~callback:(Op_download_dir_find0.download_dir_find0 tree));
+ ignore (tree#op_download_dir_tarball
+ ~callback:(Op_download_dir_tarball.download_dir_tarball tree));
+ ignore (tree#op_download_file
+ ~callback:(Op_download_file.download_file tree));
+ ignore (tree#op_file_information
+ ~callback:(Op_file_information.file_information tree));
+ ignore (tree#op_inspection_dialog
+ ~callback:(Op_inspection_dialog.inspection_dialog tree));
+ ignore (tree#op_view_file
+ ~callback:(Op_view_file.view_file tree));
+
+ tree
+
+class window =
+ (* I prototyped the basic window layout using Glade, but have
+ * implemented it by hand to give us more flexibility.
+ *)
+ let title = "Guest Filesystem Browser" in
+ let window = GWindow.window ~width:700 ~height:700 ~title () in
+ let vbox = GPack.vbox ~packing:window#add () in
+
+ (* Menus. *)
+ let connect_kvm_item, connect_xen_item, connect_none_item, _, _ =
+ make_menubar window vbox ~packing:vbox#pack () in
+
+ (* Top toolbar. *)
+ let vmcombo, refresh_button, throbber, throbber_static =
+ make_toolbar ~packing:vbox#pack () in
+
+ (* Main part of display is the file tree. *)
+ let view = make_filetree ~packing:(vbox#pack ~expand:true ~fill:true) () in
+
+ (* Status bar and progress bar. *)
+ let hbox = GPack.hbox ~spacing:4 ~packing:vbox#pack () in
+ let progress_bar = GRange.progress_bar ~packing:hbox#pack () in
+ let statusbar = GMisc.statusbar ~packing:(hbox#pack ~expand:true) () in
+ let statusbar_context = statusbar#new_context ~name:"Standard" in
+
+object (self)
+ initializer
+ ignore (statusbar_context#push title);
+ window#show ();
+
+ (* Connect up the callback for menu entries etc. These require the
+ * window_state struct in callbacks.
+ *)
+
+ (* Connect to different hypervisors. *)
+ ignore (connect_kvm_item#connect#activate
+ ~callback:(fun () -> self#connect_to (Some "qemu:///system")));
+ ignore (connect_xen_item#connect#activate
+ ~callback:(fun () -> self#connect_to (Some "xen:///")));
+ ignore (connect_none_item#connect#activate
+ ~callback:(fun () -> self#connect_to None));
+
+ (* VM combo box when changed by the user.
+ * The refresh button acts like changing the VM combo too.
+ *)
+ let combo, (model, column) = vmcombo in
+ ignore (
+ combo#connect#changed
+ ~callback:(
+ fun () ->
+ match combo#active_iter with
+ | None -> () (* nothing selected *)
+ | Some row -> self#open_domain (model#get ~row ~column)
+ )
+ );
+ ignore (
+ refresh_button#connect#clicked
+ ~callback:(
+ fun () ->
+ match combo#active_iter with
+ | None -> () (* nothing selected *)
+ | Some row -> self#open_domain (model#get ~row ~column)
+ )
+ )
+
+ (* Set the statusbar text. *)
+ method set_statusbar msg =
+ statusbar_context#pop ();
+ ignore (statusbar_context#push msg)
+
+ (* Clear the filetree. *)
+ method private clear_view () =
+ view#clear ()
+
+ (* Callback from Connect -> ... menu items. *)
+ method private connect_to uri =
+ self#clear_view ();
+ Slave.discard_command_queue ();
+ Slave.connect uri (self#when_connected uri)
+
+ (* Called back when connected to a new hypervisor. *)
+ method private when_connected uri doms =
+ self#populate_vmcombo doms
+
+ (* Populate the VM combo box. *)
+ method private populate_vmcombo doms =
+ let combo, (model, column) = vmcombo in
+ model#clear ();
+ List.iter (
+ fun { dom_name = name } ->
+ let row = model#append () in
+ model#set ~row ~column name
+ ) doms
+
+ (* When a new domain is selected by the user, eg through vmcombo. *)
+ method private open_domain name =
+ self#clear_view ();
+ Slave.discard_command_queue ();
+ Slave.open_domain name (self#when_opened_domain name)
+
+ (* Called back when domain was opened successfully. *)
+ method private when_opened_domain name data =
+ debug "when_opened_domain callback";
+ self#when_opened_common name data
+
+ (* When a set of disk images is selected by the user. *)
+ method private open_disk_images images =
+ match images with
+ | [] -> ()
+ | images ->
+ self#clear_view ();
+ Slave.discard_command_queue ();
+ Slave.open_images images (self#when_opened_disk_images images)
+
+ (* Called back when disk image(s) were opened successfully. *)
+ method private when_opened_disk_images images data =
+ match images with
+ | [] -> ()
+ | (image, _) :: _ ->
+ debug "when_opened_disk_images callback";
+ self#when_opened_common image data
+
+ (* Common code for when_opened_domain/when_opened_disk_images. *)
+ method private when_opened_common name data =
+ (* Dump some of the inspection data in debug messages. *)
+ List.iter (fun (dev, t) -> debug "filesystem: %s: %s" dev t)
+ data.insp_all_filesystems;
+ List.iter (
+ fun { insp_root = root; insp_type = typ; insp_distro = distro;
+ insp_major_version = major; insp_minor_version = minor } ->
+ debug "root device %s contains %s %s %d.%d" root typ distro major minor;
+ ) data.insp_oses;
+
+ view#add_os name data
+
+ (* Public callbacks. *)
+ method throbber_busy () =
+ (*throbber#set_pixbuf animation*)
+ (* XXX Workaround because no binding for GdkPixbufAnimation: *)
+ let file = Filename.dirname Sys.argv.(0) // "Throbber.gif" in
+ throbber#set_file file
+
+ method throbber_idle () =
+ throbber#set_pixbuf throbber_static
+
+ method progress (position, total) =
+ if position = 0L && total = 1L then
+ progress_bar#pulse ()
+ else (
+ let frac = Int64.to_float position /. Int64.to_float total in
+ if frac < 0. || frac > 1. then
+ eprintf "warning: progress bar out of range: %Ld / %Ld (%g)\n"
+ position total frac;
+ let frac = if frac < 0. then 0. else if frac > 1. then 1. else frac in
+ progress_bar#set_fraction frac
+ )
+
+ (* This is called in the main thread whenever a command fails in the
+ * slave thread. The command queue has been cleared before this is
+ * called, so our job here is to reset the main window, and if
+ * necessary to turn the exception into an error message.
+ *)
+ method failure exn =
+ let raw_msg = Printexc.to_string exn in
+ debug "failure hook: %s" raw_msg;
+
+ let title, msg = pretty_string_of_exn exn in
+ let icon = GMisc.image () in
+ icon#set_stock `DIALOG_ERROR;
+ icon#set_icon_size `DIALOG;
+ GToolbox.message_box ~title ~icon msg
+
+ (* Do what the user asked on the command line. *)
+ method run_cli_request = function