camelCaseFn

Author: pyramation

__tests__/assetlist.test.ts

@@ -12,7 +12,7 @@ it('assetlist', () => {
 it('assetlist camelCase', () => {
     const code = generateTypeScript(schema as any, {
         useSingleQuotes: true,
-        useCamelCase: true
+        camelCase: true
     });
     expect(code).toMatchSnapshot();
     writeFileSync(__dirname + '/../__fixtures__/output/assetlist.camel.ts', code);

__tests__/chain.test.ts

@@ -12,7 +12,7 @@ it('chain', () => {
 it('chain camelCase', () => {
     const code = generateTypeScript(schema as any, {
         useSingleQuotes: true,
-        useCamelCase: true
+        camelCase: true
     });
     expect(code).toMatchSnapshot();
     writeFileSync(__dirname + '/../__fixtures__/output/chain.camel.ts', code);

__tests__/contract.test.ts

@@ -12,7 +12,7 @@ it('contract', () => {
 it('contract camelCase', () => {
     const code = generateTypeScript(schema as any, {
         useSingleQuotes: true,
-        useCamelCase: true
+        camelCase: true
     });
     expect(code).toMatchSnapshot();
     writeFileSync(__dirname + '/../__fixtures__/output/contract.camel.ts', code);

__tests__/memo.test.ts

@@ -12,7 +12,7 @@ it('memo', () => {
 it('memo camelCase', () => {
     const code = generateTypeScript(schema as any, {
         useSingleQuotes: true,
-        useCamelCase: true
+        camelCase: true
     });
     expect(code).toMatchSnapshot();
     writeFileSync(__dirname + '/../__fixtures__/output/memo.camel.ts', code);

src/context.ts

@@ -3,7 +3,8 @@ import deepmerge from 'deepmerge';
 
 export interface SchemaTSOptions {
   useSingleQuotes: boolean;
-  useCamelCase: boolean;
+  camelCase?: boolean; // defaults to false
+  camelCaseFn?: (str: string) => string; // optional function to convert keys to camelCase
   strictTypeSafety: boolean;  // true uses { [k: string]: unknown; }, false uses any
 }
 
@@ -44,6 +45,7 @@ export class SchemaTSContext implements SchemaTSContextI {
 
 export const defaultOptions: SchemaTSOptions = { 
   useSingleQuotes: true, 
-  useCamelCase: false,
+  camelCase: false,
+  camelCaseFn: null,
   strictTypeSafety: true
 };

src/schema.ts

@@ -128,7 +128,9 @@ function createPropertySignature(
 ): t.TSPropertySignature {
 
   const isIdent = isValidIdentifier(key);
-  const name = ctx.options.useCamelCase ? toCamelCase(key) : key;
+  let camelCaseFn: (str: string) => string = toCamelCase;
+  if (ctx.options.camelCaseFn) camelCaseFn = ctx.options.camelCaseFn;
+  const name = ctx.options.camelCase ? camelCaseFn(key) : key;
   const propType = getTypeForProp(ctx, prop, required, schema);
   const identifier = isIdent ? t.identifier(name) : t.stringLiteral(key);
   const propSig = t.tsPropertySignature(