#!/usr/bin/env python3
#
# ex:ts=4:sw=4:sts=4:et
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
#
# Security Response Tool Commandline Tool
#
# Copyright (C) 2018       Wind River Systems
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

### Usage Examples (run from top level directory)
# Updating Defect System Issues:   ./bin/acme/srtool_defect.py -U


import os
import sys
import re
import csv
import xml.etree.ElementTree as ET
import argparse
import sqlite3
import subprocess
import json
import urllib
from time import sleep
from datetime import datetime

# load the srt.sqlite schema indexes
dir_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
sys.path.insert(0, dir_path)
from common.srt_schema import ORM

try:
    from datetime import datetime, date
    from urllib.request import urlopen, URLError
    from urllib.parse import urlparse
except ImportError:
    from urllib2 import urlopen, URLError
    from urlparse import urlparse

srtDbName = 'srt.sqlite'
srtErrorLog = 'srt_errors.txt'

#################################
# Helper methods
#

verbose = False

def debugMsg(msg):
    if verbose:
        print(msg)

overrides = {}

def set_override(key,value=None):
    if not value is None:
        overrides[key] = value
    elif key in os.environ.keys():
        overrides[key] = 'yes' if os.environ[key].startswith('1') else 'no'
    else:
        overrides[key] = 'no'
    if 'yes' == overrides[key]:
        print("OVERRIDE: %s = %s" % (key,overrides[key]))

def get_override(key):
    if key in overrides.keys():
        return 'yes' == overrides[key]
    return False

def srt_error_log(msg):
    f1=open(srtErrorLog, 'a')
    f1.write("|" + msg + "|\n" )
    f1.close()

def get_tag_key(tag,key):
    d = json.loads(tag)
    return d[key]

#################################
# class to hold fields of a Defect System issues
#

class Defect:
    project = ''
    product_id = ''

    def __init__(self,*args, **kwargs):
        self.reset()

    # reset all but persistent project info
    def reset(self):
        self.id = -1
        self.name = ''
        self.summary = ''
        self.url = ''
        self.priority = -1
        self.status = ''
        self.resolution = 'Unresolved'
        self.publish = ''

        # RCPL
        self.release_version = ''

        self.date_created = ''
        self.date_updated = ''

        # extra fields
        self.cve_status = ''
        self.vi_status  = ''
        self.vi_outcome  = ''


#################################
# Import Defect System defects
#

#################################
# Defect System list update
#

#################################
# Defect System add to investigation
#

#################################
# defect_del_from_defect_db
#

#################################
# New defect
#

#################################
# Init/Update from Defect System status
#

#################################
# main loop
#

def main(argv):
    global force_update
    global verbose
    global master_log
    global srt_user
    global srt_passwd

    parser = argparse.ArgumentParser(description='srtool_defect.py: Manage the SRTool to Defect System connection')
    parser.add_argument('--init-defect', '-i', action='store_const', const='init_defect', dest='command', help='Init and import Defect System states and update defects')
    parser.add_argument('--update-defect', '-u', action='store_const', const='update_defect', dest='command', help='Import Defect System states and update defects')
    parser.add_argument('--update-defect-list', '-l', dest='defect_update_list', help='List of Defect System defects to update in SRTool database')
    parser.add_argument('--defect_er', '-e', nargs=1, help='Query list of pending ERs under a project, review them, and assign to Rally themes')
    parser.add_argument('--force', '-f', action='store_true', dest='force_update', help='Force updates')
    parser.add_argument('--verbose', '-v', action='store_true', dest='verbose', help='Verbose debugging')
    parser.add_argument('--user', dest='user', help='User name for Defect System access')
    parser.add_argument('--passwd', dest='passwd', help='User password for Defect System access')

    parser.add_argument('--add', nargs=1, help='Add an existing defect to SRTool defect database')
    parser.add_argument('--delete', nargs=1, help='Delete an existing defect from SRTool defect database')
    parser.add_argument('--new', nargs=6, help='Create a new defect "--new <program> <summary> <description> <priority> <components> <urllink>"')
    parser.add_argument('--defect-projects', '-P', action='store_const', const='defect-projects', dest='command', help='Defect System projects')

    args = parser.parse_args()

    master_log = open("./update_logs/master_log.txt", "a")

    # Authorization
    if args.user:
        srt_user = args.user
    else:
        srt_user = os.environ.get('SRT_USER')
    if args.passwd:
        srt_passwd = args.passwd
    else:
        srt_passwd = os.environ.get('SRT_PASSWD')
    if not srt_user or not srt_passwd:
        msg = "FATAL ERROR: Missing user/password for Defect System access"
        print(msg)
        srt_error_log(msg)
        return 1

    force_update = False
    if None != args.force_update:
        force_update = args.force_update
    if None != args.verbose:
        verbose = True

    defect_list = ''
    if None != args.defect_update_list:
        defect_list = args.defect_update_list

    if args.defect_er:
        #get_defect_er(args.defect_er[0])
        print("NOT IMPLEMENTED")
    elif args.add:
        #defect_add_to_defect_db(args.add[0])
        print("NOT IMPLEMENTED")
    elif args.delete:
        #defect_del_from_defect_db(args.add[0])
        print("NOT IMPLEMENTED")
    elif args.new:
        #defect_new_defect(args.new[0],args.new[1],args.new[2],args.new[3],args.new[4],args.new[5])
        print("NOT IMPLEMENTED")
    elif 'init_defect' == args.command:
        #update_defect(True)
        print("NOT IMPLEMENTED")
    elif 'update_defect' == args.command:
        #update_defect(False)
        print("NOT IMPLEMENTED")
    elif defect_list:
        #defect_update_list(defect_list)
        print("NOT IMPLEMENTED")
    elif 'defect-projects' == args.command:
        #get_projects()
        print("NOT IMPLEMENTED")
    else:
        print("Command not found")

if __name__ == '__main__':
    global srtool_basepath

    if verbose: print("srtool_defect(%s)" % sys.argv[1:])

    # fetch any environment overrides
    #set_override('SRTDBG_MINIMAL_DB')

    srtool_basepath = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(sys.argv[0]))))
    main(sys.argv[1:])