import os
from wo.core.shellexec import WOShellExec
from wo.core.logging import Log


class SSL:

    def getExpirationDays(self, domain, returnonerror=False):
        # check if exist
        if not os.path.isfile('/etc/letsencrypt/live/{0}/cert.pem'
                              .format(domain)):
            Log.error(self, 'File Not Found: '
                      '/etc/letsencrypt/live/{0}/cert.pem'
                      .format(domain), False)
            if returnonerror:
                return -1
            Log.error(self, "Check the WordOps log for more details "
                      "`tail /var/log/wo/wordops.log` and please try again...")

        current_date = WOShellExec.cmd_exec_stdout(self, "date -d \"now\" +%s")
        expiration_date = WOShellExec.cmd_exec_stdout(self, "date -d \""
                                                      "`openssl x509 -in "
                                                      "/etc/letsencrypt/live/"
                                                      "{0}/cert.pem"
                                                      " -text -noout|grep "
                                                      "\"Not After\"|"
                                                      "cut -c 25-`\" "
                                                      "+%s".format(domain))

        days_left = int((int(expiration_date) - int(current_date)) / 86400)
        if (days_left > 0):
            return days_left
        else:
            # return "Certificate Already Expired ! Please Renew soon."
            return -1

    def getExpirationDate(self, domain):
        # check if exist
        if not os.path.isfile('/etc/letsencrypt/live/{0}/cert.pem'
                              .format(domain)):
            Log.error(self, 'File Not Found: /etc/letsencrypt/'
                      'live/{0}/cert.pem'
                      .format(domain), False)
            Log.error(self, "Check the WordOps log for more details "
                      "`tail /var/log/wo/wordops.log` and please try again...")

        expiration_date = WOShellExec.cmd_exec_stdout(self, "date -d "
                                                      "\"`openssl x509 -in "
                                                      "/etc/letsencrypt/live/"
                                                      "{0}/cert.pem"
                                                      " -text -noout|grep "
                                                      "\"Not After\"|"
                                                      "cut -c 25-`\" "
                                                      .format(domain))
        return expiration_date