import sys

import os

import helpers

print('Script to generate stub file from a nuget package')

print(' Usage: python3 ' + sys.argv[0] +

' TEMPLATE NUGET_PACKAGE_NAME [VERSION=latest] [WORK_DIR=tempDir]')

print(' The script uses the dotnet cli, codeql cli, and dotnet format global tool')

print(' TEMPLATE should be either classlib or webapp, depending on the nuget package. For example, `Swashbuckle.AspNetCore.Swagger` should use `webapp` while `newtonsoft.json` should use `classlib`.')

if len(sys.argv) < 2:

print("\nPlease supply a template name.")

exit(1)

if len(sys.argv) < 3:

print("\nPlease supply a nuget package name.")

exit(1)

thisScript = sys.argv[0]

template = sys.argv[1]

nuget = sys.argv[2]

version = helpers.get_argv(3, "latest")

relativeWorkDir = helpers.get_argv(4, "tempDir")

generator = helpers.Generator(thisScript, relativeWorkDir, template)

generator.add_nuget(nuget, version)

generator.make_stubs()

exit(0)