Dh dZddlZddlZddlmZddlZddlmZddlm Z ej e Z ej dZej ddZej d d Zd ed eZgZeseder+e d ee ddZdZdZejerejern e deeedZe den]#e$r>Ze deee dYdZ[ndZ[wwxYwe deeedZdeeeffdZeddfdede de defdZ!dededefd Z"ej#eddfdede de defd!Z$ej#efdededefd"Z%d#Z&d$Z'dS)%z Flava Function MCP Tools for CQA Test Application FastMCP-based Flava Function integration for FaaS management and operations N)Dict)FastMCP) AthenzClientFLAVA_FUNCTION_BASE_URLFLAVA_ENVIRONMENTstage FLAVA_PROJECTzflava-qaz flava-faas..z*Missing required environment variables: %szFFlava Function tools may not function properly without these variablesz/var/run/athenz/service.key.pemz /var/run/athenz/service.cert.pemF) account_nameprovider_domainkey_path cert_path use_proxyu-✅ Athenz client initialized with domain: %su.⚠️ Failed to initialize Athenz client: %sz.Authentication will fall back to basic headersuJℹ️ Athenz certificates not found (%s, %s), using basic authenticationzCQA Flava Function MCP Serverreturncddi}tr t}|r#d||d<tdntdnY#t $r2}tdt |Yd}~n"d}~wwxYwtd |S) z2Get authentication headers for Flava Function API.z Content-Typezapplication/jsonzBearer Authorizationu(✅ Added Athenz access token to headersu3⚠️ No access token received from Athenz clientu&⚠️ Failed to get access token: %sNu7ℹ️ No Athenz client available, using basic headers) ATHENZ_CLIENTget_access_tokenloggerdebugwarning Exceptionstr)headers access_tokenauth_exceptions K/Users/user/workspace/sujinbaek/cqa-test-app/products/mcp/flava_function.pyget_auth_headersrJs12G P Z(99;;L V+C\+C+C( GHHHHTUUU Z Z Z NNCSEXEX Y Y Y Y Y Y Y Y Z  NOOO NsAA&& B"0(BB" projectlimitoffsetc0 tsddiSt}td|d}d|i}|dkr||d<td|td |t jd 5}||||d }td|j|jdkrM|}d ||t|trt|nd||dcdddSd|j|j|j dcdddS#1swxYwYdS#t$rF} tdt| ddt| icYd} ~ Sd} ~ wwxYw)zGet functions from Flava API.error&FLAVA_FUNCTION_BASE_URL not configured /projects/z /functionsr!rr"u🌐 Making API request to: %su📋 Request params: %sT)follow_redirects>@)rparamstimeoutu📨 Response status: %s)successr functionscountr!r"NAPI request failed: r$ status_coderesponsezError in get_functions: %sError: )BASE_URLrrinfohttpxClientgetr2json isinstancelistlentextrr$r) r r!r"rurlr)clientr3data api_exceptions r get_functionsrC_s:'9 GEF F"$$88W8885! A::%F8  4c::: -v666 \4 0 0 0 FzzWVT"H KK2H4H I I I#s**}}#&!%*4T4*@*@GSYYYa"$         &G0DFF'3$M%                  . 999 13}3E3EFFF73}#5#57788888889sT EA5EBD8 ED8+ E8D<<E?D<E F;F FF function_namec tsddiSt}td|d|}tj5}|||d}|jdkr'|}d|||d cd d d Sd |j|j|jd cd d d S#1swxYwYd S#t$rF}t d t|ddt|icYd }~Sd }~wwxYw)z$Get function details from Flava API.r$r%r&z /functions/r()rr*r+T)r-r rDdetailsNr0r1z!Error in get_function_details: %sr4) r5rr7r8r9r2r:r>rrr$r)r rDrr?r@r3rArBs rget_function_detailsrGs9 GEF F"$$HHWHHHH \^^ vzz#wzEEH#s**}}#&%2#          G0DFF'3$M                  " 999 8#m:L:LMMM73}#5#57788888889sQ B90B9>B,; B9B, B9,B00B93B04B99 D ;D>D D cntd||dkrddiSt|||S)aGet list of functions from Flava API. Args: project: Flava project name (default: from FLAVA_PROJECT env var) limit: Maximum number of results to return (default: 10) offset: Number of results to skip for pagination (default: 0) z-Executing flava_get_functions for project: %srr$zOffset must be 0 or greater)rr6rC)r r!r"s rflava_get_functionsrIs@ KK?III zz677 % 0 00cftd|||sddiSt||S)zGet detailed information for a specific function. Args: function_name: The name of the function to get details for project: Flava project name (default: from FLAVA_PROJECT env var) zDExecuting flava_get_function_details for function: %s in project: %sr$zfunction_name is required)rr6rG)rDr s rflava_get_function_detailsrLsI KKN 6455  7 77rJctS)z5Get the configured Flava Function MCP server instance)faas_mcprJrget_flava_function_mcp_serverrPs OrJc bddttttt t dS)z)Get Flava Function MCP server informationzcqa-flava-function-mcpz1.0.0) server_nameserver_versionbase_urlflava_environment flava_projectr athenz_configured)r5rr PROVIDER_DOMAINboolrrOrJrget_flava_function_mcp_inforZs00!.&*!-00  rJ)(__doc__osloggingtypingrr7fastmcprlib.athenz_clientr getLogger__name__rgetenvr5rr rX required_varsappendrKEY_PATH CERT_PATHrpathexistsr6rathenz_exceptionrrNrintrCrGtoolrIrLrPrZrOrJrrms   ******  8 $ $ 29. / /BI17;; /:66 D 1CCMCC 42333 NN?OOO NNP - .  7>>(y 9 9I$ +      ;      III S>S    GHHHHHHHH I  KKT 72 3 3$sCx.,!r+9+9 +9),+9;>+9 +9+9+9+9\9#9c9d9999@  r11 1),1;>1 1111$ '4888!$8 8888*     s,&DE4EE