{-
  Copyright (c) Meta Platforms, Inc. and affiliates.
  All rights reserved.

  This source code is licensed under the BSD-style license found in the
  LICENSE file in the root directory of this source tree.
-}

{-# OPTIONS_GHC -Wno-orphans #-}

module Glean.Glass.SymbolId.Java (
    flattenPath
    {- instances -}

  ) where

import qualified Glean
import Glean.Glass.Types
import Glean.Glass.SymbolId.Class
import Glean.Haxl.Repos (RepoHaxl)
import Data.Text ( Text )
import qualified Data.Text as Text

import qualified Glean.Schema.CodeJava.Types as Java
import qualified Glean.Schema.CodeKotlin.Types as Kotlin
import qualified Glean.Schema.JavaAlpha.Types as Java
import qualified Glean.Schema.KotlinAlpha.Types as Kotlin
import qualified Glean.Schema.JavakotlinAlpha.Types as JavaKotlin

instance Symbol Java.Entity where
  toSymbol e = case e of
    Java.Entity_decl d -> toSymbol d
    Java.Entity_EMPTY -> return []

instance Symbol Kotlin.Entity where
  toSymbol e = case e of
    Kotlin.Entity_decl d -> toSymbol d
    Kotlin.Entity_EMPTY -> return []

instance Symbol Java.Declaration where
  toSymbol e = case e of
    Java.Declaration_class_ c -> toSymbolPredicate c
    Java.Declaration_interface_ i -> toSymbolPredicate i
    Java.Declaration_enum_ e -> toSymbolPredicate e
    Java.Declaration_method e -> toSymbolPredicate e
    Java.Declaration_ctor e -> toSymbolPredicate e
    Java.Declaration_local e -> toSymbolPredicate e
    Java.Declaration_field e -> toSymbolPredicate e
    Java.Declaration_param e -> toSymbolPredicate e
    Java.Declaration_EMPTY -> return []

instance Symbol Kotlin.Declaration where
  toSymbol e = case e of
    Kotlin.Declaration_class_ c -> toSymbolPredicate c
    Kotlin.Declaration_method e -> toSymbolPredicate e
    Kotlin.Declaration_variable e -> toSymbolPredicate e
    Kotlin.Declaration_EMPTY -> return []

instance Symbol Java.ClassDeclaration_key where
  toSymbol Java.ClassDeclaration_key{..} =
    toSymbolPredicate classDeclaration_key_name

instance Symbol Kotlin.ClassDeclaration_key where
  toSymbol Kotlin.ClassDeclaration_key{..} =
    toSymbolPredicate classDeclaration_key_name

instance Symbol Java.InterfaceDeclaration_key where
  toSymbol Java.InterfaceDeclaration_key{..} =
    toSymbolPredicate interfaceDeclaration_key_name

instance Symbol Java.EnumDeclaration_key where
  toSymbol Java.EnumDeclaration_key{..} =
    toSymbolPredicate enumDeclaration_key_name

instance Symbol Java.MethodDeclaration_key where
  toSymbol Java.MethodDeclaration_key{..} =
    toSymbolPredicate methodDeclaration_key_name

instance Symbol Kotlin.MethodDeclaration_key where
  toSymbol Kotlin.MethodDeclaration_key{..} =
    toSymbolPredicate methodDeclaration_key_name

instance Symbol Java.ConstructorDeclaration_key where
  toSymbol Java.ConstructorDeclaration_key{..} =
    toSymbolPredicate constructorDeclaration_key_name

instance Symbol Java.LocalDeclaration_key where
  toSymbol Java.LocalDeclaration_key{..} =
    toSymbolPredicate localDeclaration_key_name

instance Symbol Java.FieldDeclaration_key where
  toSymbol Java.FieldDeclaration_key{..} =
    toSymbolPredicate fieldDeclaration_key_name

instance Symbol Java.ParameterDeclaration_key where
  toSymbol Java.ParameterDeclaration_key{..} =
    toSymbolPredicate parameterDeclaration_key_name

instance Symbol Kotlin.VariableDeclaration_key where
  toSymbol Kotlin.VariableDeclaration_key{..} =
    toSymbolPredicate variableDeclaration_key_name

--
-- generic to both Java and Kotlin
--

instance Symbol JavaKotlin.MethodName_key where
  toSymbol JavaKotlin.MethodName_key{..} = do
    qname <- toSymbolPredicate methodName_key_name
    types <- mapM encodeType methodName_key_signature
    return $ case types of
      [] -> qname
      _ -> qname ++ (".t" : types)

instance Symbol JavaKotlin.Path_key where
  toSymbol JavaKotlin.Path_key{..} = path_key_container <:> path_key_base

instance Symbol JavaKotlin.Path where
  toSymbol path = toSymbol =<< Glean.keyOf path

instance Symbol JavaKotlin.QName_key where
  toSymbol JavaKotlin.QName_key{..} = qName_key_context <:> qName_key_name

instance Symbol JavaKotlin.Name where
  toSymbol name = (:[]) <$> Glean.keyOf name

--
-- symbol ids for overloading: encode parameter types
--
encodeType :: JavaKotlin.Type -> RepoHaxl u w Text
encodeType ty = do
  key <- Glean.keyOf ty
  case key of
    JavaKotlin.Type_key_primitive prim -> return prim

    JavaKotlin.Type_key_object path -> do
      Text.intercalate "." . reverse <$> (flattenPath =<< Glean.keyOf path)

    JavaKotlin.Type_key_array ty -> do
      textTy <- encodeType ty
      return (textTy <> "[]")
    JavaKotlin.Type_key_EMPTY{} -> return "<unknown>"

--
-- names
--

instance ToQName Java.Entity where
  toQName e = case e of
    Java.Entity_decl d -> toQName d
    Java.Entity_EMPTY -> return $ Left "unknown Java.Entity"

instance ToQName Kotlin.Entity where
  toQName e = case e of
    Kotlin.Entity_decl d -> toQName d
    Kotlin.Entity_EMPTY -> return $ Left "unknown Kotlin.Entity"

instance ToQName Java.Declaration where
  toQName d = case d of
    Java.Declaration_enum_ e -> Glean.keyOf e >>= toQName
    Java.Declaration_class_ e -> Glean.keyOf e >>= toQName
    Java.Declaration_interface_ e -> Glean.keyOf e >>= toQName
    Java.Declaration_method e -> Glean.keyOf e >>= toQName
    Java.Declaration_ctor e -> Glean.keyOf e >>= toQName
    Java.Declaration_field e -> Glean.keyOf e >>= toQName
    Java.Declaration_local e -> Glean.keyOf e >>= toQName
    Java.Declaration_param e -> Glean.keyOf e >>= toQName
    Java.Declaration_EMPTY -> return $ Left "Unknown Java.Declaration type"

instance ToQName Kotlin.Declaration where
  toQName d = case d of
    Kotlin.Declaration_class_ e -> Glean.keyOf e >>= toQName
    Kotlin.Declaration_method e -> Glean.keyOf e >>= toQName
    Kotlin.Declaration_variable e -> Glean.keyOf e >>= toQName
    Kotlin.Declaration_EMPTY -> return $ Left "Unknown Kotlin.Declaration type"

instance ToQName Java.ClassDeclaration_key where
  toQName Java.ClassDeclaration_key{..} =
    Glean.keyOf classDeclaration_key_name >>= toQName

instance ToQName Kotlin.ClassDeclaration_key where
  toQName Kotlin.ClassDeclaration_key{..} =
    Glean.keyOf classDeclaration_key_name >>= toQName

instance ToQName Java.InterfaceDeclaration_key where
  toQName Java.InterfaceDeclaration_key{..} =
    Glean.keyOf interfaceDeclaration_key_name >>= toQName

instance ToQName Java.EnumDeclaration_key where
  toQName Java.EnumDeclaration_key{..} =
    Glean.keyOf enumDeclaration_key_name >>= toQName

instance ToQName Java.MethodDeclaration_key where
  toQName Java.MethodDeclaration_key{..} =
    Glean.keyOf methodDeclaration_key_name >>= toQName

instance ToQName Kotlin.MethodDeclaration_key where
  toQName Kotlin.MethodDeclaration_key{..} =
    Glean.keyOf methodDeclaration_key_name >>= toQName

-- ctor qnames are indexed as "<init>" but should be the parent name + sig
instance ToQName Java.ConstructorDeclaration_key where
  toQName Java.ConstructorDeclaration_key{..} = do
    parentName <- case constructorDeclaration_key_container of
      Java.Definition_class_ cdecl -> toQName =<< Glean.keyOf cdecl
      Java.Definition_interface_ idecl -> toQName =<< Glean.keyOf idecl
      Java.Definition_enum_ edecl -> toQName =<< Glean.keyOf edecl
      Java.Definition_EMPTY -> return $
        Left "Java constructor with no parent container"
    case parentName of
      Left _err -> Glean.keyOf constructorDeclaration_key_name >>= toQName
      Right (Name n, Name rest) -> return $
        Right (Name n, Name (rest <> "." <> n))

instance ToQName Java.LocalDeclaration_key where
  toQName Java.LocalDeclaration_key{..} =
    Glean.keyOf localDeclaration_key_name >>= toQName

instance ToQName Java.FieldDeclaration_key where
  toQName Java.FieldDeclaration_key{..} =
    Glean.keyOf fieldDeclaration_key_name >>= toQName

instance ToQName Java.ParameterDeclaration_key where
  toQName Java.ParameterDeclaration_key{..} =
    Glean.keyOf parameterDeclaration_key_name >>= toQName

instance ToQName Kotlin.VariableDeclaration_key where
  toQName Kotlin.VariableDeclaration_key{..} =
    Glean.keyOf variableDeclaration_key_name >>= toQName

--
-- Generic to both Java and Kotlin
--

instance ToQName JavaKotlin.MethodName_key where
  toQName JavaKotlin.MethodName_key{..} =
    Glean.keyOf methodName_key_name >>= toQName

instance ToQName JavaKotlin.QName_key where
  toQName JavaKotlin.QName_key{..} = do
    nameStr <- Glean.keyOf qName_key_name
    context <- flattenPath =<< Glean.keyOf qName_key_context
    return $ Right (Name nameStr, Name (Text.intercalate "." (reverse context)))

flattenPath :: JavaKotlin.Path_key -> RepoHaxl u w [Text]
flattenPath JavaKotlin.Path_key{..} = do
  nameStr <- Glean.keyOf path_key_base
  case path_key_container of
    Nothing -> return [nameStr]
    Just path -> do
      rest <- flattenPath =<< Glean.keyOf path
      return (nameStr : rest)