#!/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 Tkinter as tk
import ttk

import exception

serialno_num = ""

# 判断系统类型,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"])


def get_screen_size(window):
 return window.winfo_screenwidth(),window.winfo_screenheight()


def get_window_size(window):
 return window.winfo_reqwidth(),window.winfo_reqheight()


def center_window(root, width, height):
 screenwidth = root.winfo_screenwidth()
 screenheight = root.winfo_screenheight()
 size = '%dx%d+%d+%d' % (width, height, (screenwidth - width)/2, (screenheight - height)/2)
 root.geometry(size)


class Window(object):
 device_id = ""
 device_id_list = []
 device_name_list = []
 device_name_dict = {}
 root = None
 box = None

 def __init__(self, device_id_list, root):
 self.device_id_list = device_id_list
 self.device_name_dict = get_device_name_dict(self.device_id_list)
 self.get_device_name_list()
 self.device_id = device_id_list[0]
 self.root = root
 self.box = None

 def show_window(self):
 self.root.title(u'Serialno Number')
 center_window(self.root, 300, 240)
 self.root.maxsize(600, 400)
 self.root.minsize(300, 240)

 # options = self.device_id_list
 options = self.device_name_list
 self.box = ttk.Combobox(values=options)
 self.box.current(0)
 self.box.pack(expand = tk.YES)
 self.box.bind("<<ComboboxSelected>>", self.select)
 ttk.Button(text=u"确定", command=self.ok).pack(expand = tk.YES)

 self.root.mainloop()

 def select(self, event=None):
 for key, value in self.device_name_dict.iteritems():
 if value == self.box.selection_get():
 self.device_id = key
 # self.device_id = self.box.selection_get()

 def ok(self):
 global serialno_num
 serialno_num = self.device_id
 self.root.destroy()

 def get_device_name_list(self):
 for id in self.device_id_list:
 self.device_name_list.append(self.device_name_dict.get(id))


# adb命令
def adb(args):
 global serialno_num
 if serialno_num == "":
 devices = get_device_list()
 if len(devices) == 1:
 # global serialno_num
 serialno_num = devices[0]
 else:
 root = tk.Tk()
 window = Window(devices, root)
 window.show_window() 
 cmd = "%s -s %s %s" %(command, serialno_num, str(args))
 return subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

#adb shell命令
def shell(args):
 global serialno_num
 if serialno_num == "":
 devices = get_device_list()
 if len(devices) == 1:
 serialno_num = devices[0]
 else:
 root = tk.Tk()
 window = Window(devices, root)
 window.show_window() 
 cmd = "%s -s %s shell %s" %(command, serialno_num, str(args))
 return subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

#获取设备状态
def get_state():
 return os.popen("adb -s %s get-state" %serialno_num).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\.]+")
 tmp = shell("dumpsys activity | %s mFocusedActivity" %find_util).stdout.read()
 name = ""
 try:
 name = pattern.findall(tmp)[0]
 except:
 tmp = shell("dumpsys window w | %s \/ | %s name=" %(find_util, find_util)).stdout.read()
 name = pattern.findall(tmp)[0]
 return name


# 获取当前应用的包名
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()))


def get_device_list():
 devices = []
 result = subprocess.Popen("adb devices", shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.readlines()
 result.reverse()
 for line in result[1:]:
 if "attached" not in line.strip():
 devices.append(line.split()[0])
 else:
 break
 return devices

def get_device_name_dict(devices):
 device_dict = {}
 if not devices:
 return

 for device in devices:
 cmd = "adb -s %s shell getprop ro.product.model" %device
 device_name = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.readline().strip()
 device_dict[device] = device_name
 return device_dict
 

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

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__":
 print get_focused_package_and_activity()