feat: haskellize record/field names in schema init sub-command

vst · May 24, 2024 · e5bd93a · e5bd93a
1 parent 09cdeda
commit e5bd93a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/package.yaml b/package.yaml
@@ -27,6 +27,7 @@ library:
     - autodocodec-yaml
     - brick
     - bytestring
+    - casing
     - githash
     - hashable
     - hasql

diff --git a/src/Postmap/Spec/FromSchema.hs b/src/Postmap/Spec/FromSchema.hs
@@ -6,8 +6,10 @@ module Postmap.Spec.FromSchema where
 import qualified Data.Aeson as Aeson
 import qualified Data.ByteString.Lazy as BL
 import qualified Data.List as List
+import qualified Data.Text as T
 import Postmap.Introspect (Column (..), ColumnName (..), ColumnReference (..), Table (..), TableName (..))
 import Postmap.Spec.Types (Field (..), FieldName (..), FieldReference (..), Record (..), RecordName (..), Spec (..))
+import qualified Text.Casing as Casing
 
 
 fromSchemaJson :: BL.ByteString -> Either String Spec
@@ -70,12 +72,14 @@ fromColumn Column {..} =
     }
 
 
-mkFieldNameFromColumnName :: ColumnName -> FieldName
-mkFieldNameFromColumnName = MkFieldName . unColumnName
+mkRecordNameFromTableName :: TableName -> RecordName
+mkRecordNameFromTableName =
+  MkRecordName . T.pack . Casing.pascal . T.unpack . unTableName
 
 
-mkRecordNameFromTableName :: TableName -> RecordName
-mkRecordNameFromTableName = MkRecordName . unTableName
+mkFieldNameFromColumnName :: ColumnName -> FieldName
+mkFieldNameFromColumnName =
+  MkFieldName . T.pack . Casing.camel . T.unpack . unColumnName
 
 
 mkFieldReferenceFromColumnReference :: ColumnReference -> FieldReference
-Original file line number
+Diff line change
@@ Expand Up / @@ -27,6 +27,7 @@ library: @@
         - autodocodec-yaml
         - brick
         - bytestring
+        - casing
         - githash
         - hashable
         - hasql
@@ Expand Down @@