Exemplo n.º 1
0
        public void Initialize(GetCompletionResponse response)
        {
            _response = response;
            LoadItems();
            _textBox.KeyboardHandler = this;

            if (response.Items.Count > 0)
            {
                _prefix = _textBox.Text.Substring(response.StartPosition, response.Length);
                if (_prefix.Length > 0)
                {
                    var items = _prefix.Split('.');

                    var count         = response.Items[0].UnquotedName.Count(c => c == '.') + 1;
                    var stringBuilder = new StringBuilder();
                    for (var i = Math.Max(items.Length - count, 0); i < items.Length; i++)
                    {
                        if (stringBuilder.Length > 0)
                        {
                            stringBuilder.Append('.');
                        }
                        stringBuilder.Append(items[i]);
                    }

                    _prefix = stringBuilder.ToString().ToLower();
                    Find(_prefix, 0);
                }
            }
        }
Exemplo n.º 2
0
        GetCompletionResponse IProvider.GetCompletion(ConnectionBase connection, IDbTransaction transaction, string text, int position)
        {
            var response = new GetCompletionResponse();

            //      string[] array = null;
            //
            //      switch (word)
            //      {
            //        case "from":
            //          ApplicationData appData = Application.Instance.ApplicationData;
            //          Folder folder = appData.CurrentType;
            //          const string key = "ClassNames";
            //
            //          if (folder.Properties.ContainsKey(key))
            //          {
            //            array = (string[])folder.Properties[key];
            //          }
            //          else
            //          {
            //            WmiConnection wmiConnection = (WmiConnection)connection.Wrapped;
            //            ManagementClass manClass = new ManagementClass(wmiConnection.Scope.Path);
            //            ArrayList list = new ArrayList();
            //            AddClassNames(manClass,list);
            //
            //            array = new string[list.Count];
            //            list.CopyTo(array);
            //            Array.Sort(array);
            //
            //            folder.Properties.Add("ClassNames",array);
            //            appData.Save();
            //          }
            //          break;
            //
            //        default:
            //          break;
            //      }
            //
            //      return array;
            return(response);
        }
Exemplo n.º 3
0
        GetCompletionResponse IProvider.GetCompletion(ConnectionBase connection, IDbTransaction transaction, string text, int position)
        {
            var response = new GetCompletionResponse();

            string[] array        = null;
            var      sqlStatement = new SqlParser(text);
            var      tokens       = sqlStatement.Tokens;
            var      index        = sqlStatement.FindToken(position);

            if (index >= 0 && index < tokens.Count)
            {
                var token = sqlStatement.Tokens[index];
                var value = token.Value;
            }

            if (array == null)
            {
                var    sqlObject   = sqlStatement.FindSqlObject(index);
                string commandText = null;

                if (sqlObject != null)
                {
                    string name;
                    switch (sqlObject.Type)
                    {
                    case SqlObjectTypes.Table | SqlObjectTypes.View:
                        commandText = "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES ORDER BY TABLE_NAME";
                        break;

                    case SqlObjectTypes.Table | SqlObjectTypes.View | SqlObjectTypes.Function:
                        commandText = "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES ORDER BY TABLE_NAME";
                        break;

                    case SqlObjectTypes.Table:
                        commandText = "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES ORDER BY TABLE_NAME";
                        break;

                    case SqlObjectTypes.Column:
                        name        = sqlObject.ParentName;
                        commandText = $@"SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.COLUMNS
WHERE TABLE_NAME = '{name}'
ORDER BY ORDINAL_POSITION";
                        break;

                    case SqlObjectTypes.Value:
                        var    items            = sqlObject.ParentName.Split('.');
                        var    i                = items.Length - 1;
                        var    columnName       = items[i];
                        string tableNameOrAlias = null;
                        if (i > 0)
                        {
                            i--;
                            tableNameOrAlias = items[i];
                        }

                        if (tableNameOrAlias != null)
                        {
                            var contains = sqlStatement.Tables.TryGetValue(tableNameOrAlias, out var tableName);
                            if (contains)
                            {
                                commandText = $"select distinct top 10 {columnName} from {tableName} (nolock) order by 1";
                            }
                        }

                        break;
                    }
                }

                if (commandText != null)
                {
                    Trace.WriteLine(commandText);
                    var list = new List <string>();

                    try
                    {
                        var executor = connection.Connection.CreateCommandExecutor();
                        list = executor.ExecuteReader(new ExecuteReaderRequest(commandText), 128, dataRecord => dataRecord.GetString(0)).ToList();
                    }
                    catch
                    {
                    }

                    array = new string[list.Count];
                    list.CopyTo(array);
                }
            }

            //  TODO response.Items = array;
            return(response);
        }
Exemplo n.º 4
0
        GetCompletionResponse IProvider.GetCompletion(ConnectionBase connection, IDbTransaction transaction, string text, int position)
        {
            var response = new GetCompletionResponse
            {
                FromCache = false
            };
            List <IObjectName> array = null;
            var sqlStatement         = new SqlParser(text);
            var tokens = sqlStatement.Tokens;

            sqlStatement.FindToken(position, out var previousToken, out var currentToken);

            if (currentToken != null)
            {
                var parts    = new IdentifierParser(new StringReader(currentToken.Value)).Parse().ToList();
                var lastPart = parts.Count > 0
                    ? parts.Last()
                    : null;
                var lastPartLength = lastPart != null
                    ? lastPart.Length
                    : 0;
                response.StartPosition = currentToken.EndPosition - lastPartLength + 1;
                response.Length        = lastPartLength;
                var value = currentToken.Value;
                if (value.Length > 0 && value[0] == '@')
                {
                    if (value.IndexOf("@@") == 0)
                    {
                        array = _keyWords.Where(k => k.StartsWith(value)).Select(keyWord => (IObjectName) new NonSqlObjectName(keyWord)).ToList();
                    }
                    else
                    {
                        var list = new SortedList <string, object>();

                        for (var i = 0; i < tokens.Count; i++)
                        {
                            var token   = tokens[i];
                            var keyWord = token.Value;

                            if (keyWord != null && keyWord.Length >= 2 && keyWord.IndexOf(value) == 0 && keyWord != value)
                            {
                                if (!list.ContainsKey(token.Value))
                                {
                                    list.Add(token.Value, null);
                                }
                            }
                        }

                        array = list.Keys.Select(keyWord => (IObjectName) new NonSqlObjectName(keyWord)).ToList();
                    }
                }
            }
            else
            {
                response.StartPosition = position;
                response.Length        = 0;
            }

            if (array == null)
            {
                var    sqlObject   = sqlStatement.FindSqlObject(previousToken, currentToken);
                string commandText = null;

                if (sqlObject != null)
                {
                    DatabaseObjectMultipartName name;
                    int i;

                    switch (sqlObject.Type)
                    {
                    case SqlObjectTypes.Database:
                        commandText = SqlServerObject.GetDatabases();
                        break;

                    case SqlObjectTypes.Table:
                    case SqlObjectTypes.View:
                    case SqlObjectTypes.Function:
                    case SqlObjectTypes.Table | SqlObjectTypes.View:
                    case SqlObjectTypes.Table | SqlObjectTypes.View | SqlObjectTypes.Function:
                    {
                        name = new DatabaseObjectMultipartName(connection.Database, sqlObject.Name);
                        var nameParts = sqlObject.Name != null
                                ? new IdentifierParser(new StringReader(sqlObject.Name)).Parse().ToList()
                                : null;
                        var namePartsCount = nameParts != null
                                ? nameParts.Count
                                : 0;
                        var statements = new List <string>();

                        switch (namePartsCount)
                        {
                        case 0:
                        case 1:
                        {
                            statements.Add(SqlServerObject.GetDatabases());
                            statements.Add(SqlServerObject.GetSchemas());

                            var objectTypes = sqlObject.Type.ToObjectTypes();
                            statements.Add(SqlServerObject.GetObjects("dbo", objectTypes));
                        }
                        break;

                        case 2:
                            if (nameParts[0] != null)
                            {
                                statements.Add(SqlServerObject.GetSchemas(nameParts[0]));

                                var objectTypes = sqlObject.Type.ToObjectTypes();
                                statements.Add(SqlServerObject.GetObjects(nameParts[0], objectTypes));
                            }

                            break;

                        case 3:
                        {
                            if (nameParts[0] != null && nameParts[1] != null)
                            {
                                var objectTypes = sqlObject.Type.ToObjectTypes();
                                statements.Add(SqlServerObject.GetObjects(nameParts[0], nameParts[1], objectTypes));
                            }
                        }
                        break;
                        }

                        commandText = statements.Count > 0
                                ? string.Join("\r\n", statements)
                                : null;
                    }
                    break;

                    case SqlObjectTypes.Column:
                        name = new DatabaseObjectMultipartName(connection.Database, sqlObject.ParentName);
                        string[] owners;

                        if (name.Schema != null)
                        {
                            owners = new[] { name.Schema }
                        }
                        ;
                        else
                        {
                            owners = new[] { "dbo", "sys" }
                        };

                        var sb = new StringBuilder();
                        for (i = 0; i < owners.Length; i++)
                        {
                            if (i > 0)
                            {
                                sb.Append(',');
                            }

                            sb.AppendFormat("'{0}'", owners[i]);
                        }

                        var ownersString = sb.ToString();
                        commandText = string.Format(@"declare @schema_id int
select  top 1 @schema_id = s.schema_id
from    [{0}].sys.schemas s
where   s.name  in({1})

if @schema_id is not null
begin
    declare @object_id int
    select  @object_id = o.object_id
    from    [{0}].sys.all_objects o
    where   o.name = '{2}'
            and o.schema_id = @schema_id
            and o.type in('S','U','TF','V')

    if @object_id is not null
    begin
        select  name
        from [{0}].sys.all_columns c
        where c.object_id = @object_id
        order by column_id
    end
end", name.Database, ownersString, name.Name);
                        break;

                    case SqlObjectTypes.Procedure:
                        name = new DatabaseObjectMultipartName(connection.Database, sqlObject.Name);

                        if (name.Schema == null)
                        {
                            name.Schema = "dbo";
                        }

                        commandText = string.Format(@"select
     s.name
    ,o.name
from [{0}].sys.objects o
join [{0}].sys.schemas s
on o.schema_id = s.schema_id
where   o.type in('P','X')
order by 1", name.Database);
                        break;

                    case SqlObjectTypes.Trigger:
                        commandText = "select name from sysobjects where xtype = 'TR' order by name";
                        break;

                    case SqlObjectTypes.Value:
                        var items = sqlObject.ParentName.Split('.');
                        i = items.Length - 1;
                        var    columnName       = items[i];
                        string tableNameOrAlias = null;
                        if (i > 0)
                        {
                            i--;
                            tableNameOrAlias = items[i];
                        }

                        if (tableNameOrAlias != null)
                        {
                            var contains = sqlStatement.Tables.TryGetValue(tableNameOrAlias, out var tableName);
                            if (contains)
                            {
                                string where;
                                var tokenIndex = previousToken.Index + 1;
                                if (tokenIndex < tokens.Count)
                                {
                                    var token      = tokens[tokenIndex];
                                    var tokenValue = token.Value;
                                    var indexofAny = tokenValue.IndexOfAny(new[] { '\r', '\n' });
                                    if (indexofAny >= 0)
                                    {
                                        tokenValue = tokenValue.Substring(0, indexofAny);
                                    }

                                    string like;
                                    if (tokenValue.Length > 0)
                                    {
                                        if (tokenValue.Contains('%'))
                                        {
                                            like = tokenValue;
                                        }
                                        else
                                        {
                                            like = tokenValue + '%';
                                        }
                                    }
                                    else
                                    {
                                        like = "%";
                                    }

                                    where = $"where {columnName} like N'{like}'";
                                }
                                else
                                {
                                    where = null;
                                }

                                commandText = $@"select distinct {columnName}
from
(
    select top 1000 {columnName}
    from {tableName} (readpast)
    {where}
) t";
                            }
                        }

                        break;
                    }
                }

                if (commandText != null)
                {
                    Log.Write(LogLevel.Trace, "commandText:\r\n{0}", commandText);
                    var list = new List <IObjectName>();
                    try
                    {
                        if (connection.State != ConnectionState.Open)
                        {
                            connection.OpenAsync(CancellationToken.None).Wait();
                        }

                        var executor = connection.Connection.CreateCommandExecutor();
                        executor.ExecuteReader(new ExecuteReaderRequest(commandText, null, transaction), dataReader =>
                        {
                            while (true)
                            {
                                var fieldCount = dataReader.FieldCount;
                                while (dataReader.Read())
                                {
                                    string schemaName;
                                    string objectName;

                                    if (fieldCount == 1)
                                    {
                                        schemaName = null;
                                        objectName = dataReader[0].ToString();
                                    }
                                    else
                                    {
                                        schemaName = dataReader.GetStringOrDefault(0);
                                        objectName = dataReader.GetString(1);
                                    }

                                    list.Add(new ObjectName(schemaName, objectName));
                                }

                                if (!dataReader.NextResult())
                                {
                                    break;
                                }
                            }
                        });
                    }
                    catch
                    {
                    }

                    array = list;
                }
            }

            response.Items = array;
            return(response);
        }
Exemplo n.º 5
0
        GetCompletionResponse IProvider.GetCompletion(ConnectionBase connection, IDbTransaction transaction, string text, int position)
        {
            var response = new GetCompletionResponse();

            string[] items        = null;
            var      sqlStatement = new SqlParser(text);

            sqlStatement.FindToken(position, out var previousToken, out var currentToken);

            if (currentToken != null)
            {
                response.StartPosition = currentToken.StartPosition;
                response.Length        = currentToken.EndPosition - currentToken.StartPosition + 1;
            }
            else
            {
                response.StartPosition = position;
                response.Length        = 0;
            }

            var    sqlObject   = sqlStatement.FindSqlObject(previousToken, currentToken);
            string commandText = null;
            var    cs          = new OracleConnectionStringBuilder(connection.ConnectionString);
            var    userId      = cs.UserID;

            if (sqlObject != null)
            {
                string[] parts;
                string   owner;

                switch (sqlObject.Type)
                {
                case SqlObjectTypes.Table:
                    var oracleName = new OracleName(userId, sqlObject.Name);
                    commandText          = $@"select	TABLE_NAME
from	SYS.ALL_TABLES
where	OWNER	= '{oracleName.Owner}'	
order by TABLE_NAME";
                    sqlObject.ParentName = oracleName.Owner;
                    break;

                case SqlObjectTypes.Table | SqlObjectTypes.View | SqlObjectTypes.Function:
                    var name = sqlObject.Name;

                    if (name != null)
                    {
                        parts = name.Split('.');

                        if (parts.Length > 1)
                        {
                            owner = parts[0].ToUpper();
                            sqlObject.ParentName = owner;
                            name = parts[1];
                        }
                        else
                        {
                            owner = userId;
                            sqlObject.ParentName = owner;
                        }
                    }
                    else
                    {
                        owner = userId;
                        sqlObject.ParentName = owner;
                    }

                    commandText =
                        $@"select	OBJECT_NAME
from	SYS.ALL_OBJECTS
where	OWNER	= '{owner}'
	and OBJECT_TYPE in('TABLE','VIEW')
order by OBJECT_NAME";
                    sqlObject.Name = null;

                    break;

                case SqlObjectTypes.Column:
                    var parentName = sqlObject.ParentName;

                    if (parentName != null)
                    {
                        parts = parentName.Split('.');
                        string tableName;

                        if (parts.Length == 2)
                        {
                            owner     = parts[0].ToUpper();
                            tableName = parts[1].ToUpper();
                        }
                        else
                        {
                            owner                = userId;
                            tableName            = sqlObject.ParentName.ToUpper();
                            sqlObject.ParentName = owner + '.' + tableName;
                        }

                        commandText =
                            $@"select	COLUMN_NAME
from	SYS.ALL_TAB_COLUMNS
where	OWNER = '{owner}'
	and TABLE_NAME = '{tableName}'
order by COLUMN_ID";
                    }

                    break;

                case SqlObjectTypes.Function:
                    oracleName  = new OracleName(userId, sqlObject.ParentName);
                    commandText =
                        $@"select	OBJECT_NAME
from	SYS.ALL_OBJECTS
where	OWNER	= '{oracleName.Owner
                                }'
	and OBJECT_TYPE	= 'FUNCTION'
order by OBJECT_NAME";
                    sqlObject.ParentName = oracleName.Owner;
                    break;

                //case SqlObjectTypes.Procedure:
                //    oracleName = new OracleName( userId, sqlObject.Name );
                //    break;

                default:
                    break;
                }
            }

            if (commandText != null)
            {
                var sb = new StringBuilder();
                sb.Append(_objectExplorer.SchemasNode.Connection.DataSource);
                sb.Append('.');
                sb.Append(sqlObject.Type);
                sb.Append('.');
                var parentName = sqlObject.ParentName;

                if (parentName != null)
                {
                    sb.Append(parentName.ToUpper());
                }

                var name = sqlObject.Name;

                if (!string.IsNullOrEmpty(name))
                {
                    sb.Append('.');
                    sb.Append(name.ToUpper());
                }

                var key             = sb.ToString();
                var applicationData = DataCommanderApplication.Instance.ApplicationData;
                var folderName      = ConfigurationNodeName.FromType(typeof(OracleProvider)) + ConfigurationNode.Delimiter + "CompletionCache";
                var folder          = applicationData.CreateNode(folderName);
                var contains        = folder.Attributes.TryGetAttributeValue(key, out items);
                response.FromCache = contains;

                if (!contains)
                {
                    var executor = connection.Connection.CreateCommandExecutor();
                    var table    = executor.ExecuteDataTable(new ExecuteReaderRequest(commandText));
                    var count    = table.Rows.Count;
                    items = new string[count];

                    for (var i = 0; i < count; i++)
                    {
                        items[i] = (string)table.Rows[i][0];
                    }

                    folder.Attributes.Add(key, items, null);
                }
            }

            //			SqlStatement sqlStatement = new SqlStatement(text);
            //			SqlObject sqlObject = sqlStatement.FindSqlObject(position);
            //			string commandText = null;
            //
            //			switch (sqlObject.Type)
            //			{
            //				case SqlObjectType.TableOrView:
            //					string owner = connection.Database;
            //					commandText = string.Format("select * from (select table_name from all_tables where owner = '{0}' union select view_name from all_views where owner = '{0}') order by 1",owner);
            //					break;
            //			}

            //      string commandText;
            //
            //      switch (word)
            //      {
            //        case "from":
            //          commandText = "select * from (select table_name from all_tables where owner = '{0}' union select view_name from all_views where owner = '{0}') order by 1";
            //          commandText = string.Format(commandText,objectBrowser.SchemasNode.SelectedSchema);
            //          break;
            //
            //        case "table":
            //        case "update":
            //          commandText = string.Format("select table_name from all_tables where owner='{0}'",connection.DataSource);
            //          break;
            //
            //        default:
            //          commandText = null;
            //          break;
            //      }
            //
            //      string[] items = null;
            //
            //      if (commandText != null)
            //      {
            //        string key = objectBrowser.SchemasNode.Connection.DataSource + "." +
            //          objectBrowser.SchemasNode.SelectedSchema + '.' + word;
            //
            //        Folder appData = Application.Instance.ApplicationData.CurrentType;
            //
            //        if (appData.SubFolders.Contains("CompletionCache"))
            //        {
            //          appData = (Folder)appData.SubFolders["CompletionCache"];
            //        }
            //        else
            //        {
            //          Folder folder = new Folder(appData,"CompletionCache");
            //        }
            //
            //        bool containsKey = appData.Properties.ContainsKey(key);
            //
            //        if (containsKey)
            //        {
            //          items = (string[])appData.Properties[key];
            //        }
            //        else
            //        {
            //            Cursor.Current = Cursors.WaitCursor;
            //            ArrayList list = new ArrayList();
            //            IDataReader dataReader = DataHelper.ExecuteReader(commandText,connection.Wrapped);
            //
            //            while (dataReader.Read())
            //              list.Add(dataReader.GetString(0));
            //
            //            items = new string[list.Count];
            //            list.CopyTo(items);
            //
            //            appData.Properties[key] = items;
            //
            //            Cursor.Current = Cursors.Default;
            //        }
            //      }
            //
            //      return items;

            throw new NotImplementedException();

            //response.Items = items;
            //return response;
        }
Exemplo n.º 6
0
        GetCompletionResponse IProvider.GetCompletion(ConnectionBase connection, IDbTransaction transaction, string text, int position)
        {
            var response = new GetCompletionResponse
            {
                FromCache = false
            };

            var sqlStatement = new SqlParser(text);
            var tokens       = sqlStatement.Tokens;

            sqlStatement.FindToken(position, out var previousToken, out var currentToken);

            if (currentToken != null)
            {
                var parts          = new IdentifierParser(new StringReader(currentToken.Value)).Parse();
                var lastPart       = parts.Last();
                var lastPartLength = lastPart != null ? lastPart.Length : 0;
                response.StartPosition = currentToken.EndPosition - lastPartLength + 1;
                response.Length        = lastPartLength;
            }
            else
            {
                response.StartPosition = position;
                response.Length        = 0;
            }

            var sqlObject = sqlStatement.FindSqlObject(previousToken, currentToken);

            if (sqlObject != null)
            {
                var statements = new List <string>();

                switch (sqlObject.Type)
                {
                case SqlObjectTypes.Database:
                    statements.Add(SqlServerObject.GetDatabases());
                    break;

                case SqlObjectTypes.Table | SqlObjectTypes.View | SqlObjectTypes.Function:
                {
                    var nameParts = new IdentifierParser(new StringReader(sqlObject.Name ?? string.Empty)).Parse().ToList();
                    var name      = new DatabaseObjectMultipartName(connection.Database, nameParts);

                    switch (nameParts.Count)
                    {
                    case 0:
                    case 1:
                        statements.Add(SqlServerObject.GetDatabases());
                        statements.Add(SqlServerObject.GetTables(name.Database, new[] { "BASE TABLE", "SYSTEM VIEW" }));
                        break;

                    case 2:
                        statements.Add(SqlServerObject.GetTables(name.Database, new[] { "BASE TABLE", "SYSTEM VIEW" }));
                        break;
                    }
                }
                break;

                case SqlObjectTypes.Column:
                {
                    var nameParts = new IdentifierParser(new StringReader(sqlObject.ParentName ?? string.Empty)).Parse().ToList();
                    var name      = new DatabaseObjectMultipartName(connection.Database, nameParts);
                    statements.Add(SqlServerObject.GetColumns(name.Database, name.Name));
                }
                break;
                }

                var objectNames = new List <IObjectName>();
                var executor    = DbCommandExecutorFactory.Create(connection.Connection);
                foreach (var statement in statements)
                {
                    var items = executor.ExecuteReader(new ExecuteReaderRequest(statement), 128, dataRecord => new ObjectName(null, dataRecord.GetString(0)));
                    objectNames.AddRange(items);
                }

                response.Items = objectNames;
            }

            return(response);
        }
Exemplo n.º 7
0
        GetCompletionResponse IProvider.GetCompletion(ConnectionBase connection, IDbTransaction transaction, string text, int position)
        {
            var response = new GetCompletionResponse();

            string[]     values       = null;
            SqlStatement sqlStatement = new SqlStatement(text);

            Token[] tokens = sqlStatement.Tokens;

            if (tokens.Length > 0)
            {
                Token previousToken, currentToken;
                sqlStatement.FindToken(position, out previousToken, out currentToken);
                if (currentToken != null)
                {
                    response.StartPosition = currentToken.StartPosition;
                    response.Length        = currentToken.EndPosition - currentToken.StartPosition + 1;
                }
                else
                {
                    response.StartPosition = position;
                    response.Length        = 0;
                }

                if (previousToken != null)
                {
                    Token token = previousToken;

                    if (token.Type == TokenType.KeyWord && string.Compare(token.Value, "exec", StringComparison.InvariantCultureIgnoreCase) == 0)
                    {
                        var names = new List <string>();

                        foreach (var name in TfsDataReaderFactory.Dictionary.Keys)
                        {
                            names.Add(name);
                        }

                        values = names.ToArray();
                    }
                    else if (tokens.Length > 1)
                    {
                        token = tokens[0];

                        if (token.Type == TokenType.KeyWord && string.Compare(token.Value, "exec", StringComparison.InvariantCultureIgnoreCase) == 0)
                        {
                            //token = tokens[1];
                            //TfsDataReaderFactory.DataReaderInfo info;
                            //bool contains = TfsDataReaderFactory.Dictionary.TryGetValue(token.Value, out info);

                            //if (contains)
                            //{
                            //    TfsParameterCollection parameters = info.Parameters;
                            //    TfsParameter parameter = parameters[index - 2];
                            //    Type type = parameter.Type;

                            //    if (type.IsEnum)
                            //    {
                            //        string[] names = Enum.GetNames(type);
                            //        values = names;
                            //    }
                            //}
                            var command = sqlStatement.CreateCommand(this, connection, CommandType.StoredProcedure, 0);
                            TfsDataReaderFactory.DataReaderInfo info;
                            var contains = TfsDataReaderFactory.Dictionary.TryGetValue(command.CommandText, out info);
                            if (contains)
                            {
                                var parameters     = info.Parameters;
                                var parameterIndex = previousToken.Index / 2;
                                if (parameterIndex < parameters.Count)
                                {
                                    var parameter = parameters[parameterIndex];
                                    var type      = parameter.Type;
                                    if (type.IsEnum)
                                    {
                                        var names = Enum.GetNames(type);
                                        values = names;
                                    }
                                }
                            }
                        }
                    }
                }
            }

            response.Items = values.Select(value => (IObjectName) new ObjectName(value)).ToList();
            return(response);
        }
Exemplo n.º 8
0
        GetCompletionResponse IProvider.GetCompletion(ConnectionBase connection, IDbTransaction transaction, string text, int position)
        {
            var response     = new GetCompletionResponse();
            var sqlStatement = new SqlParser(text);
            var tokens       = sqlStatement.Tokens;

            sqlStatement.FindToken(position, out var previousToken, out var currentToken);

            if (currentToken != null)
            {
                response.StartPosition = currentToken.StartPosition;
                response.Length        = currentToken.EndPosition - currentToken.StartPosition + 1;
                var value = currentToken.Value;
            }
            else
            {
                response.StartPosition = position;
                response.Length        = 0;
            }

            var sqlObject = sqlStatement.FindSqlObject(previousToken, currentToken);

            if (sqlObject != null)
            {
                string commandText = null;
                response.FromCache = false;

                switch (sqlObject.Type)
                {
                case SqlObjectTypes.Table:
                    commandText = @"
select	name
from	sqlite_master
where   type    = 'table'
order by name collate nocase";
                    break;

                case SqlObjectTypes.Table | SqlObjectTypes.View | SqlObjectTypes.Function:
                    commandText = @"
select	name
from	sqlite_master
where   type    = 'table'
order by name collate nocase";
                    break;

                case SqlObjectTypes.Index:
                    commandText = @"
select	name
from	sqlite_master
where   type    = 'index'
order by name collate nocase";
                    break;

                case SqlObjectTypes.Column:
                    commandText = $"PRAGMA table_info({sqlObject.ParentName});";
                    break;
                }

                if (commandText != null)
                {
                    var executor = DbCommandExecutorFactory.Create(connection.Connection);
                    response.Items = executor.ExecuteReader(new ExecuteReaderRequest(commandText), 128, dataRecord =>
                    {
                        var name = dataRecord.GetStringOrDefault(0);
                        return((IObjectName) new ObjectName(name));
                    }).ToList();
                }
            }

            return(response);
        }