hcornet
506716e703
Some checks failed
Deployment Verification / deploy-and-test (push) Failing after 29s
110 lines
3.0 KiB
Python
110 lines
3.0 KiB
Python
import json
|
|
import yaml
|
|
|
|
items = []
|
|
|
|
openapi = {
|
|
"openapi": "3.0.2",
|
|
"info": {
|
|
"title": "MISP",
|
|
"description": "MISP API generated from the misp book: https://github.com/MISP/misp-book/blob/master/automation/README.md",
|
|
"version": "1.0.0",
|
|
"contact": {
|
|
"name": "@frikkylikeme",
|
|
"url": "https://twitter.com/frikkylikeme",
|
|
"email": "frikky@shuffler.io"
|
|
}
|
|
},
|
|
"paths": {},
|
|
"components": {
|
|
"schemas": {},
|
|
"securitySchemes": {
|
|
"ApiKeyAuth": {
|
|
"type": "apikey",
|
|
"in": "header",
|
|
"name": "Authorization",
|
|
}
|
|
},
|
|
}
|
|
}
|
|
|
|
with open("misp.txt", "r") as tmp:
|
|
newitem = {}
|
|
recorditem = False
|
|
|
|
counter = 0
|
|
itemsplit = tmp.read().split("\n")
|
|
for item in itemsplit:
|
|
counter += 1
|
|
if item.startswith("### ") and "/" in item:
|
|
try:
|
|
path = item.split(" ")[2]
|
|
method = item.split(" ")[1].lower()
|
|
newitem = {
|
|
"path": path,
|
|
"method": method,
|
|
}
|
|
|
|
try:
|
|
openapi["paths"][path][method] = {}
|
|
except KeyError:
|
|
openapi["paths"][path] = {}
|
|
openapi["paths"][path][method] = {}
|
|
|
|
except IndexError:
|
|
newitem = {}
|
|
continue
|
|
|
|
recorditem = True
|
|
#print(newitem)
|
|
|
|
if not recorditem:
|
|
continue
|
|
|
|
if "Description" in item:
|
|
openapi["paths"][newitem["path"]][newitem["method"]]["description"] = itemsplit[counter+1]
|
|
elif "URL Arguments" in item:
|
|
parameters = []
|
|
innercnt = 0
|
|
|
|
openapi["paths"][newitem["path"]][newitem["method"]]["parameters"] = []
|
|
while True:
|
|
curline = itemsplit[counter+1+innercnt]
|
|
if "#" in curline:
|
|
break
|
|
|
|
innercnt += 1
|
|
if not curline:
|
|
continue
|
|
|
|
print(curline)
|
|
parameters.append({
|
|
"description": curline.split(" ")[1],
|
|
"in": "query",
|
|
"name": curline.split(" ")[1],
|
|
"required": True,
|
|
"schema": {"type": "string"},
|
|
})
|
|
|
|
openapi["paths"][newitem["path"]][newitem["method"]]["parameters"] = parameters
|
|
elif "Output" in item:
|
|
# FIXME
|
|
innercnt = 0
|
|
while True:
|
|
curline = itemsplit[counter+1+innercnt]
|
|
|
|
if "#" in curline:
|
|
break
|
|
|
|
innercnt += 1
|
|
if "json" in curline:
|
|
continue
|
|
#print(curline)
|
|
|
|
print(json.dumps(openapi, indent=4))
|
|
|
|
|
|
generatedfile = "generated/misp.yaml"
|
|
with open(generatedfile, "w+") as tmp:
|
|
tmp.write(yaml.dump(openapi))
|