Proper conversion Scalar for Long to String and back

2023-04-02 16:30:03 -04:00
parent 7debb27374
commit 106bda2097
2 changed files with 106 additions and 2 deletions
@@ -0,0 +1,105 @@
+package suwayomi.tachidesk.graphql.server
+
+import graphql.GraphQLContext
+import graphql.execution.CoercedVariables
+import graphql.language.StringValue
+import graphql.language.Value
+import graphql.scalar.CoercingUtil
+import graphql.schema.Coercing
+import graphql.schema.CoercingParseLiteralException
+import graphql.schema.CoercingParseValueException
+import graphql.schema.CoercingSerializeException
+import graphql.schema.GraphQLScalarType
+import java.util.Locale
+
+val GraphQLLongAsString: GraphQLScalarType = GraphQLScalarType.newScalar()
+    .name("LongString").description("A 64-bit signed integer as a String").coercing(GraphqlLongAsStringCoercing()).build()
+
+private class GraphqlLongAsStringCoercing : Coercing<Long, String> {
+    private fun toStringImpl(input: Any): String {
+        return input.toString()
+    }
+
+    private fun parseValueImpl(input: Any, locale: Locale): Long {
+        if (input !is String) {
+            throw CoercingParseValueException(
+                CoercingUtil.i18nMsg(
+                    locale,
+                    "String.unexpectedRawValueType",
+                    CoercingUtil.typeName(input)
+                )
+            )
+        }
+        return input.toLong()
+    }
+
+    private fun parseLiteralImpl(input: Any, locale: Locale): Long {
+        if (input !is StringValue) {
+            throw CoercingParseLiteralException(
+                CoercingUtil.i18nMsg(
+                    locale,
+                    "Scalar.unexpectedAstType",
+                    "StringValue",
+                    CoercingUtil.typeName(input)
+                )
+            )
+        }
+        return input.value.toLong()
+    }
+
+    private fun valueToLiteralImpl(input: Any): StringValue {
+        return StringValue.newStringValue(input.toString()).build()
+    }
+
+    @Deprecated("")
+    override fun serialize(dataFetcherResult: Any): String {
+        return toStringImpl(dataFetcherResult)
+    }
+
+    @Throws(CoercingSerializeException::class)
+    override fun serialize(
+        dataFetcherResult: Any,
+        graphQLContext: GraphQLContext,
+        locale: Locale
+    ): String {
+        return toStringImpl(dataFetcherResult)
+    }
+
+    @Deprecated("")
+    override fun parseValue(input: Any): Long {
+        return parseValueImpl(input, Locale.getDefault())
+    }
+
+    @Throws(CoercingParseValueException::class)
+    override fun parseValue(input: Any, graphQLContext: GraphQLContext, locale: Locale): Long {
+        return parseValueImpl(input, locale)
+    }
+
+    @Deprecated("")
+    override fun parseLiteral(input: Any): Long {
+        return parseLiteralImpl(input, Locale.getDefault())
+    }
+
+    @Throws(CoercingParseLiteralException::class)
+    override fun parseLiteral(
+        input: Value<*>,
+        variables: CoercedVariables,
+        graphQLContext: GraphQLContext,
+        locale: Locale
+    ): Long {
+        return parseLiteralImpl(input, locale)
+    }
+
+    @Deprecated("")
+    override fun valueToLiteral(input: Any): Value<*> {
+        return valueToLiteralImpl(input)
+    }
+
+    override fun valueToLiteral(
+        input: Any,
+        graphQLContext: GraphQLContext,
+        locale: Locale
+    ): Value<*> {
+        return valueToLiteralImpl(input)
+    }
+}
@@ -11,7 +11,6 @@ import com.expediagroup.graphql.generator.SchemaGeneratorConfig
 import com.expediagroup.graphql.generator.TopLevelObject
 import com.expediagroup.graphql.generator.hooks.FlowSubscriptionSchemaGeneratorHooks
 import com.expediagroup.graphql.generator.toSchema
-import graphql.Scalars
 import graphql.schema.GraphQLType
 import suwayomi.tachidesk.graphql.mutations.ChapterMutation
 import suwayomi.tachidesk.graphql.queries.CategoryQuery
@@ -27,7 +26,7 @@ import kotlin.reflect.KType

 class CustomSchemaGeneratorHooks : FlowSubscriptionSchemaGeneratorHooks() {
    override fun willGenerateGraphQLType(type: KType): GraphQLType? = when (type.classifier as? KClass<*>) {
-        Long::class -> Scalars.GraphQLString // encode to string for JS
+        Long::class -> GraphQLLongAsString // encode to string for JS
        else -> super.willGenerateGraphQLType(type)
    }
 }