7jdZddlmZddlZddlZddlZddlZddlmZm Z ddl m Z m Z ddl mZddl mZddlmZdd lmZmZmZdd lmZmZdd lmZed Zdd ZddZ ddZ!ddZ"ddZ#ddZ$ddZ%ddZ&e%ee e!e"e#e$dZ'dddZ(y)aCommand-line interface and command handlers for the WolfPanel Agent. Commands: run run as a service (heartbeat loop + initial discovery) register --token X register with the Central API using an install token heartbeat send a single heartbeat discover --quick|--full run discovery (read-only) update-check report whether a newer version is available status print local agent status version print the agent version ) annotationsN)ApiErrorget_api_client) AGENT_VERSION load_config)full)quick) send_once) ensure_dirsload_agent_tokenstore_credentials) get_logger setup_logging)checkclic|j}|s-tdtdtdtdyt|t|}tj j }||jtj j|jdxs|jd|jdxs|jd tj jtj jd } |j|}|jd }|jd} |r| tjdyt!|t#| |td| d|jdy#t$r }tjd |Yd }~yd }~wwxYw)zExchange an install token for long-lived agent credentials. SECURITY: the install token is used in-memory only and is never written to disk. Only the returned agent/refresh tokens are persisted. z No install token provided. z*Open: https://app.wolfpanel.net/connect zPairing Code: DEV-XXXX uH(Browser pairing is not implemented in v1 — pass --token to register.) distro_idsystemdistro_versionrelease) install_token agent_versionhostnameos_name os_version cpu_cores memory_totalzregistration failed: %sN agent_token server_idz-registration response missing required fieldsz#Registered successfully. server_id=z status=statusr)tokenprintr rquick_discoveryros_inforrget cpu_countmemory_total_bytesregisterrlogerrorr str) configargsr api_clientr%payloadresponseexcrr s =/var/www/downloads.wolfpanel.net/public_html/agent/src/cli.py cmd_registerr4&s JJM  ./ <= +, XY'J$$,,.G'--#**335;;{+Dw{{8/Dkk"23Mw{{97M$++557'..AACG&&w/,,}-K [)I I1 ABfc)nk: / {(8< z Up to date: r)r update_checkr#update_availablecurrentlatest)r-r.r/infos r3cmd_update_checkrMksm'J  +D | /0  "4<<.T[[MBC   T\\N+, r7cvt|}|jr|rdnd}td|jxsdtd|jtd|jtd|j td|j td |rd nd td |y )N registeredpending_installz server_id: zz version: z channel: z api_url: z api_mock: z token: presentmissingz state: r)r r r#rrelease_channelapi_urlapi_mock)r-r.r"states r3 cmd_statusrWxs V $E#--%L>OE L))=-=> ?@ L--. /0 L//0 12 L( )* L) *+ Le; <= L ! r7c"tty)Nr)r#r)r-r.s r3 cmd_versionrYs  - r7c<t|t|}tjd|j|j  t j||d tj||d t||tj|j ,#t$r }tjd|Yd}~kd}~wwxYw#t$r }tjd|Yd}~~d}~wwxYw#t$rtjdYywxYw) zEService mode: one initial discovery, then a resilient heartbeat loop.z)agent starting (version=%s, interval=%ss)Tr<zinitial quick scan failed: %sNzinitial discovery failed: %szagent stopping (interrupt)r)r rr*rLrheartbeat_intervalr$rC ExceptionwarningrBr timesleepKeyboardInterrupt)r-r.r/r2s r3cmd_runras'JHH 8&:N:N  & &( :FJt<96:d; fj ) JJv00 1 : 3S99: 9 2C889  -.sBB"C5-C:" C +CC  C7C22C7:DDc*tjdd}|jddd|jdd }|j d d |j dd }|jdd |j dd |j dd }|jddd|jddd|jddd|j dd |j dd |j d d! |S)"Nzwolfpanel-agentzWolfPanel Agent)prog descriptionz --verbose store_truezdebug-level logging)actionhelpcommandT)destrequiredrCz!run as a service (heartbeat loop))rgr)zregister with the Central APIz--tokenz/install token (omit to see pairing placeholder) heartbeatzsend a single heartbeatdiscoverzrun read-only discoveryz--quickzquick baseline snapshotz--fullzfull inventory (default)z --no-uploadzdo not upload snapshot update-checkz%check the release channel for updatesr!zprint local agent statusversionzprint the agent version)argparseArgumentParser add_argumentadd_subparsers add_parser)parsersub p_register p_discovers r3 build_parserrxs  $ $*;IZ [F  L?TU   Y  >CNN5BNC 1PQJI,]^NN;%>N? 1JKJIlAZ[H\@Z[M,E]^NN>(ONPNN8" $ $T *D ]F&""4<<'--W\\Z T\\ "64 00r7)returnint)rzargparse.ArgumentParser)N)rzlist[str] | Nonerr))__doc__ __future__rror@r~r^r/rrr-rrr>rrBr r$rkr identityr r r loggerrrupdaterrHr*r4r6rFrMrWrYrarxrrr7r3rs #  /-,. -( 1 h5       :0 $  1r7