Skip to content

Commit 6c6af75

Browse files
yoavcloudyoavcloud
committed
Add support for Snowflake account privileges
1 parent e9498d5 commit 6c6af75

File tree

5 files changed

+639
-71
lines changed

5 files changed

+639
-71
lines changed

‎src/ast/mod.rs

Lines changed: 254 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5453,29 +5453,107 @@ impl fmt::Display for FetchDirection {
54535453
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
54545454
#[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
54555455
pub enum Action {
5456+
AddSearchOptimization,
5457+
Apply {
5458+
apply_type: ActionApplyType,
5459+
},
5460+
ApplyBudget,
5461+
AttachListing,
5462+
AttachPolicy,
5463+
Audit,
5464+
BindServiceEndpoint,
54565465
Connect,
5457-
Create,
5466+
Create {
5467+
obj_type: Option<ActionCreateObjectType>,
5468+
},
54585469
Delete,
5459-
Execute,
5460-
Insert { columns: Option<Vec<Ident>> },
5461-
References { columns: Option<Vec<Ident>> },
5462-
Select { columns: Option<Vec<Ident>> },
5470+
EvolveSchema,
5471+
Execute {
5472+
obj_type: Option<ActionExecuteObjectType>,
5473+
},
5474+
Failover,
5475+
ImportedPrivileges,
5476+
ImportShare,
5477+
Insert {
5478+
columns: Option<Vec<Ident>>,
5479+
},
5480+
Manage {
5481+
manage_type: ActionManageType,
5482+
},
5483+
ManageReleases,
5484+
ManageVersions,
5485+
Modify {
5486+
modify_type: ActionModifyType,
5487+
},
5488+
Monitor {
5489+
monitor_type: ActionMonitorType,
5490+
},
5491+
Operate,
5492+
OverrideShareRestrictions,
5493+
Ownership,
5494+
PurchaseDataExchangeListing,
5495+
Read,
5496+
ReadSession,
5497+
References {
5498+
columns: Option<Vec<Ident>>,
5499+
},
5500+
Replicate,
5501+
ResolveAll,
5502+
Select {
5503+
columns: Option<Vec<Ident>>,
5504+
},
54635505
Temporary,
54645506
Trigger,
54655507
Truncate,
5466-
Update { columns: Option<Vec<Ident>> },
5508+
Update {
5509+
columns: Option<Vec<Ident>>,
5510+
},
54675511
Usage,
54685512
}
54695513

54705514
impl fmt::Display for Action {
54715515
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
54725516
match self {
5517+
Action::AddSearchOptimization => f.write_str("ADD SEARCH OPTIMIZATION")?,
5518+
Action::Apply { apply_type } => write!(f, "APPLY {apply_type}")?,
5519+
Action::ApplyBudget => f.write_str("APPLY BUDGET")?,
5520+
Action::AttachListing => f.write_str("ATTACH LISTING")?,
5521+
Action::AttachPolicy => f.write_str("ATTACH POLICY")?,
5522+
Action::Audit => f.write_str("AUDIT")?,
5523+
Action::BindServiceEndpoint => f.write_str("BIND SERVICE ENDPOINT")?,
54735524
Action::Connect => f.write_str("CONNECT")?,
5474-
Action::Create => f.write_str("CREATE")?,
5525+
Action::Create { obj_type } => {
5526+
f.write_str("CREATE")?;
5527+
if let Some(obj_type) = obj_type {
5528+
write!(f, " {obj_type}")?
5529+
}
5530+
}
54755531
Action::Delete => f.write_str("DELETE")?,
5476-
Action::Execute => f.write_str("EXECUTE")?,
5532+
Action::EvolveSchema => f.write_str("EVOLVE SCHEMA")?,
5533+
Action::Execute { obj_type } => {
5534+
f.write_str("EXECUTE")?;
5535+
if let Some(obj_type) = obj_type {
5536+
write!(f, " {obj_type}")?
5537+
}
5538+
}
5539+
Action::Failover => f.write_str("FAILOVER")?,
5540+
Action::ImportedPrivileges => f.write_str("IMPORTED PRIVILEGES")?,
5541+
Action::ImportShare => f.write_str("IMPORT SHARE")?,
54775542
Action::Insert { .. } => f.write_str("INSERT")?,
5543+
Action::Manage { manage_type } => write!(f, "MANAGE {manage_type}")?,
5544+
Action::ManageReleases => f.write_str("MANAGE RELEASES")?,
5545+
Action::ManageVersions => f.write_str("MANAGE VERSIONS")?,
5546+
Action::Modify { modify_type } => write!(f, "MODIFY {modify_type}")?,
5547+
Action::Monitor { monitor_type } => write!(f, "MONITOR {monitor_type}")?,
5548+
Action::Operate => f.write_str("OPERATE")?,
5549+
Action::OverrideShareRestrictions => f.write_str("OVERRIDE SHARE RESTRICTIONS")?,
5550+
Action::Ownership => f.write_str("OWNERSHIP")?,
5551+
Action::PurchaseDataExchangeListing => f.write_str("PURCHASE DATA EXCHANGE LISTING")?,
5552+
Action::Read => f.write_str("READ")?,
5553+
Action::ReadSession => f.write_str("READ SESSION")?,
54785554
Action::References { .. } => f.write_str("REFERENCES")?,
5555+
Action::Replicate => f.write_str("REPLICATE")?,
5556+
Action::ResolveAll => f.write_str("RESOLVE ALL")?,
54795557
Action::Select { .. } => f.write_str("SELECT")?,
54805558
Action::Temporary => f.write_str("TEMPORARY")?,
54815559
Action::Trigger => f.write_str("TRIGGER")?,
@@ -5498,6 +5576,174 @@ impl fmt::Display for Action {
54985576
}
54995577
}
55005578

5579+
#[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
5580+
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
5581+
#[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
5582+
pub enum ActionCreateObjectType {
5583+
Account,
5584+
Application,
5585+
ApplicationPackage,
5586+
ComputePool,
5587+
DataExchangeListing,
5588+
Database,
5589+
ExternalVolume,
5590+
FailoverGroup,
5591+
Integration,
5592+
NetworkPolicy,
5593+
OrganiationListing,
5594+
ReplicationGroup,
5595+
Role,
5596+
Share,
5597+
User,
5598+
Warehouse,
5599+
}
5600+
5601+
impl fmt::Display for ActionCreateObjectType {
5602+
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
5603+
match self {
5604+
ActionCreateObjectType::Account => write!(f, "ACCOUNT"),
5605+
ActionCreateObjectType::Application => write!(f, "APPLICATION"),
5606+
ActionCreateObjectType::ApplicationPackage => write!(f, "APPLICATION PACKAGE"),
5607+
ActionCreateObjectType::ComputePool => write!(f, "COMPUTE POOL"),
5608+
ActionCreateObjectType::DataExchangeListing => write!(f, "DATA EXCHANGE LISTING"),
5609+
ActionCreateObjectType::Database => write!(f, "DATABASE"),
5610+
ActionCreateObjectType::ExternalVolume => write!(f, "EXTERNAL VOLUME"),
5611+
ActionCreateObjectType::FailoverGroup => write!(f, "FAILOVER GROUP"),
5612+
ActionCreateObjectType::Integration => write!(f, "INTEGRATION"),
5613+
ActionCreateObjectType::NetworkPolicy => write!(f, "NETWORK POLICY"),
5614+
ActionCreateObjectType::OrganiationListing => write!(f, "ORGANIZATION LISTING"),
5615+
ActionCreateObjectType::ReplicationGroup => write!(f, "REPLICATION GROUP"),
5616+
ActionCreateObjectType::Role => write!(f, "ROLE"),
5617+
ActionCreateObjectType::Share => write!(f, "SHARE"),
5618+
ActionCreateObjectType::User => write!(f, "USER"),
5619+
ActionCreateObjectType::Warehouse => write!(f, "WAREHOUSE"),
5620+
}
5621+
}
5622+
}
5623+
5624+
#[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
5625+
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
5626+
#[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
5627+
pub enum ActionApplyType {
5628+
AggregationPolicy,
5629+
AuthenticationPolicy,
5630+
JoinPolicy,
5631+
MaskingPolicy,
5632+
PackagesPolicy,
5633+
PasswordPolicy,
5634+
ProjectionPolicy,
5635+
RowAccessPolicy,
5636+
SessionPolicy,
5637+
Tag,
5638+
}
5639+
5640+
impl fmt::Display for ActionApplyType {
5641+
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
5642+
match self {
5643+
ActionApplyType::AggregationPolicy => write!(f, "AGGREGATION POLICY"),
5644+
ActionApplyType::AuthenticationPolicy => write!(f, "AUTHENTICATION POLICY"),
5645+
ActionApplyType::JoinPolicy => write!(f, "JOIN POLICY"),
5646+
ActionApplyType::MaskingPolicy => write!(f, "MASKING POLICY"),
5647+
ActionApplyType::PackagesPolicy => write!(f, "PACKAGES POLICY"),
5648+
ActionApplyType::PasswordPolicy => write!(f, "PASSWORD POLICY"),
5649+
ActionApplyType::ProjectionPolicy => write!(f, "PROJECTION POLICY"),
5650+
ActionApplyType::RowAccessPolicy => write!(f, "ROW ACCESS POLICY"),
5651+
ActionApplyType::SessionPolicy => write!(f, "SESSION POLICY"),
5652+
ActionApplyType::Tag => write!(f, "TAG"),
5653+
}
5654+
}
5655+
}
5656+
5657+
#[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
5658+
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
5659+
#[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
5660+
pub enum ActionExecuteObjectType {
5661+
Alert,
5662+
DataMetricFunction,
5663+
ManagedAlert,
5664+
ManagedTask,
5665+
Task,
5666+
}
5667+
5668+
impl fmt::Display for ActionExecuteObjectType {
5669+
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
5670+
match self {
5671+
ActionExecuteObjectType::Alert => write!(f, "ALERT"),
5672+
ActionExecuteObjectType::DataMetricFunction => write!(f, "DATA METRIC FUNCTION"),
5673+
ActionExecuteObjectType::ManagedAlert => write!(f, "MANAGED ALERT"),
5674+
ActionExecuteObjectType::ManagedTask => write!(f, "MANAGED TASK"),
5675+
ActionExecuteObjectType::Task => write!(f, "TASK"),
5676+
}
5677+
}
5678+
}
5679+
5680+
#[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
5681+
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
5682+
#[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
5683+
pub enum ActionManageType {
5684+
AccountSupportCases,
5685+
EventSharing,
5686+
Grants,
5687+
ListingAutoFulfillment,
5688+
OrganizationSupportCases,
5689+
UserSupportCases,
5690+
Warehouses,
5691+
}
5692+
5693+
impl fmt::Display for ActionManageType {
5694+
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
5695+
match self {
5696+
ActionManageType::AccountSupportCases => write!(f, "ACCOUNT SUPPORT CASES"),
5697+
ActionManageType::EventSharing => write!(f, "EVENT SHARING"),
5698+
ActionManageType::Grants => write!(f, "GRANTS"),
5699+
ActionManageType::ListingAutoFulfillment => write!(f, "LISTING AUTO FULFILLMENT"),
5700+
ActionManageType::OrganizationSupportCases => write!(f, "ORGANIZATION SUPPORT CASES"),
5701+
ActionManageType::UserSupportCases => write!(f, "USER SUPPORT CASES"),
5702+
ActionManageType::Warehouses => write!(f, "WAREHOUSES"),
5703+
}
5704+
}
5705+
}
5706+
5707+
#[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
5708+
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
5709+
#[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
5710+
pub enum ActionModifyType {
5711+
LogLevel,
5712+
TraceLevel,
5713+
SessionLogLevel,
5714+
SessionTraceLevel,
5715+
}
5716+
5717+
impl fmt::Display for ActionModifyType {
5718+
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
5719+
match self {
5720+
ActionModifyType::LogLevel => write!(f, "LOG LEVEL"),
5721+
ActionModifyType::TraceLevel => write!(f, "TRACE LEVEL"),
5722+
ActionModifyType::SessionLogLevel => write!(f, "SESSION LOG LEVEL"),
5723+
ActionModifyType::SessionTraceLevel => write!(f, "SESSION TRACE LEVEL"),
5724+
}
5725+
}
5726+
}
5727+
5728+
#[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
5729+
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
5730+
#[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
5731+
pub enum ActionMonitorType {
5732+
Execution,
5733+
Security,
5734+
Usage,
5735+
}
5736+
5737+
impl fmt::Display for ActionMonitorType {
5738+
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
5739+
match self {
5740+
ActionMonitorType::Execution => write!(f, "EXECUTION"),
5741+
ActionMonitorType::Security => write!(f, "SECURITY"),
5742+
ActionMonitorType::Usage => write!(f, "USAGE"),
5743+
}
5744+
}
5745+
}
5746+
55015747
/// The principal that receives the privileges
55025748
#[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
55035749
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]

0 commit comments

Comments
 (0)