~cszikszoy/do-plugins/vbox

// VMItem.cs 

//

//  GNOME Do is the legal property of its developers.

//  Please refer to the COPYRIGHT file distributed with this

//  source distribution.

//

//  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 3 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, see <http://www.gnu.org/licenses/>.

 

using System;

using System.Diagnostics;

using System.Collections.Generic;

using System.IO;

using System.Xml;

using System.Threading;

 

using Do.Universe;

 

namespace VirtualBox

{

        public enum VMState

        {

                saved,

                on,

                paused,

                off,

                limbo,

                headless

        }

 

        public class VMItem : IItem

        {

                string name;

                string uuid;

                string ico_file;

                bool has_saved_states = false;

                VMState state;

                

                public VMItem (string xmlLineEntry) {

                        uuid = xmlLineEntry.Substring(xmlLineEntry.IndexOf("uuid=\"{")+7,36);

                        //find machine specific xml file

                        int StartMachinePath = xmlLineEntry.IndexOf("src=\"")+5;

                        int EndMachinePath = xmlLineEntry.LastIndexOf(".xml\" ");

                        string home = Environment.GetFolderPath (Environment.SpecialFolder.Personal);

                        string MachineSource = ("~/.VirtualBox/"+xmlLineEntry.Substring(StartMachinePath,(EndMachinePath+4) - StartMachinePath)).Replace ("~", home);

                        //parse machine specific xml file

                        try

                        {

                                StreamReader re = File.OpenText(MachineSource);

                                string input = null;

                                while ((input = re.ReadLine()) != null)

                                {

                                        if ((input.Length > 20) && (input.Substring(0,11) == "  <Machine "))

                                        {

                                                int start = input.IndexOf("OSType=\"")+8;

                                                int end = input.IndexOf("\" lastState");

                                                ico_file = "os_" + input.Substring(start,end-start) + ".png";

                                        }

                                }

                                re.Close();

                        }

                        catch //can't find icon, assign default VBox Icon

                        {

                                ico_file = "VirtualBox_64px.png@"+GetType().Assembly.FullName;

                        }

                        ico_file += "@"+GetType().Assembly.FullName;                              

                        string[] split = xmlLineEntry.Substring(StartMachinePath,EndMachinePath -StartMachinePath).Split('/');

                        name = split[1];

                        //determine the state of thte VM

                        ProcessStartInfo ps = new ProcessStartInfo ("VBoxManage", "showvminfo " + uuid);

                        ps.UseShellExecute = false;

                        ps.RedirectStandardOutput = true;

                        using (Process p = Process.Start (ps)) {

                                p.WaitForExit ();

                                string output = p.StandardOutput.ReadToEnd ();

                                int s = output.IndexOf("State:");

                                int e = output.IndexOf("\n", s);

                                string outputState = output.Substring(s, e-s);

                                //States: saved, running, paused, powered off

                                if (outputState.Contains("saved"))

                                        state = VMState.saved;

                                else if (outputState.Contains("running"))

                                        state = VMState.on;

                                else if (outputState.Contains("paused"))

                                        state = VMState.paused;

                                else if (outputState.Contains("powered off"))

                                        state = VMState.off;

                                if (output.Contains("Snapshots:"))

                                    has_saved_states = true;

                        }                   

                }

                

                public string Name { get { return name; } }

                public string Uuid { get { return uuid; } }

                public bool HasSavedStates { get { return has_saved_states; } }

                public VMState Status 

                { 

                        get { return state; }

                        set { state = value; }

                }

                public string Description { get { return "Virtual Machine: " + name; } }

                public string Icon { get { return ico_file; } }

        }

}