#!/bin/bash
# shellcheck source=/dev/null
source /etc/mosdns/lib.sh

if uci_ext ssrp; then
  if [ "$1" = "unset" ]; then
    uci set shadowsocksr.@global[0].pdnsd_enable='1'
    uci set shadowsocksr.@global[0].tunnel_forward="$WAN_DNS0:53"
  elif [ "$1" = "" ]; then
    if [ "$(uci -q get mosdns.mosdns.listen_port)" -eq 5335 ]; then
      uci set shadowsocksr.@global[0].pdnsd_enable='0'
      uci del shadowsocksr.@global[0].tunnel_forward
      uci del shadowsocksr.@global[0].adblock_url
    else
      uci set shadowsocksr.@global[0].pdnsd_enable='1'
      uci set shadowsocksr.@global[0].tunnel_forward="127.0.0.1:$(uci -q get mosdns.mosdns.listen_port)"
    fi
  fi
  uci commit shadowsocksr
  if [ "$(pid ssrplus)" ]; then
    /etc/init.d/shadowsocksr restart
  fi
fi
if uci_ext pw; then
  if [ "$1" = "unset" ]; then
    uci set passwall.@global[0].dns_mode='dns2tcp'
    uci set passwall.@global[0].dns_forward="$WAN_DNS1"
    uci set passwall.@global[0].remote_dns="$WAN_DNS1"
    uci set passwall.@global[0].dns_cache='1'
    uci set passwall.@global[0].chinadns_ng='1'
  elif [ "$1" = "" ]; then
    uci set passwall.@global[0].dns_mode='udp'
    uci set passwall.@global[0].dns_forward="127.0.0.1:$(uci -q get mosdns.mosdns.listen_port)"
    uci set passwall.@global[0].remote_dns="127.0.0.1:$(uci -q get mosdns.mosdns.listen_port)"
    uci del passwall.@global[0].dns_cache
    uci del passwall.@global[0].chinadns_ng
  fi
  uci commit passwall
  if [ "$(pid passwall)" ]; then
    /etc/init.d/passwall restart
  fi
fi

if uci_ext pw2; then
  if [ "$1" = "unset" ]; then
    uci set passwall2.@global[0].direct_dns_protocol='auto'
    uci del passwall2.@global[0].direct_dns
    uci set passwall2.@global[0].remote_dns="$WAN_DNS0"
    uci set passwall2.@global[0].dns_query_strategy='UseIPv4'
  elif [ "$1" = "" ]; then
    uci set passwall2.@global[0].direct_dns_protocol='udp'
    uci set passwall2.@global[0].direct_dns="127.0.0.1:$(uci -q get mosdns.mosdns.listen_port)"
    uci set passwall2.@global[0].remote_dns_protocol='udp'
    uci set passwall2.@global[0].remote_dns="127.0.0.1:$(uci -q get mosdns.mosdns.listen_port)"
    uci set passwall2.@global[0].dns_query_strategy='UseIP'
  fi
  uci commit passwall2
  if [ "$(pid passwall2)" ]; then
    /etc/init.d/passwall2 restart
  fi
fi

if uci_ext vssr; then
  if [ "$1" = "unset" ]; then
    uci set vssr.@global[0].pdnsd_enable='1'
  elif [ "$1" = "" ]; then
    uci set vssr.@global[0].pdnsd_enable='0'
  fi
  uci commit vssr
  if [ "$(pid vssr)" ]; then
    /etc/init.d/vssr restart
  fi
fi

exit 0