Initial project's source code check-in.

WebsitePanel.Installer/Sources/WebsitePanel.Setup/Common/SqlProcess.cs

@@ -0,0 +1,184 @@
+// Copyright (c) 2011, Outercurve Foundation.
+// All rights reserved.
+//
+// Redistribution and use in source and binary forms, with or without modification,
+// are permitted provided that the following conditions are met:
+//
+// - Redistributions of source code must  retain  the  above copyright notice, this
+//   list of conditions and the following disclaimer.
+//
+// - Redistributions in binary form  must  reproduce the  above  copyright  notice,
+//   this list of conditions  and  the  following  disclaimer in  the documentation
+//   and/or other materials provided with the distribution.
+//
+// - Neither  the  name  of  the  Outercurve Foundation  nor   the   names  of  its
+//   contributors may be used to endorse or  promote  products  derived  from  this
+//   software without specific prior written permission.
+//
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,  BUT  NOT  LIMITED TO, THE IMPLIED
+// WARRANTIES  OF  MERCHANTABILITY   AND  FITNESS  FOR  A  PARTICULAR  PURPOSE  ARE
+// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
+// ANY DIRECT, INDIRECT, INCIDENTAL,  SPECIAL,  EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+// (INCLUDING, BUT NOT LIMITED TO,  PROCUREMENT  OF  SUBSTITUTE  GOODS OR SERVICES;
+// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)  HOWEVER  CAUSED AND ON
+// ANY  THEORY  OF  LIABILITY,  WHETHER  IN  CONTRACT,  STRICT  LIABILITY,  OR TORT
+// (INCLUDING NEGLIGENCE OR OTHERWISE)  ARISING  IN  ANY WAY OUT OF THE USE OF THIS
+// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+using System;
+using System.Data.SqlClient;
+using System.IO;
+using System.Text;
+using System.Windows.Forms;
+using WebsitePanel.Setup.Actions;
+
+namespace WebsitePanel.Setup
+{
+	/// <summary>
+	/// Shows sql script process.
+	/// </summary>
+	public sealed class SqlProcess
+	{
+		private string scriptFile; 
+		private string connectionString;
+		private string database;
+
+		public event EventHandler<ActionProgressEventArgs<int>> ProgressChange;
+		
+		/// <summary>
+		/// Initializes a new instance of the class.
+		/// </summary>
+		/// <param name="file">Sql script file</param>
+		/// <param name="connection">Sql connection string</param>
+		/// <param name="db">Sql server database name</param>
+		public SqlProcess(string file, string connection, string db)
+		{
+			this.scriptFile = file;
+			this.connectionString = connection;
+			this.database = db;
+		}
+
+		private void OnProgressChange(int percentage)
+		{
+			if (ProgressChange == null)
+				return;
+			//
+			ProgressChange(this, new ActionProgressEventArgs<int>
+			{
+				EventData = percentage
+			});
+		}
+
+		/// <summary>
+		/// Executes sql script file.
+		/// </summary>
+		internal void Run()
+		{
+			int commandCount = 0;
+			int i = 0;
+			string sql = string.Empty;
+
+			try
+			{
+				using (StreamReader sr = new StreamReader(scriptFile))
+				{
+					while( null != (sql = ReadNextStatementFromStream(sr))) 
+					{
+						commandCount++;
+					}					
+				}
+			}
+			catch(Exception ex)
+			{
+				throw new Exception("Can't read SQL script " + scriptFile, ex);
+			}
+
+			Log.WriteInfo(string.Format("Executing {0} database commands", commandCount));
+			//
+			OnProgressChange(0);
+			//
+			SqlConnection connection = new SqlConnection(connectionString);
+
+			try
+			{
+				// iterate through "GO" delimited command text
+				using (StreamReader reader = new StreamReader(scriptFile))
+				{
+					SqlCommand command = new SqlCommand();
+					connection.Open();
+					command.Connection = connection;
+					command.CommandType = System.Data.CommandType.Text;
+					command.CommandTimeout = 600;
+
+					while (null != (sql = ReadNextStatementFromStream(reader)))
+					{
+						sql = ProcessInstallVariables(sql);
+						command.CommandText = sql;
+						try
+						{
+							command.ExecuteNonQuery();
+						}
+						catch (Exception ex)
+						{
+							throw new Exception("Error executing SQL command: " + sql, ex);
+						}
+
+						i++;
+						if (commandCount != 0)
+						{
+							OnProgressChange(Convert.ToInt32(i * 100 / commandCount));
+						}
+					}
+				}
+			}
+			catch (Exception ex)
+			{
+				throw new Exception("Can't run SQL script " + scriptFile, ex);
+			}
+			finally
+			{
+				connection.Close();
+			}
+		}
+
+		private string ReadNextStatementFromStream(StreamReader reader)
+		{
+			StringBuilder sb = new StringBuilder();
+			string lineOfText;
+	
+			while(true) 
+			{
+				lineOfText = reader.ReadLine();
+				if( lineOfText == null ) 
+				{
+					if( sb.Length > 0 ) 
+					{
+						return sb.ToString();
+					}
+					else 
+					{
+						return null;
+					}
+				}
+
+				if(lineOfText.TrimEnd().ToUpper() == "GO") 
+				{
+					break;
+				}
+				
+				sb.Append(lineOfText + Environment.NewLine);
+			}
+
+			return sb.ToString();
+		}
+
+		private string ProcessInstallVariables(string input)
+		{
+			//replace install variables
+			string output = input;
+			output = output.Replace("${install.database}", database);
+			return output;
+		}
+	}
+}