#!/bin/bash

echo "DNS 的安裝配置"
rpm -q bind
if [[ $? -eq 0 ]];then
    echo "DNS軟件已安裝"
else
    echo "安裝DNS軟件"
    yum install -y bind
    if [[ $? -eq 0 ]];then
        echo "安裝完成"
    else
        echo "安裝失敗"
    fi
fi
cd /etc
test -e /etc/named.conf.bak || cp /etc/named.conf /etc/named.conf.bak
echo "DNS配置"
read -p "是否配置DNS: [y/n] " ap
if [ ${ap} = "y" ]; then
sed -i 's/127.0.0.1/any/g' named.conf
sed -i 's/localhost/any/g' named.conf
while true; do
    read -p "創建區域名稱：" -t 20 qname
    grep ${qname} named.conf
    if [[ $? -eq 0 ]];then
        echo "區域名稱已被使用"
    else
        break;
    fi
done
cat << EOF >> named.conf
zone "${qname}" IN {
    type master;
    file "${qname}.zone";
    allow-update {none;};
};
EOF

ip=`ifconfig|grep inet | awk  'NR==1 {print $2}'`
echo ">>>>>>> ${ip}"
cd /var/named
echo "創建資源記錄文件....."
cp -a named.localhost ${qname}.zone
cat << EOF > ${qname}.zone
\$TTL 1D
@       IN SOA   ${qname}. root.${qname}. (
                                        0       ; serial
                                        1D      ; refresh
                                        1H      ; retry
                                        1W      ; expire
                                        3H )    ; minimum
        NS      ds.${qname}.
ds      IN A    ${ip}
EOF

while true; do
read -p "輸入主機名：" -t 20 n
read -p "輸入主機名對應IP：" -t 20 p
cat << EOF >> ${qname}.zone
${n}    IN A    ${p}
EOF
read -p "是否繼續添加：[y/n]" -t 20 op
if [ ${op} = n ]; then
    break;
fi
done
read -p "是否定義別名：[y/n]" -t 20 s
if [ ${s} = "y" ]; then
    while true; do
    read -p "輸入主機別名：" -t 20 na
    read -p "輸入主機名" -t 20 hn
    cat <<- EOF >> ${qname}.zone
    ${na}   CNAME   ${hn}
    EOF
    read -p "是否繼續添加：[y/n]" -t 20 op
    if [ ${op} = n ]; then
        break;
    fi
    done
fi
uname -a | grep el7
if [[ $? -eq 0 ]]; then
    service firewalld stop
else
    service iptables stop
fi
named-checkconf
named-checkzone ${qname} /var/named/${qname}.zone
service named restart
else 
    echo "退出腳本..."
fi