PageRenderTime 65ms CodeModel.GetById 26ms RepoModel.GetById 0ms app.codeStats 0ms

/branches/HP10.2/Source/nHydrate.DataImport.SqlClient/SchemaModelHelper.cs

#
C# | 489 lines | 417 code | 40 blank | 32 comment | 62 complexity | 0d9fc58a4be6f8b56f5fcc4d7d94ab3b MD5 | raw file
Possible License(s): JSON, CC-BY-SA-3.0
  1. #region Copyright (c) 2006-2013 nHydrate.org, All Rights Reserved
  2. // -------------------------------------------------------------------------- *
  3. // NHYDRATE.ORG *
  4. // Copyright (c) 2006-2013 All Rights reserved *
  5. // *
  6. // *
  7. // Permission is hereby granted, free of charge, to any person obtaining a *
  8. // copy of this software and associated documentation files (the "Software"), *
  9. // to deal in the Software without restriction, including without limitation *
  10. // the rights to use, copy, modify, merge, publish, distribute, sublicense, *
  11. // and/or sell copies of the Software, and to permit persons to whom the *
  12. // Software is furnished to do so, subject to the following conditions: *
  13. // *
  14. // The above copyright notice and this permission notice shall be included *
  15. // in all copies or substantial portions of the Software. *
  16. // *
  17. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, *
  18. // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES *
  19. // OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. *
  20. // IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY *
  21. // CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, *
  22. // TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE *
  23. // SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. *
  24. // -------------------------------------------------------------------------- *
  25. #endregion
  26. using System;
  27. using System.Collections.Generic;
  28. using System.Linq;
  29. using System.Text;
  30. using System.Data;
  31. using System.Text.RegularExpressions;
  32. namespace nHydrate.DataImport.SqlClient
  33. {
  34. public class SchemaModelHelper : ISchemaModelHelper
  35. {
  36. #region Public Methods
  37. public bool IsValidConnectionString(string connectionString)
  38. {
  39. var valid = false;
  40. var conn = new System.Data.SqlClient.SqlConnection();
  41. try
  42. {
  43. conn.ConnectionString = connectionString;
  44. conn.Open();
  45. valid = true;
  46. }
  47. catch (Exception ex)
  48. {
  49. valid = false;
  50. }
  51. finally
  52. {
  53. conn.Close();
  54. }
  55. return valid;
  56. }
  57. public bool IsSupportedSQLVersion(string connectionString)
  58. {
  59. var ds = DatabaseHelper.ExecuteDataset(connectionString, "SELECT SERVERPROPERTY('productversion')");
  60. var version = (string)ds.Tables[0].Rows[0][0];
  61. if (version.StartsWith("10."))
  62. return true;
  63. else if (version.StartsWith("9."))
  64. return true;
  65. else
  66. return false;
  67. }
  68. public SQLServerTypeConstants GetSQLVersion(string connectionString)
  69. {
  70. var ds = DatabaseHelper.ExecuteDataset(connectionString, "SELECT SERVERPROPERTY('productversion')");
  71. var version = (string)ds.Tables[0].Rows[0][0];
  72. if (version.StartsWith("10."))
  73. {
  74. var ds2 = DatabaseHelper.ExecuteDataset(connectionString, "SELECT SERVERPROPERTY('Edition')");
  75. var version2 = (string)ds2.Tables[0].Rows[0][0];
  76. if (version2 == "SQL Azure")
  77. return SQLServerTypeConstants.SQLAzure;
  78. else
  79. return SQLServerTypeConstants.SQL2008;
  80. }
  81. else
  82. {
  83. return SQLServerTypeConstants.SQL2005;
  84. }
  85. }
  86. #endregion
  87. internal static string GetSqlDatabaseTables()
  88. {
  89. var sb = new StringBuilder();
  90. sb.AppendLine("DECLARE @bar varchar(150)");
  91. sb.AppendLine("DECLARE @val varchar(150)");
  92. sb.AppendLine("DECLARE @tab table");
  93. sb.AppendLine("(");
  94. sb.AppendLine("xName varchar(150) NOT NULL,");
  95. sb.AppendLine("xValue varchar(150) NULL,");
  96. sb.AppendLine("xSchema varchar(150) NOT NULL");
  97. sb.AppendLine(")");
  98. sb.AppendLine("INSERT INTO @tab SELECT so.name, null, sc.name [schema] FROM sys.tables so INNER JOIN sys.schemas sc ON so.schema_id = sc.schema_id WHERE so.name <> 'dtproperties' AND (so.name <> 'sysdiagrams') AND (so.name <> '__nhydrateschema')AND (so.name <> '__nhydrateobjects') AND NOT (so.name like '__AUDIT__%')");
  99. sb.AppendLine("select xName as name, xSchema as [schema], xValue selectionCriteria from @tab WHERE xName <> 'dtproperties' ORDER BY xName");
  100. return sb.ToString();
  101. }
  102. internal static string GetSqlColumnsForTable()
  103. {
  104. return GetSqlColumnsForTable(null);
  105. }
  106. internal static string GetSqlForUniqueConstraints()
  107. {
  108. var sb = new StringBuilder();
  109. sb.AppendLine("select o.name as TableName, col.name as ColumnName");
  110. sb.AppendLine("from sys.indexes i inner join sys.objects o on i.object_id = o.object_id ");
  111. sb.AppendLine(" inner join sys.index_columns ic on i.index_id = ic.index_id and ic.object_id = o.object_id");
  112. sb.AppendLine(" INNER JOIN sys.columns col ON ic.object_id = col.object_id and ic.column_id = col.column_id");
  113. sb.AppendLine("where i.is_unique = 1 and is_primary_key = 0 and is_unique_constraint = 1");
  114. return sb.ToString();
  115. }
  116. internal static string GetSqlForIndexes()
  117. {
  118. var sb = new StringBuilder();
  119. sb.AppendLine();
  120. sb.AppendLine("SELECT ");
  121. sb.AppendLine(" ind.name as indexname");
  122. sb.AppendLine(" ,ind.is_primary_key");
  123. sb.AppendLine(" --,ind.index_id ");
  124. sb.AppendLine(" --,ic.index_column_id ");
  125. sb.AppendLine(" ,t.name as tablename");
  126. sb.AppendLine(" ,col.name as columnname");
  127. sb.AppendLine(" ,ic.is_descending_key");
  128. sb.AppendLine(" ,ic.key_ordinal");
  129. sb.AppendLine(" ,ind.type_desc");
  130. sb.AppendLine(" ,ind.is_unique_constraint");
  131. sb.AppendLine(" ,ind.is_primary_key");
  132. sb.AppendLine(" --,ic.* ");
  133. sb.AppendLine(" --,col.* ");
  134. sb.AppendLine(" --,ind.* ");
  135. sb.AppendLine("FROM sys.indexes ind ");
  136. sb.AppendLine("INNER JOIN sys.index_columns ic ");
  137. sb.AppendLine(" ON ind.object_id = ic.object_id and ind.index_id = ic.index_id ");
  138. sb.AppendLine("INNER JOIN sys.columns col ");
  139. sb.AppendLine(" ON ic.object_id = col.object_id and ic.column_id = col.column_id ");
  140. sb.AppendLine("INNER JOIN sys.tables t ");
  141. sb.AppendLine(" ON ind.object_id = t.object_id ");
  142. sb.AppendLine("WHERE (1=1) ");
  143. //sb.AppendLine(" AND ind.is_primary_key = 0 ");
  144. //sb.AppendLine(" AND ind.is_unique = 0 ");
  145. //sb.AppendLine(" AND ind.is_unique_constraint = 0 ");
  146. sb.AppendLine(" AND t.is_ms_shipped = 0 ");
  147. sb.AppendLine(" AND ic.key_ordinal <> 0");
  148. sb.AppendLine("ORDER BY ");
  149. sb.AppendLine(" ind.name, ic.key_ordinal");
  150. return sb.ToString();
  151. }
  152. internal static string GetSqlColumnsForComputed()
  153. {
  154. var sb = new StringBuilder();
  155. sb.AppendLine("select o.name as tablename, c.name as columnname, c.definition");
  156. sb.AppendLine("from sys.computed_columns c inner join sys.objects o on c.object_id = o.object_id");
  157. return sb.ToString();
  158. }
  159. internal static string GetSqlColumnsForTable(string tableName)
  160. {
  161. var sb = new StringBuilder();
  162. sb.AppendLine("SELECT");
  163. sb.AppendLine(" c.ORDINAL_POSITION as colorder,");
  164. sb.AppendLine(" c.TABLE_NAME as tablename,");
  165. sb.AppendLine(" c.COLUMN_NAME as columnname,");
  166. sb.AppendLine("(");
  167. sb.AppendLine("select top 1 c1.name");
  168. sb.AppendLine("from sys.indexes i");
  169. sb.AppendLine("join sysobjects o ON i.object_id = o.id");
  170. sb.AppendLine("join sysobjects pk ON i.name = pk.name");
  171. sb.AppendLine("AND pk.parent_obj = i.object_id");
  172. sb.AppendLine("AND pk.xtype = 'PK'");
  173. sb.AppendLine("join sys.index_columns ik on i.object_id = ik.object_id");
  174. sb.AppendLine("and i.index_id = ik.index_id");
  175. sb.AppendLine("join syscolumns c1 ON ik.object_id = c1.id");
  176. sb.AppendLine("AND ik.column_id = c1.colid");
  177. sb.AppendLine("AND c1.name = c.COLUMN_NAME");
  178. sb.AppendLine("where o.name = c.TABLE_NAME");
  179. sb.AppendLine(") as [isPrimaryKey],");
  180. sb.AppendLine(" case WHEN");
  181. sb.AppendLine(" (");
  182. sb.AppendLine(" SELECT ");
  183. sb.AppendLine(" count(*) ");
  184. sb.AppendLine(" FROM ");
  185. sb.AppendLine(" INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE foreignkeyccu");
  186. sb.AppendLine(" INNER JOIN INFORMATION_SCHEMA.TABLE_CONSTRAINTS foreignkeytc on foreignkeyccu.CONSTRAINT_NAME = foreignkeytc.CONSTRAINT_NAME AND");
  187. sb.AppendLine(" foreignkeyccu.CONSTRAINT_SCHEMA = foreignkeytc.CONSTRAINT_SCHEMA AND");
  188. sb.AppendLine(" foreignkeytc.CONSTRAINT_TYPE = 'FOREIGN KEY'");
  189. sb.AppendLine(" WHERE");
  190. sb.AppendLine(" foreignkeyccu.TABLE_SCHEMA = c.TABLE_SCHEMA AND");
  191. sb.AppendLine(" foreignkeyccu.TABLE_NAME = c.TABLE_NAME AND");
  192. sb.AppendLine(" foreignkeyccu.COLUMN_NAME = c.COLUMN_NAME ");
  193. sb.AppendLine(" ) > 0 THEN 'true' ELSE 'false' END as isForeignKey,");
  194. sb.AppendLine(" c.DATA_TYPE as datatype,");
  195. sb.AppendLine(" s.xtype,");
  196. sb.AppendLine(" c.numeric_precision AS [precision], c.numeric_scale AS [scale],");
  197. sb.AppendLine(" case when c.CHARACTER_MAXIMUM_LENGTH is null or c.CHARACTER_MAXIMUM_LENGTH > 8000 then s.length else c.CHARACTER_MAXIMUM_LENGTH end as length,");
  198. sb.AppendLine(" case when c.IS_NULLABLE = 'No' then 'false' else 'true' end as allowNull, ");
  199. sb.AppendLine(" case when c.COLUMN_DEFAULT is null then '' else c.COLUMN_DEFAULT end as defaultValue,");
  200. sb.AppendLine(" case when COLUMNPROPERTY(OBJECT_ID(c.TABLE_SCHEMA+'.'+c.TABLE_NAME), COLUMN_NAME, 'IsIdentity') = 1 then 'true' else 'false' end as isIdentity,");
  201. sb.AppendLine(" c.COLLATION_NAME AS collation");
  202. sb.AppendLine(" FROM ");
  203. sb.AppendLine(" INFORMATION_SCHEMA.COLUMNS c ");
  204. sb.AppendLine(" INNER JOIN systypes s on s.name = c.DATA_TYPE");
  205. if (!string.IsNullOrEmpty(tableName))
  206. sb.AppendLine(" WHERE c.TABLE_NAME = '" + tableName + "'");
  207. sb.AppendLine(" ORDER BY");
  208. sb.AppendLine(" c.TABLE_NAME,");
  209. sb.AppendLine(" c.ORDINAL_POSITION");
  210. return sb.ToString();
  211. }
  212. internal static string GetSqlForRelationships()
  213. {
  214. var sb = new StringBuilder();
  215. sb.AppendLine("SELECT ");
  216. sb.AppendLine(" KCU1.CONSTRAINT_NAME AS 'FK_CONSTRAINT_NAME'");
  217. sb.AppendLine(" , KCU1.TABLE_NAME AS 'FK_TABLE_NAME'");
  218. sb.AppendLine(" , KCU1.COLUMN_NAME AS 'FK_COLUMN_NAME' ");
  219. sb.AppendLine(" , KCU2.TABLE_NAME AS 'UQ_TABLE_NAME'");
  220. sb.AppendLine(" , KCU2.COLUMN_NAME AS 'UQ_COLUMN_NAME'");
  221. sb.AppendLine(" , so.id");
  222. sb.AppendLine("FROM ");
  223. sb.AppendLine(" INFORMATION_SCHEMA.REFERENTIAL_CONSTRAINTS RC");
  224. sb.AppendLine(" JOIN INFORMATION_SCHEMA.KEY_COLUMN_USAGE KCU1");
  225. sb.AppendLine(" ON KCU1.CONSTRAINT_CATALOG = RC.CONSTRAINT_CATALOG ");
  226. sb.AppendLine(" AND KCU1.CONSTRAINT_SCHEMA = RC.CONSTRAINT_SCHEMA");
  227. sb.AppendLine(" AND KCU1.CONSTRAINT_NAME = RC.CONSTRAINT_NAME");
  228. sb.AppendLine("JOIN INFORMATION_SCHEMA.KEY_COLUMN_USAGE KCU2");
  229. sb.AppendLine(" ON KCU2.CONSTRAINT_CATALOG = RC.UNIQUE_CONSTRAINT_CATALOG ");
  230. sb.AppendLine(" AND KCU2.CONSTRAINT_SCHEMA = RC.UNIQUE_CONSTRAINT_SCHEMA");
  231. sb.AppendLine(" AND KCU2.CONSTRAINT_NAME = RC.UNIQUE_CONSTRAINT_NAME");
  232. sb.AppendLine(" AND KCU2.ORDINAL_POSITION = KCU1.ORDINAL_POSITION");
  233. sb.AppendLine("JOIN sysobjects so");
  234. sb.AppendLine(" ON KCU1.CONSTRAINT_NAME = so.name");
  235. sb.AppendLine("WHERE");
  236. sb.AppendLine(" so.xtype = 'F'");
  237. sb.AppendLine("ORDER BY");
  238. sb.AppendLine(" KCU1.CONSTRAINT_NAME,");
  239. sb.AppendLine(" KCU1.ORDINAL_POSITION");
  240. return sb.ToString();
  241. }
  242. internal static string GetSqlIndexesForTable()
  243. {
  244. var sb = new StringBuilder();
  245. sb.AppendLine("select t.name as tablename, i.name as indexname, c.name as columnname, i.is_primary_key");
  246. sb.AppendLine("from sys.tables t");
  247. sb.AppendLine("inner join sys.indexes i on i.object_id = t.object_id");
  248. sb.AppendLine("inner join sys.index_columns ic on ic.object_id = t.object_id");
  249. sb.AppendLine("inner join sys.columns c on c.object_id = t.object_id and");
  250. sb.AppendLine("ic.column_id = c.column_id");
  251. //sb.AppendLine("select o.name as tablename, i.name as indexname, i.is_primary_key from sys.objects o inner join sys.indexes i on o.object_id = i.object_id where o.[type] = 'U'");
  252. return sb.ToString();
  253. }
  254. internal static string GetSqlForViews()
  255. {
  256. var sb = new StringBuilder();
  257. sb.AppendLine("select s.name as schemaname, v.name, m.definition from sys.views v inner join sys.sql_modules m on v.object_id = m.object_id inner join sys.schemas s on s.schema_id= v.schema_id");
  258. return sb.ToString();
  259. }
  260. internal static string GetViewBody(string sql)
  261. {
  262. var regEx = new Regex(@"CREATE VIEW[\r\n\s]*[a-zA-Z0-9\[\]_\.]*[\r\n\s]*AS[\r\n\s]*([\s\S\r\n]*)", RegexOptions.IgnoreCase);
  263. var match = regEx.Match(sql);
  264. if (match != null && match.Groups != null && match.Groups.Count == 2)
  265. sql = match.Groups[1].Value;
  266. else
  267. {
  268. sql = sql.Replace("\r", string.Empty);
  269. var arr = sql.Split('\n').ToList();
  270. var sb = new StringBuilder();
  271. var inBody = false;
  272. foreach (var lineText in arr)
  273. {
  274. //This is FAR from perfect. It assumes the creation line ends with the "AS" keyword for a stored proc
  275. if (inBody)
  276. {
  277. sb.AppendLine(lineText);
  278. }
  279. else if (!inBody && (lineText.ToLower().Trim().EndsWith(" as") || lineText.ToLower().Trim() == "as"))
  280. {
  281. inBody = true;
  282. }
  283. }
  284. sql = sb.ToString();
  285. }
  286. return sql.Trim();
  287. }
  288. internal static string GetSqlForViewsColumns()
  289. {
  290. var sb = new StringBuilder();
  291. sb.AppendLine("select v.name as viewname, c.name as columnname, c.system_type_id, c.max_length, c.precision, c.scale, c.is_nullable from sys.views v inner join sys.columns c on v.object_id = c.object_id order by v.name, c.name");
  292. return sb.ToString();
  293. }
  294. internal static string GetSqlForFunctions()
  295. {
  296. var sb = new StringBuilder();
  297. sb.AppendLine("select o.*, s.name as schemaname from sys.objects o inner join sys.schemas s on o.schema_id = s.schema_id WHERE [type] IN ('FN', 'IF', 'TF') and o.name <> 'fn_diagramobjects'");
  298. return sb.ToString();
  299. }
  300. internal static string GetFunctionBody(string schema, string name, string connectionString)
  301. {
  302. var sb = new StringBuilder();
  303. var ds = DatabaseHelper.ExecuteDataset(connectionString, "sp_helptext '[" + schema + "].[" + name + "]'");
  304. foreach (DataRow dr in ds.Tables[0].Rows)
  305. {
  306. var t = (string)dr["Text"] + string.Empty;
  307. sb.AppendLine(t.Replace("\r\n", string.Empty).Replace("\r", string.Empty).Replace("\n", string.Empty));
  308. }
  309. var sql = sb.ToString();
  310. var regEx = new Regex(@"CREATE\s*FUNCTION[\r\n\s]*[a-zA-Z0-9\[\]_\.]*.*RETURNS.*AS[\r\n\s]+(RETURN[\s\S\r\n]*)", RegexOptions.IgnoreCase | RegexOptions.Singleline);
  311. var match = regEx.Match(sql);
  312. if (match != null && match.Groups != null && match.Groups.Count == 2)
  313. {
  314. sql = match.Groups[1].Value;
  315. }
  316. else
  317. {
  318. regEx = new Regex(@"CREATE\s*FUNCTION[\r\n\s]*[a-zA-Z0-9\[\]_\.]*.*RETURNS.*(BEGIN[\s\S\r\n]*)", RegexOptions.IgnoreCase | RegexOptions.Singleline);
  319. match = regEx.Match(sql);
  320. if (match != null && match.Groups != null && match.Groups.Count == 2)
  321. {
  322. sql = match.Groups[1].Value;
  323. }
  324. else
  325. {
  326. regEx = new Regex(@"CREATE\s*FUNCTION[\r\n\s]*[a-zA-Z0-9\[\]_\.]*.*RETURNS.*AS[\r\n\s]+([\s\S\r\n]*)", RegexOptions.IgnoreCase | RegexOptions.Singleline);
  327. match = regEx.Match(sql);
  328. if (match != null && match.Groups != null && match.Groups.Count == 2)
  329. sql = match.Groups[1].Value;
  330. else
  331. System.Diagnostics.Debug.Write(string.Empty);
  332. }
  333. }
  334. return sql.Trim();
  335. }
  336. internal static string GetSqlForStoredProceduresParameters()
  337. {
  338. var sb = new StringBuilder();
  339. sb.AppendLine("SELECT dbo.syscolumns.xtype, dbo.sysobjects.name, dbo.sysobjects.id,");
  340. sb.AppendLine(" dbo.syscolumns.name AS ColName,");
  341. sb.AppendLine(" dbo.systypes.name AS ColType,");
  342. sb.AppendLine(" dbo.syscolumns.isnullable,");
  343. sb.AppendLine(" dbo.syscolumns.length,");
  344. sb.AppendLine(" dbo.syscolumns.isoutparam");
  345. sb.AppendLine("FROM dbo.sysobjects INNER JOIN");
  346. sb.AppendLine(" dbo.syscolumns ON dbo.sysobjects.id = dbo.syscolumns.id INNER JOIN");
  347. sb.AppendLine(" dbo.systypes ON dbo.syscolumns.xtype = dbo.systypes.xtype");
  348. sb.AppendLine("WHERE (dbo.sysobjects.category = 0) AND");
  349. sb.AppendLine(" (dbo.sysobjects.xtype = 'P') AND");
  350. sb.AppendLine(" NOT (dbo.sysobjects.name LIKE 'gen_%') AND");
  351. sb.AppendLine(" NOT (dbo.sysobjects.name LIKE 'dt_%') AND");
  352. sb.AppendLine(" NOT (dbo.sysobjects.name LIKE 'sp[_]%diagram%') AND");
  353. sb.AppendLine(" dbo.systypes.name <> 'sysname' AND");
  354. sb.AppendLine(" (dbo.sysobjects.uid in (select uid from dbo.sysusers))");
  355. sb.AppendLine("ORDER BY");
  356. sb.AppendLine(" dbo.sysobjects.name, dbo.syscolumns.name");
  357. return sb.ToString();
  358. }
  359. internal static string GetSqlForStoredProceduresColumns(StoredProc sp)
  360. {
  361. var sb = new StringBuilder();
  362. System.Windows.Forms.Application.DoEvents();
  363. sb.AppendLine("SET FMTONLY ON");
  364. sb.Append("EXEC [" + (string.IsNullOrEmpty(sp.Schema) ? "dbo" : sp.Schema) + "].[" + sp.Name + "] ");
  365. foreach (var parameter in sp.ParameterList)
  366. {
  367. if (parameter.DataType == SqlDbType.UniqueIdentifier)
  368. sb.Append("@" + parameter.Name + "='540C6D43-5645-40FB-980F-2FF126BFBD5E'");
  369. else if (parameter.IsTextType())
  370. sb.Append("@" + parameter.Name + "=''");
  371. else if (parameter.IsNumericType())
  372. sb.Append("@" + parameter.Name + "=0");
  373. else if (parameter.IsBinaryType())
  374. sb.Append("@" + parameter.Name + "=0x0");
  375. else if (parameter.DataType == SqlDbType.Bit)
  376. sb.Append("@" + parameter.Name + "=0");
  377. else if (parameter.IsDateType())
  378. sb.Append("@" + parameter.Name + "='2000-01-01'");
  379. else
  380. System.Diagnostics.Debug.Write(string.Empty);
  381. if (sp.ParameterList.IndexOf(parameter) < sp.ParameterList.Count - 1)
  382. sb.Append(", ");
  383. }
  384. sb.AppendLine();
  385. return sb.ToString();
  386. }
  387. internal static string GetSqlForStoredProceduresBody(string schema, string spName, string connectionString)
  388. {
  389. var sb = new StringBuilder();
  390. var ds = DatabaseHelper.ExecuteDataset(connectionString, "sp_helptext '[" + schema + "].[" + spName + "]'");
  391. if (ds.Tables.Count > 0)
  392. {
  393. foreach (DataRow dr in ds.Tables[0].Rows)
  394. {
  395. sb.Append(((string)dr[0]).Replace("\r", string.Empty));
  396. }
  397. var arr = sb.ToString().Split('\n').ToList();
  398. sb = new StringBuilder();
  399. var inBody = false;
  400. foreach (var lineText in arr)
  401. {
  402. var lineText2 = StripComments(lineText);
  403. //This is FAR from perfect. It assumes the creation line ends with the "AS" keyword for a stored proc
  404. if (inBody)
  405. {
  406. sb.AppendLine(lineText);
  407. }
  408. else if (!inBody && (lineText2.ToLower().Trim().EndsWith(" as") || lineText2.ToLower().Trim() == "as"))
  409. {
  410. inBody = true;
  411. }
  412. }
  413. }
  414. return sb.ToString().Trim();
  415. }
  416. internal static string GetSqlForStoredProcedures()
  417. {
  418. return GetSqlForStoredProcedures(null);
  419. }
  420. internal static string GetSqlForStoredProcedures(string name)
  421. {
  422. var sb = new StringBuilder();
  423. sb.AppendLine("SELECT OBJECT_SCHEMA_NAME(id) as schemaname, dbo.sysobjects.id, dbo.sysobjects.xtype, dbo.sysobjects.name");
  424. sb.AppendLine("FROM dbo.sysobjects");
  425. sb.AppendLine("WHERE (dbo.sysobjects.category = 0) AND");
  426. sb.AppendLine(" (dbo.sysobjects.xtype = 'P') AND");
  427. sb.AppendLine(" NOT (dbo.sysobjects.name LIKE 'gen_%') AND");
  428. sb.AppendLine(" NOT (dbo.sysobjects.name LIKE 'dt_%') AND");
  429. sb.AppendLine(" NOT (dbo.sysobjects.name LIKE 'sp[_]%diagram%') AND");
  430. if (!string.IsNullOrEmpty(name))
  431. {
  432. sb.AppendLine(" (dbo.sysobjects.name = '" + name + "') AND");
  433. }
  434. sb.AppendLine(" (dbo.sysobjects.uid in (select uid from dbo.sysusers))");
  435. sb.AppendLine("ORDER BY dbo.sysobjects.name");
  436. return sb.ToString();
  437. }
  438. /// <summary>
  439. /// Not perfect!! Just strips off what looks like comment
  440. /// </summary>
  441. private static string StripComments(string sql)
  442. {
  443. if (string.IsNullOrEmpty(sql)) return sql;
  444. var index = sql.IndexOf("--");
  445. if (index == -1) return sql;
  446. return sql.Substring(0, index);
  447. }
  448. }
  449. }