/// /// Copyright (c) 2013-2017 Sensus Metering Systems /// using System; using System.IO; using System.Collections.Generic; using System.Windows.Forms; using log4net; using Config.Entities; using TBF.Resources; using TBF.Forms; namespace TBF { public class Program { public const string HomeDir = "C:\\Tbf\\"; /// Contains subdirectories Results, Logs, Images, ... /// log4net static ILog log; const string log4netConfigFName = "log4netConfig.xml"; /// Program information public static string Version; /// version string public static DateTime BuildDateTime; /// date and time of program build /// Local settings public static LocalSettings LocalSettings; public const string LocalSettingsFileName = "config.xml"; public const string LocalSettingsBackupName = "config.backup.xml"; public static System.Globalization.CultureInfo AltCulture; /// /// Selected procedure data /// - before the main program window is open, objects referenced by this variable /// are loaded from the database. /// - user interface displays data stored in objects referenced by this variable. /// When user modifies procedure settings and confirms changes (choses to save them), /// data referenced by this variable are updated and written to the database. /// - when users starts a test procedure, state machine creates its own copy /// of procedure data and uses them for the test. Changes of 'SelectedProcedure' /// done during the test do not have any influence on the currently running test. /// public static Config.Entities.Procedure SelectedProcedure = null; /// /// Main window object. /// public static MainWnd MainWnd; /// /// The main entry point for the application. /// [STAThread] static void Main() { AppDomain.CurrentDomain.UnhandledException += new UnhandledExceptionEventHandler(MyHandler); /// Program version string System.Reflection.Assembly thisAssembly = System.Reflection.Assembly.GetExecutingAssembly(); Version ver = thisAssembly.GetName().Version; Version = string.Format("{0}.{1}.{2}", ver.Major, ver.Minor, ver.Build); BuildDateTime = new System.IO.FileInfo(thisAssembly.Location).LastWriteTime; /// Local program configuration directory including the trailing backslash string locAppDataDir = Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData) + Path.DirectorySeparatorChar + "TestBenchFramework" + Path.DirectorySeparatorChar; /// Check whether local application data directory exists. /// If not, create it and copy the initial configuration to this directory. /// This is done only once after a clean program installation on the first program start-up try { if (!Directory.Exists(locAppDataDir)) { Directory.CreateDirectory(locAppDataDir); Environment.CurrentDirectory = locAppDataDir; string sampleConfig = Application.StartupPath + Path.DirectorySeparatorChar + "SampleConfig" + Path.DirectorySeparatorChar; if (File.Exists(sampleConfig + LocalSettingsFileName)) { File.Copy(sampleConfig + LocalSettingsFileName, LocalSettingsFileName); } if (File.Exists(sampleConfig + log4netConfigFName)) { File.Copy(sampleConfig + log4netConfigFName, log4netConfigFName); } if (File.Exists(sampleConfig + Config.Data.SQLiteDbFName)) { File.Copy(sampleConfig + Config.Data.SQLiteDbFName, Config.Data.SQLiteDbFName); } File.SetAttributes(LocalSettingsFileName, FileAttributes.Normal); File.SetAttributes(log4netConfigFName, FileAttributes.Normal); File.SetAttributes(Config.Data.SQLiteDbFName, FileAttributes.Normal); } else { Environment.CurrentDirectory = locAppDataDir; } } catch (Exception e) { log.Error("A problem when creating the local application data directory and preparing the initial configuration", e); return; } /// Configure and start logging log4net.Config.XmlConfigurator.Configure(new System.IO.FileInfo(locAppDataDir + log4netConfigFName)); log = LogManager.GetLogger(typeof(Program)); log.Fatal("--------------------------------------------------------------------------------"); log.Fatal(string.Format("TBF ver.{0}", Version)); /// /// Load the local settings /// Program.LocalSettings = LocalSettings.Load(Program.LocalSettingsFileName); if (Program.LocalSettings == null || Program.LocalSettings.TestBenches == null) { /// Loading local seetings from regular config file failed. Use the backup Program.LocalSettings = LocalSettings.Load(Program.LocalSettingsBackupName); if (Program.LocalSettings == null || Program.LocalSettings.TestBenches == null) { log.FatalFormat("Local settings: Could not load file {0}, nor {1}.", Program.LocalSettingsFileName, Program.LocalSettingsBackupName); log.Fatal("Application terminated."); return; /// Fatal error } else { LocalSettings.Save(); /// Save the settings to overwrite the wrong file log.FatalFormat("Local settings: Could not load file {0}, successfully loaded {1}", Program.LocalSettingsFileName, Program.LocalSettingsBackupName); log.FatalFormat("BatchNr = {0}", LocalSettings.BatchNr); } } else { /// Loading local seetings from the regular config file was successful. Update the backup File.Copy(Program.LocalSettingsFileName, Program.LocalSettingsBackupName, true); log.FatalFormat("Local settings: Successfully loaded from {0}", Program.LocalSettingsFileName); log.FatalFormat("BatchNr = {0}", LocalSettings.BatchNr); } try { string culture = LocalSettings.Language.Replace('_', '-'); System.Threading.Thread.CurrentThread.CurrentUICulture = new System.Globalization.CultureInfo(culture); } catch (Exception e) { string msg = e.Message; MessageBox.Show("Selected language not supported.\nUsing English.", "Warning", MessageBoxButtons.OK, MessageBoxIcon.Asterisk); System.Threading.Thread.CurrentThread.CurrentUICulture = new System.Globalization.CultureInfo("en"); } //AltCulture = null; AltCulture = new System.Globalization.CultureInfo("SK"); /// This belongs to Application.Run(new MainWnd()) ... /// and should be executed before opening 'loginDlg' Application.EnableVisualStyles(); Application.SetCompatibleTextRenderingDefault(false); /// Ensure that there is at least one bench in the configuration while (Program.LocalSettings.BenchesCount == 0) { log.Warn("No test benches in the local configuration -> display an appropriate dialog."); /// Ask what to do, ask for the password, open 'DatabaseSetingsDlg' and continue on OK if ((new Forms.NoBenchOrDatabaseDlg { Message = Strings.NoBenchMsg }.ShowDialog() != DialogResult.OK) || (new Users.Forms.LoginDlg(true).ShowDialog() == DialogResult.Cancel) || (new BenchesDlg().ShowDialog() == DialogResult.Cancel)) { return; /// Exit program } log.Warn("Loading the local settings again."); Program.LocalSettings = LocalSettings.Load(Program.LocalSettingsFileName); if (Program.LocalSettings == null) return; /// Fatal error } /// User login bool retryLogin = true; /// true = stay in a login loop do { LoginDlgWithBenchSelection loginDlgBench; if (LocalSettings.LastBenchName != null) { loginDlgBench = new LoginDlgWithBenchSelection(LocalSettings.LastBenchName); } else { loginDlgBench = new LoginDlgWithBenchSelection(); } loginDlgBench.Method = LocalSettings.LoginMethod; DialogResult dr = loginDlgBench.ShowDialog(); if (dr == DialogResult.Cancel) { return; /// Login canceled --> Exit application } else { for (int i = 0; i < LocalSettings.BenchesCount; i++) { if (LocalSettings.TestBenches[i].BenchName == loginDlgBench.BenchName) { log.FatalFormat("Test bench name: {0}", loginDlgBench.BenchName); Users.IUser loadedUser = null; try { bool authorized = false; if (User.IsPowerUser(loginDlgBench.Alias, loginDlgBench.Password)) { loadedUser = new User(loginDlgBench.Alias, 6, true); authorized = loadedUser.Authorize(loginDlgBench.Alias, loginDlgBench.Password); } /// /// Try authorisation with Users DB first /// if (!authorized) { bool settingsChanged = false; if (string.IsNullOrEmpty(LocalSettings.TestBenches[i].UsersDBSettings.ConnectionString)) { LocalSettings.TestBenches[i].UsersDBSettings.ConnectionString = LocalSettings.TestBenches[i].ProceduresDBSettings.ConnectionString; LocalSettings.TestBenches[i].UsersDBSettings.DbType = LocalSettings.TestBenches[i].ProceduresDBSettings.DbType; settingsChanged = true; } /// Load and authorise user from the UsersDB database try { Users.Entities.DBType dbType = LocalSettings.TestBenches[i].UsersDBSettings.DbType; string connStr = LocalSettings.TestBenches[i].UsersDBSettings.ConnectionString; switch (loginDlgBench.Method) { default: case Users.Entities.LoginMethod.UserName: loadedUser = Users.Entities.User.LoadUserByName(loginDlgBench.Alias, dbType, connStr); if (loadedUser != null) authorized = loadedUser.Authorize(loginDlgBench.Alias, loginDlgBench.Password); break; case Users.Entities.LoginMethod.FullName: loadedUser = Users.Entities.User.LoadUserByFullName(loginDlgBench.Alias, dbType, connStr); if (loadedUser != null) authorized = loadedUser.AuthorizeFullName(loginDlgBench.Alias, loginDlgBench.Password); break; case Users.Entities.LoginMethod.Number: int number; if (!int.TryParse(loginDlgBench.Alias, out number)) break; loadedUser = Users.Entities.User.LoadUserByNumber(number, dbType, connStr); if (loadedUser != null) authorized = loadedUser.AuthorizeNumber(number, loginDlgBench.Password); break; } } catch { authorized = false; } if (settingsChanged && authorized) LocalSettings.Save(); } /// /// One more attempt: authorisation with Config DB /// if (!authorized) { /// Load and authorise user from the Config database try { switch (loginDlgBench.Method) { default: case Users.Entities.LoginMethod.UserName: loadedUser = Config.Entities.User.LoadUserByName(loginDlgBench.Alias, LocalSettings.TestBenches[i].ProceduresDBSettings); if (loadedUser != null) authorized = loadedUser.Authorize(loginDlgBench.Alias, loginDlgBench.Password); break; case Users.Entities.LoginMethod.FullName: loadedUser = Config.Entities.User.LoadUserByFullName(loginDlgBench.Alias, LocalSettings.TestBenches[i].ProceduresDBSettings); if (loadedUser != null) authorized = loadedUser.AuthorizeFullName(loginDlgBench.Alias, loginDlgBench.Password); break; case Users.Entities.LoginMethod.Number: int number; if (!int.TryParse(loginDlgBench.Alias, out number)) break; loadedUser = Config.Entities.User.LoadUserByNumber(number, LocalSettings.TestBenches[i].ProceduresDBSettings); if (loadedUser != null) authorized = loadedUser.AuthorizeNumber(number, loginDlgBench.Password); break; } } catch { authorized = false; } } if (authorized) { LocalSettings.LoginMethod = loginDlgBench.Method; /// Save the login method actually used if (LocalSettings.TestBenches[i].IsRealBench) { /// This is a real bench ==> check if another instance is running string processName = System.IO.Path.GetFileNameWithoutExtension(thisAssembly.Location); if (System.Diagnostics.Process.GetProcessesByName(processName).Length > 1) { MessageBox.Show(Strings.Another_instance_is_running, Strings.Error, MessageBoxButtons.OK, MessageBoxIcon.Asterisk); return; } } log.InfoFormat("Password accepted, bench '{0}' parameters loaded.", LocalSettings.TestBenches[i].BenchName); /// Copy the selected bench settings to CurrentBench. /// Clone() guarantees that current bench settings wont be modified /// when user modifies the database settings in DatabaseSettingsDlg. Config.Data.CurrentBench = (Config.DatabaseSettings)LocalSettings.TestBenches[i].Clone(); /// Save the selection to local settings LocalSettings.LastBenchName = Config.Data.CurrentBench.BenchName; Results.DB.DbType = Config.Data.CurrentBench.WaterMetersDBSettings.DbType; Results.DB.ConnectionString = Config.Data.CurrentBench.WaterMetersDBSettings.ConnectionString; Results.DB.LoadSharedData(); int maxBatchNr = Results.DB.GetMaxSavedBatchNr(); if (Program.LocalSettings.BatchNr <= maxBatchNr) { log.FatalFormat("Max. BatchNr in DB = {0}, LocalSettings.BatchNr = {1}", maxBatchNr, Program.LocalSettings.BatchNr); Program.LocalSettings.BatchNr = maxBatchNr + 1; Program.LocalSettings.Save(); log.FatalFormat("LocalSettings.BatchNr updated to {0}", Program.LocalSettings.BatchNr); } /// This is to trigger an exception in case of a power user and no Config database IList listOfProcedures = Config.FluentCommon.CreateSession(Users.Entities.DBKind.Config) .QueryOver() .Where(x => (x.ProcedureState == ProcedureState.Active)) .And(x => (x.Name == LocalSettings.LastProcedureName)) .List(); retryLogin = false; break; } if (retryLogin) { /// Close Fluent NHibernate MessageBox.Show(Strings.Invalid_user_password_or_bench, Strings.Error, MessageBoxButtons.OK); } } catch (Exception) { /// Database connect failed, ask what to do switch ((new Forms.NoBenchOrDatabaseDlg { Message = Strings.NoDatabaseMsg, ShowRetry = true }).ShowDialog()) { case DialogResult.Abort: return; /// Exit program case DialogResult.Retry: break; /// Retry the loop default: /// Open 'DatabaseSetingsDlg' and retry DB connect on OK if (new Users.Forms.LoginDlg(true).ShowDialog() == DialogResult.Cancel || new BenchesDlg().ShowDialog() == DialogResult.Cancel) { return; /// Exit program } Program.LocalSettings = LocalSettings.Load(Program.LocalSettingsFileName); if (Program.LocalSettings == null) return; /// Fatal error break; /// Retry the loop } } } // if benchname is OK } // for all benches in local settings loop } // not Canceled } //do while (retryLogin); /// /// Load the last procedure or create a new, default one /// if (LocalSettings.LastProcedureName != null) { IList listOfProcedures = Config.FluentCommon.CreateSession(Users.Entities.DBKind.Config) .QueryOver() .Where(x => (x.ProcedureState == ProcedureState.Active)) .And(x => (x.Name == LocalSettings.LastProcedureName)) .List(); if (listOfProcedures.Count == 1) { log.InfoFormat("Pre-selecting the procedure {0}", LocalSettings.LastProcedureName); SelectedProcedure = listOfProcedures[0]; } } try { /// Open the main application window log.Info("Creating the main window"); MainWnd = new MainWnd(); log.Info("Opening the main window"); Application.Run(MainWnd); log.Info("The main window was closed"); } catch (Exception e) { LogException(log, "Exception in Application.Run(MainWnd)", e); } finally { /// Save local settings LocalSettings.Save(); } /// Close Fluent NHibernate /// Todo log.Info("Exiting application."); } static void MyHandler(object sender, UnhandledExceptionEventArgs args) { LogException(log, "Unhandled exception", (Exception)args.ExceptionObject); } static void LogException(ILog log, string description, Exception e) { log.FatalFormat("---------------( {0} )---------------", description); log.FatalFormat("Message : {0}", e.Message); if (e.InnerException != null) { log.FatalFormat("InnerMessage : {0}", e.InnerException.Message); } log.FatalFormat("StackTrace : {0}{1}", Environment.NewLine, e.StackTrace); log.Fatal("--------------------------------------"); } } }