#!/usr/bin/env python3
import csv

# 用户提供的钓鱼网站信息
new_threats = [
    ["钓鱼网站", "Silver fox 团伙", "2", "baiduwenshen.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "11bucketyun.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "supervt.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "ossbaiwenj.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "kefubahaohonsheng.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "shimo-oss1.oss-cn-hangzhou.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "jubaopengosssi.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "winios2024.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "jpbdoss.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "423down.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "osstesto.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "51yunpio.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "mustdll.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "condongjkhdsgdsd.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "alibwj.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙-Sload 远控木马", "2", "adll.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "jubaopengosser.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙-Sload 远控木马", "2", "aexe.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "whitefile.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "shunfengoss.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "titi2-4.oss-cn-hangzhou.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "alidll.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "bucketossbj.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "muchengoss.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "variety.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "aliyunlianjieoss.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "alivt.oss-cn-hongkong.aliyuncs.com"],
    ["钓鱼网站", "Silver fox 团伙", "2", "wangzheguilaioss.oss-cn-hongkong.aliyuncs.com"]
]

# 读取现有的威胁数据库文件
csv_file = "/root/dns/static/domain-info/threats/threats-database.csv"

# 存储已存在的域名
existing_domains = set()

# 读取文件并收集已存在的域名
with open(csv_file, 'r', encoding='utf-8') as f:
    reader = csv.reader(f)
    next(reader)  # 跳过表头
    for row in reader:
        if len(row) >= 4:
            existing_domains.add(row[3])

# 过滤出不存在的新威胁
unique_new_threats = []
for threat in new_threats:
    domain = threat[3]
    if domain not in existing_domains:
        unique_new_threats.append(threat)
        existing_domains.add(domain)  # 添加到已存在集合中，避免重复添加

# 将新威胁添加到文件末尾
if unique_new_threats:
    with open(csv_file, 'a', newline='', encoding='utf-8') as f:
        writer = csv.writer(f)
        writer.writerows(unique_new_threats)
    print(f"已添加 {len(unique_new_threats)} 个新的钓鱼网站到威胁数据库")
else:
    print("所有提供的钓鱼网站已存在于威胁数据库中，无需添加")