generate-clients.sh

#!/bin/bash

CONTAINER=openapitools/openapi-generator-cli:v5.0.1
CONTAINER_TOOL=$([ -x /usr/bin/podman ] && echo podman || echo docker)

usage() {
    echo -e "Usage:\n $0 [ --docker | --podman ] [ --inventory ] [ --vmaas ] [ --rbac ]" >&2
}

OPTS=$(getopt --longoptions=docker,podman,help,inventory,vmaas,rbac -n ${0##*/} -- dphvir "$@")

if [ $? != 0 ] ; then echo "Terminating..." >&2 ; exit 1 ; fi
eval set -- "$OPTS"

declare -A GENERATE
while : ; do
    case "$1" in
        -d|--docker) CONTAINER_TOOL=docker
            ;;
        -p|--podman) CONTAINER_TOOL=podman
            ;;
        -i|--inventory) GENERATE[inventory]=1
            ;;
        -v|--vmaas) GENERATE[vmaas]=1
            ;;
        -r|--rbac) GENERATE[rbac]=1
            ;;
        --) break
            shift
            ;;
        -h|--help|*) usage
                  exit 1
            ;;
    esac
    shift
done

# if none of API was explictly set generate all of them
if [ -z "${GENERATE[*]}" ] ; then
    GENERATE=([inventory]=1 [vmaas]=1 [rbac]=1)
fi

# For now, we skip the dateTime parsing because of incompatible formats ( inventory does not produce fully compliant

function filter_oneof() {
    # filter out oneOf types which openapi generator can't handle
    sed -i '
            # inventory
            s|"additionalProperties":{"oneOf":\[{"$ref":"#/components/schemas/SystemProfileNestedObject","x-scope":\[""\]},{"not":{"type":"object"}}\]}|"additionalProperties":{"not":{"type":"object"}}|g;
            s|{"oneOf":\[{"type":"string"},{"type":"boolean"}\]}|{"type":"string"}|g;
            s|Fqdn|fqdn|g;

            # rbac
            s|"CrossAccountRequestDetail":{"oneOf":\[{"\$ref":"#/components/schemas/CrossAccountRequestDetailByAccount"},{"\$ref":"#/components/schemas/CrossAccountRequestDetailByUseId"}\]}|"CrossAccountRequestDetail":{"\$ref":"#/components/schemas/CrossAccountRequestDetailByAccount"}|g;
            s|"items":{"oneOf":\[{"\$ref":"#/components/schemas/CrossAccountRequestByAccount"},{"\$ref":"#/components/schemas/CrossAccountRequestByUserId"}\]}|"items":{"\$ref":"#/components/schemas/CrossAccountRequestByAccount"}|g;

            # vmaas
            s|"type": {"oneOf": \[{"type": "string", "example": "security"}, {"type": "array", "items": {"type": "string", "example": "security"}}\]},|"type": {"type": "array", "items": {"type": "string", "example": "security"}},|;
            s|"severity": {"oneOf": \[{"type": "string", "enum": \["Low", "Moderate", "Important", "Critical", null\], "nullable": true}, {"type": "array", "items": {"type": "string", "enum": \["Low", "Moderate", "Important", "Critical", null\], "nullable": true}}\]}},| "severity": {"type": "array", "items": {"type": "string", "enum": \["Low", "Moderate", "Important", "Critical", null\], "nullable": true}}},|;
            s|"items":{"oneOf":\[{"properties":{"count":{"type":"integer"},"value":{"type":"string"}},"type":"object"},{"properties":{"count":{"type":"integer"},"value":{"type":"boolean"}},"type":"object"}\]},|"items":{"properties":{"count":{"type":"integer"},"value":{"type":"string"}},"type":"object"},|g;
            s|{"\$ref": "#/components/schemas/VulnerabilityList"}|{"type": "array","items": { "type": "string","example": ""}}|g;

            # workaround for converting nullable date-time to nullable strings
            s|"format":"date-time","nullable":true,"type":"string"|"nullable":true,"type":"string"|g;
            ' $1
}

function generate_client() {
  NAME=$1
  SPEC_URL=$2

  LOCAL_JSON="$NAME/api/openapi.json"
  curl $SPEC_URL -o "$LOCAL_JSON"

  filter_oneof "$LOCAL_JSON"

  HERE=$(pwd)
  # remove old code and doc
  find "$HERE/$NAME" -name client_test.go -prune -o \( -name '*.go' -o -name '*.md' \) -print0 \
        | xargs -0 rm -f

  $CONTAINER_TOOL run --rm -v ${HERE}:/local:z --security-opt=label=disable $CONTAINER generate \
      -i "/local/$LOCAL_JSON" \
      -g go \
      --api-package $NAME \
      -p packageName=$NAME,isGoSubmodule=true \
      --git-host "github.com" --git-user-id RedHatInsights --git-repo-id patchman-clients \
      --type-mappings DateTime=string \
      -o /local/$NAME

    if [ "$NAME" == "inventory" ] ; then
        sed -i 's/uNKNOWNBASETYPE/createCheckIn/g; s/UNKNOWN_BASE_TYPE/CreateCheckIn/g;' $NAME/api_hosts.go
    fi
}


$CONTAINER_TOOL image exists $CONTAINER || $CONTAINER_TOOL pull $CONTAINER

# Generate Inventory client
if [ -n "${GENERATE[inventory]}" ] ; then
    if [[ -z $INVENTORY_ADDRESS ]]; then
        echo "Using default inventory address (CI)"
        INVENTORY_ADDRESS=https://ci.cloud.redhat.com
    fi
generate_client inventory "${INVENTORY_ADDRESS}/api/inventory/v1/openapi.json"
fi

# Generate Vmaas client
if [ -n "${GENERATE[vmaas]}" ] ; then
    if [[ -z $VMAAS_ADDRESS ]]; then
        echo "Using default vmaas address (CI)"
        VMAAS_ADDRESS=https://webapp-vmaas-ci.5a9f.insights-dev.openshiftapps.com
    fi
generate_client vmaas "${VMAAS_ADDRESS}/api/vmaas/v3/openapi.json"
fi

# Generate RBAC client
if [ -n "${GENERATE[rbac]}" ] ; then
    if [[ -z $RBAC_ADDRESS ]]; then
        echo "Using default RBAC address (CI)"
        RBAC_ADDRESS=https://ci.cloud.redhat.com
    fi
generate_client rbac "${RBAC_ADDRESS}/api/rbac/v1/openapi.json"
fi