Support pure foreign functions

well-typed · Apr 19, 2024 · e86be9f · e86be9f
1 parent f70b040
commit e86be9f
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/test/Test/TraceForeignCalls/UsePlugin.hs b/test/Test/TraceForeignCalls/UsePlugin.hs
@@ -2,17 +2,29 @@
 {-# OPTIONS_GHC -fplugin=Plugin.TraceForeignCalls #-}
 
 module Test.TraceForeignCalls.UsePlugin (
+    -- * IO functions
     answerIO
+  , answerPure
   , slowAddIO
+  , slowAddPure
   ) where
 
 import Foreign.C
 
-foreign import capi "test_cbits.h answer"   c_answerIO  :: IO CInt
-foreign import capi "test_cbits.h slow_add" c_slowAddIO :: CLong -> CLong -> IO CLong
+foreign import capi "test_cbits.h answer"   c_answerIO   :: IO CInt
+foreign import capi "test_cbits.h answer"   c_answerPure ::    CInt
+
+foreign import capi "test_cbits.h slow_add" c_slowAddIO   :: CLong -> CLong -> IO CLong
+foreign import capi "test_cbits.h slow_add" c_slowAddPure :: CLong -> CLong ->    CLong
 
 answerIO :: IO Int
 answerIO = fromIntegral <$> c_answerIO
 
+answerPure :: Int
+answerPure = fromIntegral $ c_answerPure
+
 slowAddIO :: Int -> Int -> IO Int
 slowAddIO a b = fromIntegral <$> c_slowAddIO (fromIntegral a) (fromIntegral b)
+
+slowAddPure :: Int -> Int -> Int
+slowAddPure a b = fromIntegral $ c_slowAddPure (fromIntegral a) (fromIntegral b)