#
# This file generated and managed by Chef.
#
import sys
import deepsecurity as api
from deepsecurity.rest import ApiException as api_exception

def get_deployment_script(api, configuration, api_version, api_exception):
  deployment_script = api.AgentDeploymentScript()
  deployment_script.platform = 'linux'
  deployment_script.policyID = 67
  deployment_scripts_api = api.AgentDeploymentScriptsApi(api.ApiClient(configuration))

  try:
    deployment_script = deployment_scripts_api.generate_agent_deployment_script(api_version, agent_deployment_script = deployment_script)
    with open("deploy.sh", "w") as file:
      file.write(deployment_script.script_body + '\n')

  except api_exception as e:
    print("An exception occurred when calling AgentDeploymentScriptsApi.generate_agent_deployment_script: %s\n" % e)
    sys.exit(1)

if __name__ == '__main__':
  # Add Deep Security Manager host information to the api client configuration
  configuration = api.Configuration()
  configuration.host = 'https://app.deepsecurity.trendmicro.com/api'

  # Authentication
  configuration.api_key['api-secret-key'] = '<%= node['trend_micro']['ds_agent_api_secret_key'] %>'
  
  # Version
  api_version = 'v1'

  get_deployment_script(api, configuration, api_version, api_exception)