Inspect.hs

{-# LANGUAGE CPP, NoImplicitPrelude, OverloadedStrings, DoAndIfThenElse, FlexibleContexts #-}

{- |
Description:    Generates inspections when asked for by the frontend.

-}
module IHaskell.Eval.Inspect (inspect) where

import           ClassyPrelude
import qualified Prelude as P

import           Data.List.Split (splitOn)

import           Exception (ghandle)

import           IHaskell.Eval.Evaluate (Interpreter)
import           IHaskell.Display
import           IHaskell.Eval.Util (getType)
import           IHaskell.Types

-- | Characters used in Haskell operators.
operatorChars :: String
operatorChars = "!#$%&*+./<=>?@\\^|-~:"

-- | Whitespace characters.
whitespace :: String
whitespace = " \t\n"

-- | Compute the identifier that is being queried.
getIdentifier :: String -> Int -> String
getIdentifier code pos = identifier
  where
    chunks = splitOn whitespace code
    lastChunk = P.last chunks :: String
    identifier =
      if all (`elem` operatorChars) lastChunk
        then "(" ++ lastChunk ++ ")"
        else lastChunk


inspect :: String -- ^ Code in the cell
        -> Int    -- ^ Cursor position in the cell
        -> Interpreter (Maybe Display)
inspect code pos = do
  let identifier = getIdentifier code pos
      handler :: SomeException -> Interpreter (Maybe a)
      handler _  = return Nothing
  response <- ghandle handler (Just <$> getType identifier)
  let prefix = identifier ++ " :: "
      fmt str = Display [plain $ prefix ++ str]
  return $ fmt <$> response