dZddlmZddlmZddlmZddlmZddlmZddlm Z dZ d Z d Z d Z d Zd ZdZdZy)zBHelper methods for configuring web security scanner command flags.)absolute_import)division)unicode_literals)wss_base)base) exceptionsctjd}|jt|jt |jt |jt |gS)zOHook to add additional authentication flags. Returns: Auth flag group F)mutex)r ArgumentGroup AddArgument _TypeFlag _UserFlag _PasswordFlag_UrlFlag) auth_groups ;lib/googlecloudsdk/command_lib/web_security_scanner/auth.py AuthFlagsrs\ !!.* % % ) $ c:tjddddddS)z(Returns a flag for setting an auth type. --auth-typezDisables Authentication.z%Authentication with a Google account.z%Authentication with a custom account.)nonegooglecustomzType of authentication to use.)choiceshelp_str)rChoiceArgumentrrr r (s+  ,;; 0 rc0tjddS)z(Returns a flag for setting an auth user.z --auth-userz The test user to log in as. Required if `--auth-type` is not 'none'. 'google' login requires a full email address. Cannot be your own account. helprArgumentrrrrr5s    rc0tjddS)z,Returns a flag for setting an auth password.z--auth-passwordzQ Password for the test user. Required if `--auth-type` is not 'none'. rr!rrrrr?s    rc0tjddS)z'Returns a flag for setting an auth url.z --auth-urlz URL of the login page for your site. Required if `--auth-type` is 'custom'. Otherwise, it should not be set. rr!rrrrrHs    rctj}|jDt|j|j |j grtjdd|S|jdk(rDt|j|j |j grtjdd|S|j|jj|_ |jj|j_|jdk(r[t|ddgd|jj!|j|j |jj_|S|jd k(rft|gd d |jj%|j|j |j |jj_|Stj(d|j) aModify request hook to set scan config details. Args: unused_ref: not used parameter to modify request hooks args: Parsed args namespace request: The partially filled request object. Returns: Request object with Authentication message filled out. rz+Required when setting authentication flags.rz4No other auth flags can be set with --auth-type=noner auth_user auth_password)usernamepasswordr)r&r'auth_url)r(r)loginUrl)rWebSecurityScannerCommand auth_typeanyr&r'r*rRequiredArgumentExceptionInvalidArgumentException scanConfigmessages ScanConfigAuthenticationauthentication"_RequireAllFlagsOrRaiseForAuthType GoogleAccount googleAccount CustomAccount customAccountUnknownArgumentException) unused_refargsrequestcs rSetScanConfigAuthr@Rs((*! ^^ DNND.. >?  0 0 F HH N ^^v DNND.. >?  / / O QQ N ..0G&'jj&?&?&A'# ^^x&tk?-K'/167jj6N6N$*<*<7O7>G%%3 . ~~!& 8(D67jj6N6N##7O7 G%%3 .  - -mT^^ LLrcv|j|S|jr|xjdz c_|Sd|_|S)z%Adds auth-specific fieldmask entries.z,authenticationr5)r- updateMask)r<r=r>s rAddAuthFieldMaskrCsB ^^ N  ++ .*G .rct|}|D]M}|| |jdd}tjdj |dj |y)N_-z--{0}zRequired by --auth-type={0})varsreplacerr/format)r=flagsr-argvarsflagdasheds rr6r6s\ J'dt}||C%f  0 0 .. ' . .y 9 ;;rN)__doc__ __future__rrr+googlecloudsdk.api_lib.web_security_scannerrgooglecloudsdk.callioperrrr rrrr@rCr6rrrrRsFI&'@(.     *Z ;r