o i L@sddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZddlZddlmZddlmZd ZGd d d eZdddZGdddZdS)) annotationsN)web)parse)args) AppSettings) TypedDictdefaultc@s.eZdZUded<ded<ded<ded<dS)FileInfostrpathintsizemodifiedcreatedN)__name__ __module__ __qualname____annotations__rr./mnt/c/Users/fbmor/ComfyUI/app/user_manager.pyr s  r r r relative_toreturncCs:tj||tjdtj|tj|tj|dS)N/)r rrr)osr relpathreplacesepgetsizegetmtimegetctime)r rrrr get_file_infos    r!c@s>eZdZddZddZddZdd d Zd d Zd dZdS) UserManagercCst}t||_tj|s#tj|ddtj s#t dt dtj rRtj | rMt| }t||_WddS1sFwYdSi|_dSddi|_dS)NTexist_okzd****** User settings have been changed to be stored on the server instead of browser storage. ******zg****** For multi-user setups add the --multi-user CLI argument to enable multiple user profiles. ******r ) folder_pathsget_user_directoryrsettingsrr existsmakedirsr multi_userloggingwarningisfileget_users_fileopenjsonloadusers)selfuser_directoryfrrr__init__%s    " zUserManager.__init__cCstjtdS)Nz users.json)rr joinr%r&r3rrrr.8szUserManager.get_users_filecCsPd}tjrd|jvr|jd}|tjrtd|||jvr&td||S)Nr z comfy-userzUnknown user: )rr*headers startswithr%SYSTEM_USER_PREFIXKeyErrorr2)r3requestuserrrrget_request_user_id;s     zUserManager.get_request_user_iduserdataTc Cs|dkr t}ntd|||}t|}|durdS|}tj||f|kr-dS|durQd|vr:t |}tj tj ||}tj||f|krQdStj |d} |rhtj | shtj| dd|S)Nr@zUnknown filepath type:%rTr#)r%r&r<r?get_public_user_directoryrr commonpathrunquoteabspathr7splitr(r)) r3r=filetype create_dirroot_dirr> user_rootr parentrrrget_request_user_filepathHs(     z%UserManager.get_request_user_filepathcCs|}|s td|tjrtdtdd|}|tjr%td|dtt }||j |<t | d}t |j |Wd|S1sNwY|S)Nzusername not providedzSystem User prefix not allowedz[^a-zA-Z0-9-_]+-_w)strip ValueErrorr:r%r;resubr uuiduuid4r2r/r.r0dump)r3nameuser_idr5rrradd_useris     zUserManager.add_userc sj||dfdd}|dfdd}|dfdd}|d fd d }dfdd |dfdd}|dfdd}|dfdd}|dfdd} dS)Nz/userscsBtjr tdjdSj|ddd}tdtj|dS)Nserver)storager2FrI)r\migrated) rr*r json_responser2rMrr r()r=user_dirr8rr get_users~s z)UserManager.add_routes..get_usersc s|IdH}|d}|jvrtjddiddSz|}Wnty@}ztjdt|iddWYd}~Sd}~wwt|S)NusernameerrorzDuplicate username.status)r0r2valuesrr_rZrRr )r=bodyrbrYer8rr post_userss" z*UserManager.add_routes..post_usersz /userdatacs |jjdd}|stjdddS||s!tjdddStjs.tjdd dS|jjd d d k}|jjd d d k|jjd d d k|r`tj t dd}n tj t d}dfdd fddt j ||dD}t |S)a List user data files in a specified directory. This endpoint allows listing files in a user's data directory, with options for recursion, full file information, and path splitting. Query Parameters: - dir (required): The directory to list files from. - recurse (optional): If "true", recursively list files in subdirectories. - full_info (optional): If "true", return detailed file information (path, size, modified time). - split (optional): If "true", split file paths into components (only applies when full_info is false). Returns: - 400: If 'dir' parameter is missing. - 403: If the requested path is not allowed. - 404: If the requested directory does not exist. - 200: JSON response with the list of files or file information. The response format depends on the query parameters: - Default: List of relative file paths. - full_info=true: List of dictionaries with file details. - split=true (and full_info=false): List of lists, each containing path components. dirrdzDirectory not providedrftextzInvalid directoryzDirectory not foundrecursetrue full_inforFz*** full_pathr rFileInfo | str | list[str]cs>rt|Stj|tjd}r|g|dS|S)Nr)r!rr rrrrF)rurel_path)rsr split_pathrrprocess_full_paths  zGUserManager.add_routes..listuserdata..process_full_pathcs g|] }tj|r|qSr)rr r-).0ru)ryrr s  z@UserManager.add_routes..listuserdata..) recursiveN)rur rrv)rel_urlquerygetrResponserMrr r(lowerr7globescaper_)r= directoryrqpatternresultsr8)rsr ryrxr listuserdatas(    z,UserManager.add_routes..listuserdataz /v2/userdatac s|jjdd}zt|}Wn$ty4}ztd|d|tj dddWYd}~Sd}~wwzj |dd d }|rIj ||d d }n|}Wn!t ym}ztd |tj d d dWYd}~Sd}~ww|swtj dddSt j |s||krtgStj dddSt j |stj dddSg}zt j|ddD]y\}}}|D]} t j || } t j | |t jd} || | ddq|D]P} t j || } t j | |t jd} | | dd}zt | }|j|d<|j|d<Wnty}ztd| d|WYd}~nd}~ww||qqWn%tyD}ztd|d|tj dddWYd}~Sd}~ww|jdd d!t|S)"a List files and directories in a user's data directory. This endpoint provides a structured listing of contents within a specified subdirectory of the user's data storage. Query Parameters: - path (optional): The relative path within the user's data directory to list. Defaults to the root (''). Returns: - 400: If the requested path is invalid, outside the user's data directory, or is not a directory. - 404: If the requested path does not exist. - 403: If the user is invalid. - 500: If there is an error reading the directory contents. - 200: JSON response containing a list of file and directory objects. Each object includes: - name: The name of the file or directory. - type: 'file' or 'directory'. - path: The relative path from the user's data root. - size (for files): The size in bytes. - modified (for files): The last modified timestamp (Unix epoch). r rlz!Failed to decode path parameter: z , Error: rdz$Invalid characters in path parameterrmNFr]zAccess denied for user: roz!Invalid user specified in requestzInvalid path requestedrpzRequested path not foundz!Requested path is not a directoryT)topdownrr)rXr rHrGrrzCould not stat file z: zError listing directory iz Error reading directory contentscSs|ddk|dfS)NrHrrX)r)xrrr<szBUserManager.add_routes..list_userdata_v2..)key)r}r~rrrD Exceptionr+r,rrrMr<rr r(r_isdirwalkr7rrrappendstatst_sizest_mtimeOSErrorrcsort)r=requested_rel_pathribase_user_pathtarget_abs_pathrrootdirsfilesdir_namedir_pathrw file_name file_path entry_infostats stat_errorr8rrlist_userdata_v2s         z0UserManager.add_routes..list_userdata_v2FrGcsZ|j|d}|stjddS||}|stjddS|r+tj|s+tjddS|S)Nrdrerorp) match_inforrrrMrr r()r= check_existsparamrGr r8rrget_user_data_path@s    z2UserManager.add_routes..get_user_data_pathz/userdata/{file}cs&|dd}t|ts|St|S)NTr) isinstancer r FileResponser=r rrr getuserdataNs    z+UserManager.add_routes..getuserdatac sd|}t|ts |S|jdddk}|jdddk}|s/tj|r/tj dddSzD| IdH}tj |}t j |d \}}z#t|d  }||Wdn1s\wYt||Wn t|Wn$ty} ztd |d | tj d ddWYd} ~ Sd} ~ ww|d} |rt|| } ntj|| } t| S)aC Upload or update a user data file. This endpoint handles file uploads to a user's data directory, with options for controlling overwrite behavior and response format. Query Parameters: - overwrite (optional): If "false", prevents overwriting existing files. Defaults to "true". - full_info (optional): If "true", returns detailed file information (path, size, modified time). If "false", returns only the relative file path. Path Parameters: - file: The target file path (URL encoded if necessary). Returns: - 400: If 'file' parameter is missing. - 403: If the requested path is not allowed. - 409: If overwrite=false and the file already exists. - 200: JSON response with either: - Full file information (if full_info=true) - Relative file path (if full_info=false) The request body should contain the raw file content to be written. overwriterrfalsersFile already existsrmN)rkwbzError saving file 'z': rdz@Invalid filename. Please avoid special characters like :\/*?"<>|)rfreason)rr r~rrrr r(rrreaddirnametempfilemkstempfdopenwriterunlinkrr+r,rMr!rr_) r=r rrsrhrfdtmp_pathr5ri user_pathresprr3rr post_userdataVsD       z-UserManager.add_routes..post_userdatacs2|dd}t|ts|St|tjddS)NTrre)rr rremoverrrrrrdelete_userdatas     z/UserManager.add_routes..delete_userdataz/userdata/{file}/move/{dest}cs|dd}t|ts|S|ddd}t|ts|S|jdddk}|jd ddk}|s?tj|r?tj d d d St d |d|dt |||d}|r_t||}ntj||}t|S)a Move or rename a user data file. This endpoint handles moving or renaming files within a user's data directory, with options for controlling overwrite behavior and response format. Path Parameters: - file: The source file path (URL encoded if necessary) - dest: The destination file path (URL encoded if necessary) Query Parameters: - overwrite (optional): If "false", prevents overwriting existing files. Defaults to "true". - full_info (optional): If "true", returns detailed file information (path, size, modified time). If "false", returns only the relative file path. Returns: - 400: If either 'file' or 'dest' parameter is missing - 403: If either requested path is not allowed - 404: If the source file does not exist - 409: If overwrite=false and the destination file already exists - 200: JSON response with either: - Full file information (if full_info=true) - Relative file path (if full_info=false) TrFdest)rrrrrrrsrrrmzmoving 'z' -> ''N)rr r~rrrr r(rrr+infoshutilmoverMr!rr_)r=sourcerrrsrrrrr move_userdatas$       z-UserManager.add_routes..move_userdata)FrG)r' add_routesrpostdelete) r3routesrarjrrrrrrrrrr{s$   ?i> zUserManager.add_routesN)r@T) rrrr6r.r?rMrZrrrrrr"$s ! r")r r rr rr ) __future__rr0rrSrUrrr+raiohttprurllibrcomfy.cli_argsrr% app_settingsrtypingr default_userr r!r"rrrrs&