#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
Created on 2015年1月23日

@author: xuxu
'''

import os
import platform
import re
import subprocess
import time

import exception

#判断系统类型，windows使用findstr，linux使用grep
system = platform.system()
if system is "Windows":
    find_util = "findstr"
else:
    find_util = "grep"
    
#判断是否设置环境变量ANDROID_HOME
if "ANDROID_HOME" in os.environ:
    if system == "Windows":
        command = os.path.join(os.environ["ANDROID_HOME"], "platform-tools", "adb.exe")
    else:
        command = os.path.join(os.environ["ANDROID_HOME"], "platform-tools", "adb")
else:
    raise EnvironmentError(
        "Adb not found in $ANDROID_HOME path: %s." %os.environ["ANDROID_HOME"])
    
#adb命令
def adb(args):
    cmd = "%s %s" %(command, str(args))
    return subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

#adb shell命令
def shell(args):
    cmd = "%s shell %s" %(command, str(args))
    return subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

#获取设备状态
def get_state():
    return os.popen("adb get-state").read().strip()

#获取对应包名的pid
def get_app_pid(pkg_name):   
    if system is "Windows":
        string = shell("ps | findstr %s$" %pkg_name).stdout.read()
        
    string = shell("ps | grep -w %s" %pkg_name).stdout.read()

    if string == '':
        return "the process doesn't exist."

    pattern = re.compile(r"\d+")
    result = string.split(" ")
    result.remove(result[0])

    return  pattern.findall(" ".join(result))[0]

#杀掉对应包名的进程
def kill_process(pkg_name):
    pid = get_app_pid(pkg_name)
    
    result = shell("kill %s" %str(pid)).stdout.read().split(": ")[-1]
    
    if result != "":
        raise exception.SriptException("Operation not permitted or No such process")

#获取设备上当前应用的包名与activity    
def get_focused_package_and_activity():
    pattern = re.compile(r"[a-zA-Z0-9\.]+/.[a-zA-Z0-9\.]+")
    out = shell("dumpsys window w | %s \/ | %s name=" %(find_util, find_util)).stdout.read()

    return pattern.findall(out)[0]

#获取当前应用的包名
def get_current_package_name():
    return get_focused_package_and_activity().split("/")[0]

#获取当前设备的activity
def get_current_activity():
    return get_focused_package_and_activity().split("/")[-1]

#时间戳
def timestamp():
    return time.strftime('%Y-%m-%d-%H-%M-%S',time.localtime(time.time()))

#连接设备
# adb("kill-server").wait()
# adb("start-server").wait()
adb("wait-for-device")

if get_state() != "device":
    adb("kill-server").wait()
    adb("start-server").wait()
    
if get_state() != "device":
    raise exception.SriptException("Device not run")
    

if __name__ == "__main__":
    pass