import os
import argparse
from slither import Slither
from slither.utils.colors import red
import logging
from .slither_format import slither_format

logging.basicConfig()
logging.getLogger("Slither").setLevel(logging.INFO)

available_detectors = ["unused-state",
                       "solc-version",
                       "pragma",
                       "naming-convention",
                       "external-function",
                       "constable-states",
                       "constant-function"]

detectors_to_run = []

def parse_args():
    """
    Parse the underlying arguments for the program.
    :return: Returns the arguments for the program.
    """
    parser = argparse.ArgumentParser(description='slither_format',
                                     usage='slither_format filename')

    parser.add_argument('filename', help='The filename of the contract or truffle directory to analyze.')
    parser.add_argument('--solc', help='solc path', default='solc')
    parser.add_argument('--verbose', '-v', help='verbose mode output',action='store_true',default=False)
    
    group_detector = parser.add_argument_group('Detectors')
    group_detector.add_argument('--detect',
                                help='Comma-separated list of detectors, defaults to all, '
                                'available detectors: {}'.format(
                                    ', '.join(d for d in available_detectors)),
                                action='store',
                                dest='detectors_to_run',
                                default='all')
    return parser.parse_args()


def main():
    # ------------------------------
    #       Usage: python3 -m slither_format filename
    #       Example: python3 -m slither_format contract.sol
    # ------------------------------
    # Parse all arguments
    args = parse_args()

    # Perform slither analysis on the given filename
    slither = Slither(args.filename, is_truffle=os.path.isdir(args.filename), solc=args.solc, disable_solc_warnings=True)

    # Format the input files based on slither analysis
    slither_format(args, slither)
if __name__ == '__main__':
    main()