"""Common asset GraphQL queries."""

create_asset_upload_token = """
mutation(
    $assetType: String!
) {
  createAssetUploadToken(
    assetType: $assetType
  ) {
    bucket
    filename
    credentials {
        token
        awsAccessKeyId
        awsSecretAccessKey
        expiration
    }
  }
}
"""

save_asset_info = """
mutation (
  $assetType: String!,
  $filename: String!,
  $productId: Int!,
  $upc: String!,
  $trackUniqueId: Int!,
  $token: String!,
  $originalFilename: String,
  $isCorrection: Boolean
) {
  saveAssetInfo(
    input: {
      assetType: $assetType,
      filename: $filename
      productId: $productId,
      upc: $upc,
      trackUniqueId: $trackUniqueId,
      token: $token,
      originalFilename: $originalFilename,
      isCorrection: $isCorrection
    }
  ) {
    status
  }
}
"""

get_asset_status_v2_query = """
query($filename: String!){
  assetStatus(
      filename: $filename
  ) {
    status
  }
}
"""

get_product_track_assets_query = """
query($productId: ID!, $useV2: Boolean){
  productTrackAssets(
      productId: $productId
      useV2: $useV2
  ) {
    assets {
        trackUniqueId
        status
        correction {
          trackUniqueId
          status
        }
    }
  }
}
"""

valid_artwork_query = """
query($productId: String!, $requireV1: Boolean!) {
  product(productId: $productId, cached: false) {
    isArtworkValid(requireV1: $requireV1)
  }
}
"""