org.web3j.openapi.server.ExceptionMappings.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2020 Web3 Labs Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 */
package org.web3j.openapi.server

import com.fasterxml.jackson.core.JsonParseException
import com.fasterxml.jackson.databind.JsonMappingException
import org.eclipse.jetty.http.HttpStatus
import org.web3j.openapi.core.ErrorResponse
import org.web3j.protocol.exceptions.TransactionException
import org.web3j.tx.exceptions.ContractCallException
import javax.servlet.http.HttpServletRequest
import javax.ws.rs.NotFoundException
import javax.ws.rs.core.Context
import javax.ws.rs.core.HttpHeaders
import javax.ws.rs.core.Response
import javax.ws.rs.core.Response.Status
import javax.ws.rs.core.Response.Status.Family
import javax.ws.rs.core.UriInfo
import javax.ws.rs.ext.ExceptionMapper

sealed class BaseExceptionMapper(
    private val status: Response.StatusType
) : ExceptionMapper {

    @Context
    private lateinit var uriInfo: UriInfo

    @Context
    private lateinit var request: HttpServletRequest

    override fun toResponse(exception: E): Response {

        val error = ErrorResponse(
            title = exception.message ?: status.reasonPhrase,
            userAgent = request.getHeader(HttpHeaders.USER_AGENT),
            responseStatus = status.statusCode,
            requestMethod = request.method,
            requestUrl = uriInfo.requestUri.toString()
        )

        return Response.status(status.statusCode).entity(error).build()
    }
}

class JsonMappingExceptionMapper : BaseExceptionMapper(Status.BAD_REQUEST)
class JsonParseExceptionMapper : BaseExceptionMapper(Status.BAD_REQUEST)
class TransactionExceptionMapper : BaseExceptionMapper(Status.BAD_REQUEST)
class UnsupportedOperationExceptionMapper : BaseExceptionMapper(Status.BAD_REQUEST)
class IllegalArgumentExceptionMapper : BaseExceptionMapper(CustomStatus.UNPROCESSABLE_ENTITY)
class ContractCallExceptionMapper : BaseExceptionMapper(CustomStatus.UNPROCESSABLE_ENTITY)
class IllegalStateExceptionMapper : BaseExceptionMapper(CustomStatus.UNPROCESSABLE_ENTITY)
class RuntimeExceptionMapper : BaseExceptionMapper(Status.INTERNAL_SERVER_ERROR)
class NotFoundExceptionMapper : BaseExceptionMapper(Status.NOT_FOUND)

enum class CustomStatus(
    private val _statusCode: Int,
    private val _reasonPhrase: String
) : Response.StatusType {

    UNPROCESSABLE_ENTITY(
        HttpStatus.Code.UNPROCESSABLE_ENTITY.code,
        HttpStatus.Code.UNPROCESSABLE_ENTITY.message
    );

    override fun getStatusCode(): Int = _statusCode
    override fun getFamily(): Family = Family.familyOf(_statusCode)
    override fun getReasonPhrase(): String = _reasonPhrase
}