o iy@sddlZddlZddlZddlZddlZddlZddlmZddlZe e Z ddZ de de fddZd d d e d ede fd dZde de de de fddZdede fddZ       d=de de de dedBdedBdededBdedBdede dBfdd Ze d!kre ejeed"sGd#d$d$ZeZed%d&d'd(d)id*d+id,d-d.id/ed0d1d2d3d4d5id6d7d8ed9d&d:d,d;d<dSdS)>N)Anyc Csft}tj|d}z tj|ddW|Sty2}ztd|t ||WYd}~Sd}~ww)zZEnsures the API log directory exists within ComfyUI's temp directory and returns its path.api_logsTexist_okz'Error creating API log directory %s: %sN) folder_pathsget_temp_directoryospathjoinmakedirs Exceptionloggererrorstr) base_temp_dirlog_direrA/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/util/request_logger.pyget_log_directorysrnamereturncCs,|sdStdd|}|d}|sd}|S)Nlogz[^A-Za-z0-9._-]+_z ._)resubstrip)r sanitizedrrr_sanitize_filename_components r )lengthpartsr cGs"td|dd|S)N|utf-8)hashlibsha1r encode hexdigest)r r!rrr _short_hash%s"r(r operation_id request_urlc Cstjd}t|}t|pd|pd}d}|d}d|d}|s&d}td|t|d} td | t|t|} t|| krM|d | d }tj ||||S) zgBuild log filepath. We keep it well under common path length limits aiming for <= 240 characters total.z%Y%m%d_%H%M%S_%frz.logop<Nz ._-) datetimenowstrftimerr(maxlenrstriprr r ) rr)r* timestampslughmax_total_pathprefixsuffixmax_filename_len max_slug_lenrrr_build_log_filepath)s   r?datacCs~t|trz|dWStydt|dYSwt|ttfr;z tj|dddWSt y:t |YSwt |S)z5Helper to format data (dict, str, bytes) for logging.r#z[Binary data of length z bytes]F)indent ensure_ascii) isinstancebytesdecodeUnicodeDecodeErrorr5dictlistjsondumps TypeErrorr)r@rrr_format_data_for_logging=s     rMrequest_methodrequest_headersrequest_params request_dataresponse_status_coderesponse_headersresponse_content error_messagec  Cszt} t| ||} g} | dtj| d|| d| d|| d||rB| dt||rN| dt||dur\| d t|| d |durm| d ||ry| dt||dur| d t|| r| d | z)t| ddd} | d | Wdn1swYt d| WWdSt y}zt d| t|WYd}~WdSd}~wwt y}z t d|WYd}~dSd}~ww)z Logs API request and response details to a file in the temp/api_logs directory. Filenames are sanitized and length-limited for cross-platform safety. If we still fail to write, we fall back to appending into api.log. z Timestamp: zOperation ID: zE------------------------------ REQUEST ------------------------------zMethod: zURL: z Headers: zParams: Nz Data/Body: zG ------------------------------ RESPONSE ------------------------------z Status Code: z Content: zError: wr#)encoding zAPI log saved to: %szError writing API log to %s: %sz'[DEBUG] log_request_response failed: %s)rr?appendr1r2 isoformatrMopenwriter r debugr rrlogging)r)rNr*rOrPrQrRrSrTrUrfilepath log_contentfr_log_errrlog_request_responseLsJ   "rc__main__rc@seZdZddZdS)MockFolderPathscCs(tjtjtd}tj|dd|S)Ntemp_test_logsTr)rr r dirname__file__r )selfprrrrsz"MockFolderPaths.get_temp_directoryN)__name__ __module__ __qualname__rrrrrres retest_operation_getGETzhttps://api.example.com/test AuthorizationzBearer testtokenparam1value1messagezSuccess!)r)rNr*rOrPrRrTtest_operation_post_errorPOSTzhttps://api.example.com/submitvaluenum{)keynestedzConnection timed out)r)rNr*rQrUtest_binary_responsez!https://api.example.com/image.pngsPNG  IHDR...)r)rNr*rRrT)NNNNNNN)r1r$rJr^rrtypingrr getLoggerrkr rrrintr(r?rMrHrcsetLevelDEBUGhasattrrerrrrs      6