#!/usr/bin/python3

import requests
import sys
import json
import ipinfo
import os

def virusTotal(ip):
    URL = "https://www.virustotal.com/api/v3/ip_addresses/"+ip
    headers = {
        'x-apikey': '84f08fdfaec261212fc5cda36822aec9c98649e31e5394e8ded96ada75d5fe8c',
        }
    response = requests.get(URL, headers=headers)
    
    content = response.content
    data = content.decode('utf-8').replace("'", '"')
    #data = json.loads(data, ',')
    #data = json.dumps(data, indent=4)
    
    with open(newFile+"/virustotal.json", "w") as outfile:
        outfile.write(data)
    #print(response.status_code)
    #print(data)

def ipInfo(ip):
    access_token = 'ecfd9fa58fc0f0'
    handler = ipinfo.getHandler(access_token)
    details = handler.getDetails(ip)
    
    ##data = json.dumps(details, indent=4)
    #print(type(details))
    with open(newFile+"/ipinfo.json", "w") as outfile:
        json.dump(details.all, outfile)
    

ip = sys.argv[1]
if not ip:
   raise ValueError

newFile = ip + "-results"

os.mkdir(newFile)

virusTotal(ip)
ipInfo(ip)