Dark magic of code transformation

Valeriy Kuzmin, Lilt, 2021

Problem: DTO type sharing

#[derive(TypescriptDefinition, TypeScriptify, 
Deserialize, Serialize)]
#[serde(tag = "tag")]
pub enum InventoryAction {
    Unknown,
    Split { from: Uuid, count: i32 },
    Merge { from: Uuid, to: Uuid },
    Move { item: Uuid, index: i32 },
}

Rust

Problem: DTO type sharing

export type InventoryActionUnknown = { tag: "Unknown" };

export type InventoryActionSplit = {
    tag: "Split",
    from: string,
    count: number,
};

// more variants...

export type InventoryAction = InventoryActionUnknown | InventoryActionSplit;


export class InventoryActionBuilder {
    static InventoryActionUnknown = (): InventoryActionUnknown => ({
         tag: "Unknown"
    });
    static InventoryActionSplit = ({
         from,
         count
    }): InventoryActionSplit => ({
         tag: "Split",
         from,
         count
    });

    // more variants...
}

Typescript

Problem: DTO type sharing

export type InventoryAction =
  | { tag: 'Unknown' }
  | { tag: 'Split'; from: Uuid; count: number }
  | { tag: 'Merge'; from: Uuid; to: number }
  | { tag: 'Move'; item: Uuid; index: number };

typescript-definitions crate

Task 1: Advanced unions

export type InventoryAction = smth;

ExportNamedDeclaration with TSTypeAliasDeclaration

.replaceWith((ex: ASTPath<ExportNamedDeclaration>) => {
  if (ex.value.declaration.type !== 'TSTypeAliasDeclaration') {
  	return ex.value;
  }
  // ...
}

Task 1: Advanced unions

... type InventoryAction = Elem1 | Elem2;

type id = Identifier + type annotation of TSUnionType

// ...
// separate declaration from export
const p = ex.value.declaration;
// returns p.id.typeName if p.id.type === 'Identifier'
const mainUnionName = getTypeDeclaraiontName(p);
if (!mainUnionName) {
  return ex.value;
}
// extract the right part of the assignment
const union = p.typeAnnotation;
if (!isTSUnionType(union)) {
   // not a union, not interesting
  return ex.value;
}
// ...
        

Task 1: Advanced unions

type C = {tag: "A"} | {tag: "B"};

union.types ➟ [['CA','A'], ['CB', 'B']] ➟ Record<Tag, ExtractedTypeName>

union.types ➟ [['A', <A type obj>], ['B', <B type obj>]] ➟ Record<Tag, ExtractedType>

type CA = {tag: "A"};
type CB = {tag: "B"};
type C = CA | CB;

Task 1: Advanced unions

{tag: "A", foo: "bar"}
type CA = {tag: "A", foo: "bar"};

TSTypeLiteral

TSPropertySignature[]

key.type = 'Identifier'

valuer.type = 'Identifier' | 'TSNumberKeyWord' | 'TSStringKeyword' | 'TSLiteralType'

ObjectTypeAnnotation

Task 2: Builder classes

export type C = CA | CB;

ExportNamedDeclaration with type = 'TSTypeAliasDeclaration'

.find(j.ExportNamedDeclaration)
.insertAfter((ex: ASTPath<ExportNamedDeclaration>) => {
  if (ex.value.declaration.type !== 'TSTypeAliasDeclaration') {
    return [];
  }
  const union = ex.value.declaration.typeAnnotation;
// ...

Task 2: Builder classes

return j.exportNamedDeclaration(
  j.classDeclaration.from({
    id: j.identifier(builderClassName),
    body: j.classBody(
      union.types
      .map((subType) => {
        if (!isTsTypeReference(subType)) {
          return null;
        }
        if (!isIdentifier(subType.typeName)) {
          return null;
        }
        return j.classProperty.from({
          key: j.identifier(subType.typeName.name),
          static: true,
          accessibility: 'public',
          declare: false,
          value: makeBuilderMethod(
            subType.typeName.name,
            typeMap,
            j,
            mainTypeName
          ),
        });
      })
      .filter((t) => !!t)
    ),
  })
)

The end!

Questions?

jscodeshift-2-at-lilt

By Valeriy Kuzmin

jscodeshift-2-at-lilt

  • 200