-
Notifications
You must be signed in to change notification settings - Fork 9
/
CommandLineWorkerDba.cs
120 lines (101 loc) · 4.85 KB
/
CommandLineWorkerDba.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#region Copyright Notice
/* Copyright (c) 2017, Deb'jyoti Das - [email protected]
All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are not permitted.Neither the name of the
'Deb'jyoti Das' 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 Deb'jyoti Das '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 Debjyoti OR Deb'jyoti OR Debojyoti Das OR Eyedia 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.
*/
#region Developer Information
/*
Author - Debjyoti Das ([email protected])
Created - 11/12/2017 3:31:31 PM
Description -
Modified By -
Description -
*/
#endregion Developer Information
#endregion Copyright Notice
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Eyedia.Aarbac.Framework;
using System.Configuration;
namespace Eyedia.Aarbac.Command
{
public class CommandLineWorkerDba: CommandLineWorker
{
const string __csformat = "data source={0};Integrated security=True;";
const string __csformatuser = "data source={0};User Id={1};Password={2};";
public CommandLineWorkerDba():base()
{
}
/// <summary>
/// Create new rbac database on specified MS Sql Server instance. usage: aarbac -c id -s servername [-i false] [-u username] [-p password] [-n dbname]
/// </summary>
/// <param name="options"></param>
public void CreateNew(Options options)
{
bool errored = false;
if (string.IsNullOrEmpty(options.Server))
{
WriteErrorLine("Sql server name is required. Please use -s <servername>");
errored = true;
}
if (!options.IntegratedSecurity)
{
if (string.IsNullOrEmpty(options.UserName))
{
WriteErrorLine("Sql server user name is required when integrated security is false. Please use -u <username>");
errored = true;
}
if (string.IsNullOrEmpty(options.Password))
{
WriteErrorLine("Sql server name user password is required required when integrated security is false. Please use -p <password>");
}
Console.WriteLine("Please use -is <true/false> to set integrated security");
}
if (string.IsNullOrEmpty(options.Name))
options.Name = "aarbac";
if (errored)
return;
string connectionString = string.Empty;
if (options.IntegratedSecurity)
connectionString = string.Format(__csformat, options.Server);
else
connectionString = string.Format(__csformatuser, options.Server, options.UserName, options.Password);
RbacDba dba = new RbacDba(connectionString);
if (dba.DatabaseExists(options.Name))
{
WriteErrorLine("Database {0} already exist!", options.Name);
return;
}
Console.Write("Contacting server...".PadRight(RbacDba.__maxInfoLen, '.'));
new RbacDba(connectionString).CreateDatabase(options.Name);
WriteColor(ConsoleColor.Green, "All done!" + Environment.NewLine);
Console.WriteLine();
string connectionStringWithDb = connectionString + "Initial Catalog=" + options.Name;
var config = ConfigurationManager.OpenExeConfiguration(ConfigurationUserLevel.None);
var connectionStringsSection = (ConnectionStringsSection)config.GetSection("connectionStrings");
if (connectionStringsSection.ConnectionStrings["aarbac"] != null)
connectionStringsSection.ConnectionStrings["aarbac"].ConnectionString = connectionStringWithDb;
else
connectionStringsSection.ConnectionStrings.Add(new ConnectionStringSettings("aarbac", connectionStringWithDb));
config.Save();
ConfigurationManager.RefreshSection("connectionStrings");
}
}
}