private static string QuoteIdentifier(ISqlDialect dialect, SqlFormatProperties props, string ident, SymbolPosition original)
{
switch (props.IdentifierQuoteMode)
{
case SqlIdentifierQuoteMode.Original:
if (original != null)
{
return(GetCasedString(original.GetOriginalToken(), props.IdentifierCase));
}
return(dialect.QuoteIdentifier(GetCasedString(ident, props.IdentifierCase)));
break;
case SqlIdentifierQuoteMode.Plain:
if (MustBeQuoted(ident, dialect))
{
return(dialect.QuoteIdentifier(GetCasedString(ident, props.IdentifierCase)));
}
return(GetCasedString(ident, props.IdentifierCase));
break;
case SqlIdentifierQuoteMode.Quoted:
return(dialect.QuoteIdentifier(GetCasedString(ident, props.IdentifierCase)));
break;
}
throw new InternalError("DAE-00048 Unexpected idquote mode");
}
private static void DumpSeparatorIfNeeded(StringBuilder sb, SqlFormatProperties props, SqlFormatterState state, SymbolPosition orig)
{
if (state == null)
{
return;
}
if (state.LineFeedNeeded)
{
DumpEoln(sb, props, state);
state.LineFeedNeeded = false;
state.SeparatorNeeded = false;
state.WasDataOnCurrentLine = false;
}
if (state.SeparatorNeeded && state.WasDataOnCurrentLine)
{
if (props.IndentationLevel == SqlIndentationLevel.Original)
{
if (orig != null)
{
sb.Append(orig.GetOriginalSeparator());
}
else
{
sb.Append(' ');
}
}
else
{
sb.Append(' ');
}
state.SeparatorNeeded = false;
}
}
public SqlDumper(ISqlOutputStream stream, ISqlDialect dialect, SqlFormatProperties props)
{
m_stream = stream;
m_props = props;
m_dialect = dialect;
m_DDA = dialect.CreateDataAdapter();
m_formatterState.DDA = m_DDA;
}
public static SqlFormatProperties CreateOriginal()
{
var res = new SqlFormatProperties();
res.IndentationLevel = SqlIndentationLevel.Original;
res.SqlCommandCase = CharacterCase.Original;
res.CleanupSpecificObjectCode = false;
return res;
}
private static string QuoteFullName(ISqlDialect dialect, SqlFormatProperties props, NameWithSchema name)
{
if (props.QualifierMode == SqlQualifierMode.OmitAll || name.Schema == null)
{
return(QuoteIdentifier(dialect, props, name.Name, null));
}
return(QuoteIdentifier(dialect, props, name.Schema, null) + "." + QuoteIdentifier(dialect, props, name.Name, null));
}
private string GetSqlText()
{
var props = new SqlFormatProperties();
props.IdentifierQuoteMode = m_settings.QuoteIdentifiers ? SqlIdentifierQuoteMode.Quoted : SqlIdentifierQuoteMode.Plain;
props.SqlCommandCase = m_settings.KeywordCase == CharacterCase2.Upper ? CharacterCase.Upper : CharacterCase.Lower;
props.QualifierMode = m_settings.AddSchema ? SqlQualifierMode.Original : SqlQualifierMode.OmitAll;
return(m_anal.Dialect.GenerateScript(GenerateSql, props));
}
public GenerateDataSqlJobCommand(IDataSqlGenerator generator, IFilePlace place, ISqlDialect dialect, TableDataFrame dataFrame, DataFrameRowsExtractor rows, ConnectionPack connpack, SqlFormatProperties formatProps)
{
Generator = generator;
Place = place;
Dialect = dialect;
ConnPack = connpack;
m_rows = rows;
m_dataFrame = dataFrame;
FormatProps = formatProps;
}
public GenerateSqlJobCommand(IAppObjectSqlGenerator generator, IFilePlace place, ISqlDialect dialect, AppObject[] objs, ConnectionPack connpack, SqlFormatProperties formatProps)
: this()
{
Generator = generator;
Place = place;
Dialect = dialect;
Args.AddRange(objs);
ConnPack = connpack;
FormatProps = formatProps;
}
public static string GetSqlLiteral(SqlFormatProperties props, IDialectDataAdapter dda, SqlFormatterState state, IBedValueReader reader, DbTypeBase dsttype)
{
if (props.BinaryStrings)
{
switch (reader.GetFieldType())
{
case TypeStorage.String:
if (props.BinaryStrings)
{
return(dda.GetSqlLiteral(props.RealBinaryEncoding.GetBytes(reader.GetString()), dsttype));
}
break;
}
}
return(dda.GetSqlLiteral(reader, dsttype));
}
private static void DumpEoln(StringBuilder sb, SqlFormatProperties props, SqlFormatterState state)
{
if (props.IndentationLevel != SqlIndentationLevel.SingleLine)
{
sb.Append("\n");
if (state != null)
{
for (int j = 0; j < state.IndentLevel * props.Indentation; j++)
{
sb.Append(" ");
}
}
}
else
{
sb.Append(" ");
}
}
public void Edit()
{
if (!EditEnabled())
{
return;
}
var dbconn = this.FindDatabaseConnection(ConnPack);
if (dbconn.DatabaseCaps.ExecuteSql)
{
OpenQueryParameters pars = new OpenQueryParameters();
IPhysicalConnection newconn = Connection.CreateConnection();
string dbname = DatabaseName;
if (dbname != null)
{
newconn.AfterOpen += ConnTools.ChangeDatabaseCallback(dbname);
}
pars.GenerateSql = delegate(IPhysicalConnection conn)
{
var props = new SqlFormatProperties {
OmitVersionTests = true
};
var dbs = conn.PhysicalFactory.CreateDatabaseSource(conn, dbname);
var so = new SpecificObjectStructure(dbs.LoadSpecificObjectDetail(DbObjectType, DbObjectName));
if (dbconn.Dialect != null)
{
so.CreateSql = dbconn.Dialect.ReformatSpecificObject(so.ObjectType, so.CreateSql);
}
string drop = newconn.Dialect.GenerateScript(dmp => { dmp.DropSpecificObject(so); }, props);
string create = newconn.Dialect.GenerateScript(dmp => { dmp.CreateSpecificObject(so); }, props);
return(drop + "\n\nGO\n\n" + create);
};
MainWindow.Instance.OpenContent(new QueryFrame(newconn, pars));
}
else
{
var frm = new SpecificObjectFrame(dbconn.CloneSource(), LoadStructure(), new ObjectEditorPars {
});
MainWindow.Instance.OpenContent(frm);
}
}
public static ISqlDumper CreateDumper(this ISqlDialect dialect, TextWriter tw, SqlFormatProperties props)
{
SqlOutputStream sqlo = new SqlOutputStream(dialect, tw, props);
return(dialect.CreateDumper(sqlo, props));
}
private static void WriteFormattedValue(ISqlDialect dialect, SqlFormatProperties props, StringBuilder sb, object val, char fmt, object orig, SqlFormatterState state, IDialectDataAdapter dda)
{
switch (fmt)
{
case 'i': // quote identifier
DumpSeparatorIfNeeded(sb, props, state, (SymbolPosition)orig);
if (val is string)
{
sb.Append(QuoteIdentifier(dialect, props, (string)val, (SymbolPosition)orig));
}
else if (val is IColumnReference)
{
sb.Append(QuoteIdentifier(dialect, props, ((IColumnReference)val).ColumnName, (SymbolPosition)orig));
}
else
{
throw new InternalError("DAE-00043 Identifier must be of type string or IColumnReference");
}
DataDumped(state);
break;
case 'f': // quote full name
DumpSeparatorIfNeeded(sb, props, state, (SymbolPosition)orig);
if (val is NameWithSchema)
{
sb.Append(QuoteFullName(dialect, props, (NameWithSchema)val));
}
else if (val is IFullNamedObject)
{
sb.Append(QuoteFullName(dialect, props, ((IFullNamedObject)val).FullName));
}
else
{
throw new InternalError("DAE-00044 Full name must be of type NameWithSchema or IFullNamedObject");
}
DataDumped(state);
break;
case 's': // string - copy character data
if (val != null)
{
DumpSeparatorIfNeeded(sb, props, state, (SymbolPosition)orig);
sb.Append(val.ToString());
DataDumped(state);
}
break;
case 'k': // keyword
DumpSeparatorIfNeeded(sb, props, state, (SymbolPosition)orig);
if (!(val is string))
{
throw new InternalError("DAE-00045 Identifier must be of type string");
}
if (props.SqlCommandCase == CharacterCase.Original && orig != null)
{
sb.Append(((SymbolPosition)orig).GetOriginalToken());
}
else
{
foreach (char c2 in (string)val)
{
sb.Append(GetCasedChar(c2, props.SqlCommandCase));
}
}
DataDumped(state);
break;
case 'K': // multi-word keyword
if (!(val is IEnumerable <string>))
{
throw new InternalError("DAE-00046 Identifier must be of type string");
}
if (orig != null)
{
foreach (var sym in ((IEnumerable <SymbolPosition>)orig))
{
DumpSeparatorIfNeeded(sb, props, state, sym);
sb.Append(GetCasedString(sym.GetOriginalToken(), props.SqlCommandCase));
if (state != null)
{
state.SeparatorNeeded = true;
}
else
{
sb.Append(" ");
}
DataDumped(state);
}
}
else
{
foreach (string s in ((IEnumerable <string>)val))
{
DumpSeparatorIfNeeded(sb, props, state, null);
sb.Append(GetCasedString(s, props.SqlCommandCase));
if (state != null)
{
state.SeparatorNeeded = true;
}
else
{
sb.Append(" ");
}
DataDumped(state);
}
}
break;
case 'v': // value - copy character data
DumpSeparatorIfNeeded(sb, props, state, (SymbolPosition)orig);
if (props.UseOriginalValues && orig != null)
{
sb.Append(((SymbolPosition)orig).GetOriginalToken());
}
else
{
var vth = val as ValueTypeHolder;
if (vth != null)
{
sb.Append(GetSqlLiteralAndRead(props, dda, state, vth.Value, vth.DbType));
}
else
{
sb.Append(GetSqlLiteralAndRead(props, dda, state, val));
}
}
DataDumped(state);
break;
case 't': // version test
if (val != null && !props.OmitVersionTests)
{
sb.Append(val.ToString());
}
break;
default:
throw new InternalError("DAE-00047 Unknown format character: " + fmt);
}
}
public static string Format(ISqlDialect dialect, SqlFormatProperties props, SqlFormatterState state, string format, params object[] args)
{
IDialectDataAdapter dda = null;
if (state != null)
{
dda = state.DDA;
}
if (dda == null)
{
dda = dialect.CreateDataAdapter();
}
int argindex = 0;
StringBuilder sb = new StringBuilder();
int i = 0;
while (i < format.Length)
{
char c = format[i];
switch (c)
{
case '^': // SQL keyword
{
i++;
SymbolPosition original = null;
if (format[i] == ':')
{
original = (SymbolPosition)args[argindex];
argindex++;
i++;
}
DumpSeparatorIfNeeded(sb, props, state, original);
while (i < format.Length && (Char.IsLetter(format, i) || format[i] == '_'))
{
sb.Append(GetCasedChar(format[i], props.SqlCommandCase));
i++;
}
DataDumped(state);
}
break;
case '&': // indentation & spacing
{
i++;
c = format[i];
i++;
char level = '0';
if (c == '1' || c == '2' || c == '3' || c == '5')
{
level = c;
c = format[i];
i++;
}
if (level != '0')
{
// indentation levels
if (props.IndentationLevel == SqlIndentationLevel.Original || props.IndentationLevel == SqlIndentationLevel.SingleLine)
{
if (c == 'n' || c == 's')
{
if (state != null)
{
state.SeparatorNeeded = true;
}
else
{
sb.Append(" ");
}
}
// when original indentation is used, don't use our separators
break;
}
bool valid = (props.IndentationLevel == SqlIndentationLevel.Compact && (level == '2' || level == '5')) ||
(props.IndentationLevel == SqlIndentationLevel.Large && (level == '3' || level == '5'));
if (!valid)
{
break; // mark is not for this indentation level
}
}
switch (c)
{
case '&':
sb.Append("&");
break;
case 'n':
if (state == null)
{
DumpEoln(sb, props, state);
}
else
{
state.LineFeedNeeded = true;
}
break;
case '>':
if (state != null)
{
state.IndentLevel++;
}
break;
case '<':
if (state != null)
{
state.IndentLevel--;
}
break;
case 's':
if (state != null)
{
state.SeparatorNeeded = true;
}
else
{
sb.Append(" ");
}
break;
default:
throw new InternalError("DAE-00041 Unknown & formatting instruction:" + c);
}
}
break;
case '%': // format parameter
{
i++;
c = format[i];
if (c == '%')
{
sb.Append('%');
i++;
}
else if (c == ',') // comma separated list
{
i++;
c = format[i];
bool ok = false;
if (args[argindex] is IEnumerable)
{
ok = true;
}
if (args[argindex] is IBedRecord && c == 'v')
{
ok = true;
}
if (!ok)
{
throw new InternalError("DAE-00042 List must be of type Enumerable");
}
bool was = false;
if (args[argindex] is IEnumerable)
{
foreach (object item in (IEnumerable)args[argindex])
{
if (was)
{
sb.Append(", ");
}
WriteFormattedValue(dialect, props, sb, item, c, null, state, dda);
was = true;
}
}
else
{
var rec = (IBedRecord)args[argindex];
for (int x = 0; x < rec.FieldCount; x++)
{
if (was)
{
sb.Append(", ");
}
rec.ReadValue(x);
sb.Append(GetSqlLiteral(props, dda, state, rec));
was = true;
}
}
argindex++;
i++;
}
else if (c == ':')
{
object orig = args[argindex];
argindex++;
i++;
c = format[i];
object arg = args[argindex];
argindex++;
i++;
WriteFormattedValue(dialect, props, sb, arg, c, orig, state, dda);
}
else
{
WriteFormattedValue(dialect, props, sb, args[argindex], c, null, state, dda);
argindex++;
i++;
}
}
break;
default:
{
if (Char.IsWhiteSpace(c))
{
if (state != null)
{
state.SeparatorNeeded = false;
}
}
else
{
DumpSeparatorIfNeeded(sb, props, state, null);
}
sb.Append(c);
i++;
}
break;
}
}
return(sb.ToString());
}
public static Job CreateDataJob(IDataSqlGenerator generator, IFilePlace place, ISqlDialect dialect, TableDataFrame dataFrame, DataFrameRowsExtractor rows, ConnectionPack connpack, SqlFormatProperties formatProps, JobProperties jobProps)
{
return(Job.FromCommand(new GenerateDataSqlJobCommand(generator, place, dialect, dataFrame, rows, connpack, formatProps), jobProps));
}
public GenerateSqlJobCommand()
{
Args = new List <AppObject>();
FormatProps = new SqlFormatProperties();
}
public static string GetSqlLiteralAndRead(SqlFormatProperties props, IDialectDataAdapter dda, SqlFormatterState state, object val)
{
return(GetSqlLiteralAndRead(props, dda, state, val, null));
}
public static Job CreateJob(IAppObjectSqlGenerator generator, IFilePlace place, ISqlDialect dialect, AppObject[] objs, ConnectionPack connpack, SqlFormatProperties formatProps, JobProperties jobProps)
{
return(Job.FromCommand(new GenerateSqlJobCommand(generator, place, dialect, objs, connpack, formatProps), jobProps));
}
private void RefreshPreview()
{
if (!m_initialized)
{
return;
}
var dial = cbxDialect.SelectedItem as ISqlDialect;
if (dial == null && m_db != null)
{
dial = m_db.Dialect;
}
if (dial == null && m_dataFrame != null)
{
dial = m_dataFrame.GetDialect();
}
if (dial == null && m_objs != null)
{
dial = m_objs[0].Dialect;
}
if (dial == null)
{
dial = GenericDialect.Instance;
}
m_previewRows = tbxRows.Text.SafeIntParse(10);
m_dialect = dial;
m_sqlgen = lbxGenerator.SelectedItem as ISqlGeneratorCommon;
m_rowsExtractor = (DataFrameRowsExtractor)lbxRows.SelectedItem;
m_formatProps = sqlFormatPropsFrame1.Value;
if (m_curRun != null)
{
return;
}
if (m_dialect != null && m_sqlgen != null)
{
pictureBox1.BringToFront();
if (m_objs != null)
{
if (m_db != null)
{
m_db.Connection.BeginInvoke((Action)DoRefreshPreview, m_invoker.CreateInvokeCallback(RefreshedPreview));
}
else
{
m_curRun = DoRefreshPreview;
m_curRun.BeginInvoke(m_invoker.CreateInvokeCallback(RefreshedPreview), null);
//DoRefreshPreview();
//RefreshedPreview(null);
}
}
if (m_dataFrame != null)
{
if (m_dataFrame.TabularData.Connection != null)
{
m_dataFrame.TabularData.Connection.BeginInvoke((Action)DoRefreshPreview, m_invoker.CreateInvokeCallback(RefreshedPreview));
}
else
{
ThreadPool.BeginInvoke(DoRefreshPreviewFunc, m_invoker.CreateInvokeCallback(RefreshedPreview));
}
}
}
}
public virtual ISqlDumper CreateDumper(ISqlOutputStream stream, SqlFormatProperties props)
{
return(new SqlDumper(stream, this, props));
}
public static string GetSqlLiteralAndRead(SqlFormatProperties props, IDialectDataAdapter dda, SqlFormatterState state, object val, DbTypeBase dsttype)
{
state._Holder.ReadFrom(val);
return(GetSqlLiteral(props, dda, state, state._Holder, dsttype));
}
public static string GenerateScript(this ISqlDialect dialect, Action <ISqlDumper> script, SqlFormatProperties props)
{
StringWriter sw = new StringWriter();
ISqlDumper dmp;
if (dialect != null)
{
dmp = dialect.CreateDumper(sw, props);
}
else
{
dialect = new GenericDialect();
dmp = new InfoSqlDumper(new SqlOutputStream(dialect, sw, props), dialect, props);
}
script(dmp);
return(sw.ToString());
}
public InfoSqlDumper(ISqlOutputStream fw, ISqlDialect dialect, SqlFormatProperties props)
: base(fw, dialect, props)
{
}
public static string GetSqlLiteral(SqlFormatProperties props, IDialectDataAdapter dda, SqlFormatterState state, IBedValueReader reader)
{
return(GetSqlLiteral(props, dda, state, reader, null));
}
public virtual ISqlDumper CreateDumper(ISqlOutputStream stream, SqlFormatProperties props)
{
return(m_dialect.CreateDumper(stream, props));
}
protected virtual void GetFormatProps(SqlFormatProperties formatProps)
{
}
public SqlOutputStream(ISqlDialect dialect, TextWriter tw, SqlFormatProperties props)
{
m_dialect = dialect;
m_props = props;
m_tw = tw;
}