From b5ae7b76551989470f3f970c3923f717251454f0 Mon Sep 17 00:00:00 2001
From: rwelgosh <raymond.welgosh@gmail.com>
Date: Thu, 26 Dec 2024 18:41:38 -0500
Subject: [PATCH] Added autocomplete for ADD and DROP identity statements

---
 pylib/cqlshlib/cql3handling.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/pylib/cqlshlib/cql3handling.py b/pylib/cqlshlib/cql3handling.py
index 457582f3d3..bbe0a5242e 100644
--- a/pylib/cqlshlib/cql3handling.py
+++ b/pylib/cqlshlib/cql3handling.py
@@ -268,6 +268,7 @@ JUNK ::= /([ \t\r\f\v]+|(--|[/][/])[^\n\r]*([\n\r]|$)|[/][*].*?[*][/])/ ;
                   | <schemaChangeStatement>
                   | <authenticationStatement>
                   | <authorizationStatement>
+                  | <identityStatement>
                   ;
 
 <dataChangeStatement> ::= <insertStatement>
@@ -316,6 +317,10 @@ JUNK ::= /([ \t\r\f\v]+|(--|[/][/])[^\n\r]*([\n\r]|$)|[/][*].*?[*][/])/ ;
                            | <listPermissionsStatement>
                            ;
 
+<identityStatement> ::= <addIdentityStatement>
+                      | <dropIdentityStatement>
+                      ;
+
 # timestamp is included here, since it's also a keyword
 <simpleStorageType> ::= typename=( <identifier> | <stringLiteral> | "timestamp" ) ;
 
@@ -1623,6 +1628,14 @@ def username_name_completer(ctxt, cass):
     session = cass.session
     return [maybe_quote(list(row.values())[0].replace("'", "''")) for row in session.execute("LIST USERS")]
 
+syntax_rules += r'''
+<addIdentityStatement> ::= "ADD" "IDENTITY" ( "IF" "NOT" "EXISTS" )? <stringLiteral>
+                             "TO" "ROLE" <rolename>
+                        ;
+<dropIdentityStatement> ::= "DROP" "IDENTITY" ( "IF" "NOT" "EXISTS" )? <stringLiteral>
+                          ;
+'''
+
 
 @completer_for('rolename', 'role')
 def rolename_completer(ctxt, cass):