import streamlit as st
import subprocess

# 创建一个标题
st.title('Python 解析域名IP地址')

# 创建一个文本框并接收用户输入
user_input = st.text_input('输入域名http://')

# 判断用户是否有输入，并输出一句问候语
if user_input != '':
    st.write('正在解析：,', user_input, '......')
    # 检测dig是否安装
    try:
        subprocess.check_output(["dig", "-v"])
    except subprocess.CalledProcessError:
        # 如果未安装，则安装dig
        subprocess.call(["sudo", "apt-get", "update"])
        subprocess.call(["sudo", "apt-get", "install", "-y", "dnsutils"])

    # 使用subprocess运行dig命令并获取输出
    result = subprocess.run(["dig", "+noall", "+answer", user_input ], capture_output=True)

    # 将输出转换为表格
    output_lines = result.stdout.decode('utf-8').strip().split('\n')
    table = [line.split() for line in output_lines]

    # 将表格输出到streamlit界面
    st.table(table)
else:
    st.write('这是一个dig的UI界面，Stremlit太强到了')