using System;
using SQLDMO;
using System.Drawing;
using System.Collections;
using System.ComponentModel;
using System.Windows.Forms;
using System.Data;

namespace bakServe040907
{
	
	/// <summary>
	/// 數據庫備份類說明
	/// </summary>
	public class bakServer
	{
		string ServerName;   //數據服務器名稱
		string UserName;     //用戶名稱
		string Password;     //用戶密碼
		string message;      //消息提示 
		ProgressBar PBar;    //狀態條控件

		public bakServer()
		{
			
		}

		/// <summary>
		/// 取得數據庫服務器列表
		/// </summary>
		/// <returns>數據庫服務器列表</returns>
		public ArrayList GetServerList() 
		{ 
			ArrayList alServers = new ArrayList() ; 
			SQLDMO.Application sqlApp = new SQLDMO.ApplicationClass() ; 
			try 
			{ 
				SQLDMO.NameList serverList = sqlApp.ListAvailableSQLServers() ; 
				for(int i = 1;i<= serverList.Count;i++) 
				{ 
					alServers.Add(serverList.Item(i)) ; 
				} 
			} 
			catch(Exception e) 
			{ 
//				throw(new Exception("取數據庫服務器列表出錯："+e.Message)) ; 
				message = "取數據庫服務器列表出錯：" +e.Message;

			} 
			finally 
			{ 
				sqlApp.Quit() ; 
			} 
			return alServers ; 
		} 


		/// <summary>
		/// 錯誤消息處理
		/// </summary>
		/// <returns>消息信息</returns>
		public string Msg()
		{		
			return message;
		}

		/// <summary>
		/// 取得指定數據庫列表
		/// </summary>
		/// <param name="strServerName">服務器名稱</param>
		/// <param name="strUserName">用戶名稱</param>
		/// <param name="strPwd">用戶密碼</param>
		/// <returns>數據庫列表</returns>
		public ArrayList GetDbList(string strServerName,string strUserName,string strPwd) 
		{ 
			ServerName = strServerName ; 
			UserName = strUserName ; 
			Password = strPwd ; 

			
			ArrayList alDbs = new ArrayList() ; 
			SQLDMO.Application sqlApp = new SQLDMO.ApplicationClass() ; 
			SQLDMO.SQLServer svr = new SQLDMO.SQLServerClass() ; 
			try 
			{ 
				svr.Connect(ServerName,UserName,Password) ; 
				foreach(SQLDMO.Database db in svr.Databases) 
				{ 
					if(db.Name!=null) 
						alDbs.Add(db.Name) ; 
				} 
			} 
			catch(Exception err) 
			{ 
//				throw(new Exception("連接數據庫出錯："+e.Message)) ; 
				message = "連接數據庫出錯：" +err.Message;
			} 
			finally 
			{ 
				svr.DisConnect() ; 
				sqlApp.Quit() ; 
			} 
			return alDbs ; 
		} 


		//數據庫的備份和實時進度顯示代碼： 
		/// <summary>
		/// 數據庫的備份和實時進度顯示
		/// </summary>
		/// <param name="strDbName">數據庫名稱</param>
		/// <param name="strFileName">備份文件名</param>
		/// <param name="pgbMain">狀態條控件名稱</param>
		/// <param name="strServerName">服務器名稱</param>
		/// <param name="strUserName">用戶名稱</param>
		/// <param name="strPwd">密碼</param>
		/// <returns>備份成功返回true ，否則返回false</returns>
		public bool BackUPDB(string strDbName,string strFileName, ProgressBar pgbMain,string strServerName,string strUserName,string strPwd) 
		{ 
			ServerName = strServerName ; 
			UserName = strUserName ; 
			Password = strPwd ; 

			PBar = pgbMain; 
			SQLDMO.SQLServer svr = new SQLDMO.SQLServerClass() ; 
			try 
			{ 
				svr.Connect(ServerName,UserName,Password) ; 
				SQLDMO.Backup bak = new SQLDMO.BackupClass(); 
				bak.Action = 0 ; 
				bak.Initialize = true ; 
				SQLDMO.BackupSink_PercentCompleteEventHandler pceh = new SQLDMO.BackupSink_PercentCompleteEventHandler(Step); 
				bak.PercentComplete += pceh; 

				bak.Files = strFileName; 
				bak.Database = strDbName; 
				bak.SQLBackup(svr); 
				return true ; 
			} 
			catch(Exception err) 
			{ 
//				throw(new Exception("備份數據庫失敗"+err.Message)) ; 
				message = "備份數據庫失敗：" +err.Message;
                return false ; 
			} 
			finally 
			{ 
				svr.DisConnect() ; 
			} 
		} 


		/// <summary>
		/// 顯示進度條的當前進度
		/// </summary>
		/// <param name="message"></param>
		/// <param name="percent"></param>
		private void Step(string message,int percent) 
		{ 
			PBar.Value = percent ; 
		} 
		//		其中，這兩個語句實現了進度的實時顯示： 
		//		SQLDMO.BackupSink_PercentCompleteEventHandler pceh = new SQLDMO.BackupSink_PercentCompleteEventHandler(Step); 
		//		bak.PercentComplete += pceh; 
		//		Step就是上面private void Step(string message,int percent) 的方法名稱，它用來顯示進度條的當前進度。 



		/// <summary>
		/// 數據庫的恢復和殺死進程
		/// </summary>
		/// <param name="strDbName">數據庫名稱</param>
		/// <param name="strFileName">備份文件名</param>
		/// <param name="pgbMain">狀態條控件名稱</param>
		/// <param name="strServerName">服務器名稱</param>
		/// <param name="strUserName">用戶名稱</param>
		/// <param name="strPwd">密碼</param>
		/// <returns>恢復成功返回true ，否則返回false</returns>
		public bool RestoreDB(string strDbName,string strFileName, ProgressBar pgbMain,string strServerName,string strUserName,string strPwd ) 
		{ 
			PBar = pgbMain ; 
			SQLDMO.SQLServer svr = new SQLDMO.SQLServerClass() ; 
			try 
			{ 
				ServerName = strServerName ; 
				UserName = strUserName ; 
				Password = strPwd ; 

				svr.Connect(ServerName,UserName,Password) ; 
				SQLDMO.QueryResults qr = svr.EnumProcesses(-1) ; 
				int iColPIDNum = -1 ; 
				int iColDbName = -1 ; 
				for(int i=1;i<=qr.Columns;i++) 
				{ 
					string strName = qr.get_ColumnName(i) ; 
					if (strName.ToUpper().Trim() == "SPID") 
					{ 
						iColPIDNum = i ; 
					} 
					else if (strName.ToUpper().Trim() == "DBNAME") 
					{ 
						iColDbName = i ; 
					} 
					if (iColPIDNum != -1 && iColDbName != -1) 
						break ; 
				} 

				for(int i=1;i<=qr.Rows;i++) 
				{ 
					int lPID = qr.GetColumnLong(i,iColPIDNum) ; 
					string strDBName = qr.GetColumnString(i,iColDbName) ; 
					if (strDBName.ToUpper() == strDbName.ToUpper()) 
						svr.KillProcess(lPID) ; 
				} 

				SQLDMO.Restore res = new SQLDMO.RestoreClass() ; 
				res.Action = 0 ; 
				SQLDMO.RestoreSink_PercentCompleteEventHandler pceh = new SQLDMO.RestoreSink_PercentCompleteEventHandler(Step); 
				res.PercentComplete += pceh; 
				res.Files = strFileName ; 

				res.Database = strDbName ; 
				res.ReplaceDatabase = true ; 
				res.SQLRestore(svr) ; 
				return true ; 
			} 
			catch(Exception err) 
			{ 
//				throw(new Exception("恢復數據庫失敗,請關閉所有和該數據庫連接的程序！"+err.Message)) ; 
				message = "恢復數據庫失敗,請關閉所有和該數據庫連接的程序！" +err.Message;
				return false;
			} 
			finally 
			{ 
				svr.DisConnect() ; 
			} 
		} 
	}
}