guardContent function

Implementation

String guardContent(
  ServerLifecycleComponent component,
  String Function(Spec) formatter,
) {
  final (:positioned, :named) = getParams(
    component.params,
    defaultExpression: createGetFromDi(),
    useField: true,
  );

  final parameters = createConstructorParameters(component.params);
  final fields = createFields(component.params);
  final generics = createGenerics(component.genericTypes);

  final clazz = Class(
    (p) => p
      ..name = component.guardClass.className
      ..implements.add(refer((Guard).name))
      ..constructors.add(
        Constructor(
          (p) => p
            ..constant = true
            ..requiredParameters.add(
              Parameter(
                (p) => p
                  ..name = 'di'
                  ..toThis = true
                  ..named = false,
              ),
            )
            ..optionalParameters.addAll(parameters),
        ),
      )
      ..types.addAll(generics)
      ..fields.add(
        Field(
          (p) => p
            ..type = refer('DI')
            ..name = 'di'
            ..modifier = FieldModifier.final$,
        ),
      )
      ..fields.addAll(fields)
      ..methods.add(
        Method(
          (p) => p
            ..name = 'protect'
            ..returns = refer('Future<${(GuardResult).name}>')
            ..modifier = MethodModifier.async
            ..annotations.add(refer('override'))
            ..requiredParameters.add(
              Parameter(
                (p) => p
                  ..name = 'context'
                  ..type = refer((Context).name),
              ),
            )
            ..body = Block.of([
              declareFinal('component')
                  .assign(refer(component.name).newInstance(positioned, named))
                  .statement,
              const Code('\n'),
              declareFinal('guards')
                  .assign(
                    literalList([
                      ...createComponentMethods(component.guards),
                    ], refer('FutureOr<${(GuardResult).name}> Function()')),
                  )
                  .statement,
              const Code('\n'),
              forInLoop(
                declaration: declareFinal('guard'),
                iterable: refer('guards'),
                body: Block.of([
                  declareFinal('result')
                      .assign(
                        createSwitchPattern(refer('guard').call([]), {
                          declareFinal(
                            'future',
                            type: refer('Future<${(GuardResult).name}>'),
                          ): refer(
                            'future',
                          ),
                          declareFinal(
                            'result',
                            type: refer((GuardResult).name),
                          ): refer(
                            'Future',
                          ).property('value').call([refer('result')]),
                        }).awaited,
                      )
                      .statement,
                  const Code('\n'),
                  ifStatement(
                    refer('result').property('isBlock'),
                    body: refer('result').returned.statement,
                  ).code,
                ]),
              ).code,
              const Code('\n'),
              declareConst(
                (GuardResult).name,
              ).property('pass').call([]).returned.statement,
            ]),
        ),
      ),
  );

  return formatter(clazz);
}