update sh

bin/container/domainctl.sh

@@ -1,11 +1,24 @@
 #!/usr/bin/env bash
 CK_RESULT=''
-HTTPD_CONF='httpd_config.conf'
+LSDIR='/usr/local/lsws'
+LS_HTTPD_CONF="${LSDIR}/conf/httpd_config.xml"
+OLS_HTTPD_CONF="${LSDIR}/conf/httpd_config.conf"
 
 help_message(){
     echo 'Command [-add|-del] [domain_name]'
-    echo 'Example 1: domain-ctl.sh -add example.com'
-    echo 'Example 2: domain-ctl.sh -del example.com'
+    echo 'Example 1: domainctl.sh -add example.com'
+    echo 'Example 2: domainctl.sh -del example.com'
+}
+
+check_lsv(){
+    if [ -f ${LSDIR}/bin/openlitespeed ]; then
+        LSV='openlitespeed'
+    elif [ -f ${LSDIR}/bin/litespeed ]; then
+        LSV='lsws'
+    else
+        echo 'Version not exist, abort!'
+        exit 1     
+    fi
 }
 
 dot_escape(){
@@ -23,7 +36,7 @@ fst_match_after(){
     FIRST_NUM_AFTER=$(tail -n +${1} ${2} | grep -n -m 1 ${3} | awk -F ':' '{print $1}')
 }
 lst_match_line(){
-    fst_match_after ${1} ${2} '}'
+    fst_match_after ${1} ${2} ${3}
     LAST_LINE_NUM=$((${FIRST_LINE_NUM}+${FIRST_NUM_AFTER}-1))
 }
 
@@ -47,34 +60,73 @@ www_domain(){
     WWW_DOMAIN=$(echo www.${1})
 }
 
+add_ls_domain(){
+    fst_match_line 'ccl.xml</templateFile>' ${LS_HTTPD_CONF}
+    NEWNUM=$((FIRST_LINE_NUM+1))
+    sed -i "${NEWNUM}i \ \ \ \ \ \ <member>\n \ \ \ \ \ \ \ <vhName>${DOMAIN}</vhName>\n \ \ \ \ \ \ \ <vhDomain>${DOMAIN},${WWW_DOMAIN}</vhDomain>\n \ \ \ \ \ \ </member>" ${LS_HTTPD_CONF}
+}
+
+add_ols_domain(){
+    perl -0777 -p -i -e 's/(vhTemplate centralConfigLog \{[^}]+)\}*(^.*listeners.*$)/\1$2
+  member '${DOMAIN}' {
+    vhDomain              '${DOMAIN},${WWW_DOMAIN}'
+  }/gmi' ${OLS_HTTPD_CONF}
+}
+
 add_domain(){
+    check_lsv
     dot_escape ${1}
     DOMAIN=${ESCAPE}
     www_domain ${1}
-    check_duplicate "member.*${DOMAIN}" ${HTTPD_CONF}
-    if [ "${CK_RESULT}" != '' ]; then
-        echo "# It appears the domain already exist! Check the ${HTTPD_CONF} if you believe this is a mistake!"
-        exit 1
-    else
-        perl -0777 -p -i -e 's/(vhTemplate centralConfigLog \{[^}]+)\}*(^.*listeners.*$)/\1$2
-  member '${1}' {
-    vhDomain              '${1},${WWW_DOMAIN}'
-  }/gmi' ${HTTPD_CONF}     
+    if [ "${LSV}" = 'lsws' ]; then
+        check_duplicate "vhDomain.*${DOMAIN}" ${LS_HTTPD_CONF}
+        if [ "${CK_RESULT}" != '' ]; then
+            echo "# It appears the domain already exist! Check the ${LS_HTTPD_CONF} if you believe this is a mistake!"
+            exit 1
+        fi    
+    elif [ "${LSV}" = 'openlitespeed' ]; then
+        check_duplicate "member.*${DOMAIN}" ${OLS_HTTPD_CONF}
+        if [ "${CK_RESULT}" != '' ]; then
+            echo "# It appears the domain already exist! Check the ${OLS_HTTPD_CONF} if you believe this is a mistake!"
+            exit 1
+        fi        
     fi
+    add_ls_domain
+    add_ols_domain
+}
+
+del_ls_domain(){
+    fst_match_line "<vhName>*${1}" ${LS_HTTPD_CONF}
+    FIRST_LINE_NUM=$((FIRST_LINE_NUM-1))
+    lst_match_line ${FIRST_LINE_NUM} ${LS_HTTPD_CONF} '</member>'
+    sed -i "${FIRST_LINE_NUM},${LAST_LINE_NUM}d" ${LS_HTTPD_CONF}
+}
+
+del_ols_domain(){
+    fst_match_line ${1} ${OLS_HTTPD_CONF}
+    lst_match_line ${FIRST_LINE_NUM} ${OLS_HTTPD_CONF} '}'
+    sed -i "${FIRST_LINE_NUM},${LAST_LINE_NUM}d" ${OLS_HTTPD_CONF}    
 }
 
 del_domain(){
+    check_lsv
     dot_escape ${1}
     DOMAIN=${ESCAPE}
-    check_duplicate "member.*${DOMAIN}" ${HTTPD_CONF}
-    if [ "${CK_RESULT}" = '' ]; then
-        echo "# We couldn't find the domain you wanted to remove! Check the ${HTTPD_CONF} if you believe this is a mistake!"
-        exit 1
-    else
-        fst_match_line ${1} ${HTTPD_CONF}
-        lst_match_line ${FIRST_LINE_NUM} ${HTTPD_CONF}
-        sed -i "${FIRST_LINE_NUM},${LAST_LINE_NUM}d" ${HTTPD_CONF}
+    if [ "${LSV}" = 'lsws' ]; then
+        check_duplicate "vhDomain.*${DOMAIN}" ${LS_HTTPD_CONF}
+        if [ "${CK_RESULT}" = '' ]; then
+            echo "# Domain non-exist! Check the ${LS_HTTPD_CONF} if you believe this is a mistake!"
+            exit 1
+        fi
+    elif [ "${LSV}" = 'openlitespeed' ]; then
+        check_duplicate "member.*${DOMAIN}" ${OLS_HTTPD_CONF}
+        if [ "${CK_RESULT}" = '' ]; then
+            echo "# Domain non-exist! Check the ${OLS_HTTPD_CONF} if you believe this is a mistake!"
+            exit 1
+        fi        
     fi
+    del_ls_domain ${1}
+    del_ols_domain ${1}
 }
 
 check_input ${1}