z(dZddlZddlZddlmZddlmZddlmZejdGdd Z d d Z d e fd Z y)z:Utility for validating Firestore Mongo connection strings.N)List)log) console_attrT)frozencheZdZUdZeeed<eeed<eeed<eeed<eeed<dZy) ValidationResultsz>Container class for results of validating a connection string.headersinfowarningserrorsfooterscdj|j|jz|jz|jz|j zS)N )joinr r r r r )selfs ;lib/googlecloudsdk/command_lib/firestore/connection_util.py__str__zValidationResults.__str__#s> 99 tyy 4==04;;>M N)__name__ __module__ __qualname____doc__rstr__annotations__rrrrrs;F 9 S / I s) 9rrc x dd|dg}gg}gdg}d}d}dIfd } |jdsjdn=|tdd}tjd|} | rb| j d}d t| j d z}jd |d ||t|t|zd zd}tjd |} | s*jd|d|jdns| j d|k7r+jd| j dd|d|dnjd|d|dd}tjd|} | sjdn| j d|k7r+jd| j dd|d|dnjd|d|t| j ddzd}|jdsjdnW|tdd}tjd|} | sjd|d |d!n| j d|k7rR| j dr(jd"| j dd#|dn.jd$|d%|d!njd"|d|t| j ddzd}i|j d&} | D]E} | s| j d'} t| d k7rjd(| d);| d| d*<G| d+d,| d-d,| d.d/d0vrd0}d0=|d1k(r1| d2d3d45|sjd6njd7n|d8k(r&|sjd9nxjd:nf|d;k(r|rjd<| d=d>d?5nBjd@|dn,|rjdA|d |dBnjdCjD]\}}|jdD|d'|d s|jdEn|jdF|dGt|||HS)JzDValidate the specified connection_string for the specified database.zP--------------------------------------------------------------------------------zEvaluating connection string: Nc |vrj|d|d|dy|}|=||k7r!j|d|d|d|d|d yj|d|dy)Nz#The connection string must specify =.The parameter z is set to z%. The connection string must specify )append) param_nameexpected_valueerror_description actual_valuer extra_paramsr s r CheckParamz,ValidateConnectionString..CheckParam>s% mm   \>*! - "*-l z "  '  ! |;nAl!N+1 . zl!N#3156rz mongodb://z0The connection string must start with mongodb://z^([^:]*):([^@]*)@*z&The connection string specifies user: z and password: zA^([0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12})\.z@The database address must start with a valid UUID. For database z, use the value ztNOTE: for password based authentication, the connection string can also start with mongodb://username:password@UUID.z the UUID z2 in the connection string does not match the UUID z for the current database rz The UUID z is correct.%z ^([^\.]+)\.zIThe database address must have the form: UUID.location.firestore.goog:443z The location z6 in the connection string does not match the location zfirestore.goog:443/zRThe database address must end with firestore.goog:443 as the domain name and port.z ^([^\?]*)\?zMThe connection string must specify the database id. For the current database z: it should have the form UUID.location.firestore.goog:443/?zThe database name z> in the connection string does not match the current database zoThe database name in the connection string is empty. It is recommended to explicitly specify the database name z, e.g. firestore.goog:443/&rr z= appears malformed. Expected something in the form key=value.r loadBalancedtruetls retryWritesfalse authMechanismPLAIN authSourcez $externalzUsing PLAIN authentication. )r$zsThe username and an access token should be specified in the connection string when PLAIN authentication is enabled.z=username and access token specified for PLAIN authentication.z SCRAM-SHA-256zeThe username and password should be specified in the connection string when SCRAM-SHA-256 is enabled.z2username and password specified for SCRAM-SHA-256.z MONGODB-OIDCzZThe username should not be specified when using the MONGODB-OIDC authentication mechanism.authMechanismPropertiesz(ENVIRONMENT:gcp,TOKEN_RESOURCE:FIRESTOREz#Using MONGODB-OIDC authentication. z%Unsupported authentication mechanism z,Since the connection string specified user: zw, the connection must also be configured with an appropriate authentication mechanism, e.g. authMechanism=SCRAM-SHA-256znNo authMechanism specified. The connection string must specify one of the supported authentication mechanisms.zUnknown parameter z4Did not detect any errors in this connection string.zJTIP: You can use 'gcloud firestore databases connection-string --database=z(' to construct valid connection strings.)r r r r r )) startswithr!lenrematchgroupsplititemsr)connection_stringdb_uiddb_location_id database_idr r r userpasswordr'r;entriesentrypartsauth_mechanismkvr r&r s @@@rValidateConnectionStringrK)s&'8&9: ' $ ( & J' $ (7&  ' ' 5 mmFG )#l*;*=> HH)+< =E [[^ds5;;q>**h kk 24&9#* & ,CIH ,E,I,KL HHL E  mm %&6vh @  mm B  {{1~ mm ekk!n%&"8#=k]! M  kkIfX\23)"#. HH E  mm -  {{1~' mm %++a.)*$$2#34)]! -  kkM.!1>?)#ekk!n*=*A*CD  ' '(= > mm & )#.C*D*FG HH^%6 7E  mm &&1]3..9]! =   {{1~$ Q  Q 01;;F-q J   Im5k]! E  kk&{m<@A)#ekk!n*=*A*CDL%%c*G kk#e Uq UG$0 0 "'q U1X~v&uf}g&,&#O4n  ' 7 "  <  --L ++M  _ , --A ++J K ^ +  --7   % 6C  =n=MQOP  :4&A"$+ +    F ""$1oo*1#Qqc34%  NNIJ NN !]"J L     rvalidation_resultsctj}|jD]!}tjj |#|j D]6}tjj |jddd|8|jD]6}tjj |jddd|8|jD]6}tjj |jddd|8|jD]!}tjj |#y) z@Renders the connection string validation results to the console.zINFO:green zWARNING:yellowzERROR:redN) rGetConsoleAttrr rstatusPrintr Colorizer r r )rLconheaderr warningerrorfooters rPrettyPrintValidationResultsr[s ##%#"**fJJV+ %%dJJ Wg67q?@&#,,gJJ Z:;1WIFG-!((eJJ Xu56aw?@)"**fJJV+r)NNN) r dataclassesr:typingrgooglecloudsdk.corergooglecloudsdk.core.consoler dataclassrrKr[rrrras_A #4d#  $ "  m` 5F r