�
�hG � � � d Z ddlZddlmZ ddlZddlZddlZddlm a m
Z
m Z de fd�Z
defd�Zde fd �Zd
e defd �Zd
e defd �Zdee e f fd
�Zy)u6
应用管理模块 - 提供Android应用管理功能
� N)�Tuple� )�_device�
set_device�
get_device�returnc �J � t j ddgdd�� t j ddgdd�� j } d| vry t j � }|sy
t |� |j d � |j d � |j d
� |j d� t j d� |j y# t j $ r Y yt $ r Y yw xY w# Y �lxY w# Y yxY w# t $ r}dt |� � �cY d}~S d}~ww xY w)un
初始化 uiautomator2，包括安装和启动服务
Returns:
str: 初始化结果信息
�adb�devicesT��check�capture_outputz=Failed to initialize UIAutomator2: ADB service is not runningz8Failed to initialize UIAutomator2: ADB command not found)r �text�devicez6Failed to initialize UIAutomator2: No device connectedzAFailed to initialize UIAutomator2: Could not establish connectionzahttps://github.com/openatx/android-uiautomator-server/releases/download/2.3.1/app-uiautomator.apkzfhttps://github.com/openatx/android-uiautomator-server/releases/download/2.3.1/app-uiautomator-test.apk�com.github.uiautomator�com.github.uiautomator.test� z%UIAutomator2 initialized successfullyzIFailed to initialize UIAutomator2: Service not running after installationz#Failed to initialize UIAutomator2: N)�
subprocess�run�CalledProcessError�FileNotFoundError�stdout�u2�connectr � app_install� app_start�time�sleep�info� Exception�str)r r �es �8/Volumes/dz/code/uiautomator2-mcp/src/mcp_android/app.py�init_uiautomator2r$ s1 � � ,>� N� �N�N�E�9�-�T�$� O� �.�.�%��!3�D�t�T�[�[�� �7� "�K� ��� �� �V� �6�� � � � � C�
D� � � � H�
I�
���1�2����6�7� �
�
�1�
� _� �K�K�:��G �,�,� S�R� � N�M� N��, � �� _�^�� � >�4�S��V�H�=�=��>�s� �C �(D �D � D �)"C3 � 7D � C: �C0�#D �%C0�-D �/C0�0D �3C7�5D �:C>�<D �>D � D"�
D �D"� D"c � � ddddddd�} t j ddgdd�� d| d< t � t j � at du| d < | d sd
| d
< | S t j d| d< t j g d�� j }d|v | d< t j t j d�| d< | S # t j t f$ r
d | d
< | cY S w xY w# t $ r }d t |� � �| d
< | cY d}~S d}~ww xY w# d| d< Y ��xY w# d| d< Y | S xY w# t $ r}t |� | d
< | cY d}~S d}~ww xY w)up
检查 uiautomator2 是否正确安装和运行
Returns:
dict: 包含检查结果的字典
FN)�
adb_server�device_connected�service_running�
app_installed� device_info�errorr
r Tr r& z#ADB server not running or not foundr+ zFailed to connect device: r' zNo device connectedr( )�pm�list�packagesr r) )�serialr r* zFailed to get device info)
r r r r r r r r r! r �shell�outputr/ )�statusr" r. s r# �check_uiautomator2r3 @ s� � � � !� � ���
�F�.� � �N�N�E�9�-�T�$� O�#'�F�<� � �?�
� �*�*�,��
&-�D�%8��!�"��(�)� 3�F�7�O��M� .� �L�L�(,�F�$� %�
�=�=�!;�<�C�C��":�h�"F��� � @�!�.�.� � � �%�F�=� !� �
��M �-�-�/@�A� � C�F�7�O��M� �� �
�$>�s�1�v�h�"G��w�� �
��
�� .�(-�F�$� %�� @�$?�F�=� !��
�� � � �a�&��w���
���s� � C �D* �C( �D* � D �3(D* � $D � D* � C%�"D* �$C%�%D* �( D�1D �D�D* � D�D* �D�D* � D'�$D* �'D* �* E
�3E�E
�E
c � � t �yt j d� t j d� t j d� t j d� t j d� t j d� t j
d� t j
d� t j d� t j y # Y y
xY w# t $ r} d t | � � �cY d} ~ S d} ~ ww xY w) uQ
重启 uiautomator2 服务
Returns:
str: 重启结果信息
Nz=Device not initialized. Please call init_uiautomator2() firstr r r z$am force-stop com.github.uiautomatorz)am force-stop com.github.uiautomator.testr z+UIAutomator2 service restarted successfullyzAFailed to restart UIAutomator2: Service not running after restartz(Failed to restart UIAutomator2 service: ) r �app_stopr r r0 r r r r! )r" s r# �restart_uiautomator2r6 � s� � �C� �?�R� ���1�2����6�7� �
�
�1�
� �
�
�<�=��
�
�A�B� �
�
�1�
� ���2�3����7�8� �
�
�1�
� W� �L�L�@�� W�V�� � C�9�#�a�&��B�B��C�s; �C �B=C �C �C �C � C � D �(
C;�5D �;D � package_namec � � t � } |j | � |j | d�� y# Y �xY w# t $ r Y yw xY w)uq
启动应用
Args:
package_name: 应用包名
Returns:
bool: 是否启动成功
T)�waitF)r r5 r r �r7 r s r# � start_appr; � sT � �
�� �� � �O�O�L� )�
��� �D��1��� � ��
� ���s �
9 �2 �9 �6�9 � A�Ac �Z � t � }|j | � y# t $ r Y yw xY w)uq
停止应用
Args:
package_name: 应用包名
Returns:
bool: 是否停止成功
TF)r r5 r r: s r# �stop_appr= � s/ � ��� ����� �%��� � ���s � � *�*c �j � t � } | j � }|d |d fS # t $ r Y yw xY w)u{
获取当前运行的应用包名和活动名
Returns:
Tuple[str, str]: (package_name, activity_name)
�package�activity)� rA )r � app_currentr )r �currents r# �get_current_apprD � sE � � �\�F���$�$�&���y�!�7�:�#6�6�6�� � ���s �& � 2�2)�__doc__� uiautomator2r �typingr r r �os�androidr r r r! r$ �dictr3 r6 �boolr; r= rD � � r# �<module>rN s� ��� � � � � � 4� 4�2>�3� 2>�h>�D� >�@"C�c� "C�H�C� �D� �.
�3�
�4�
� ��s�C�x� � rM
