lib/Method/Generate/Constructor.pm

   1 package Method::Generate::Constructor;
   2
   3 use strictures 1;
   4 use Sub::Quote;
   5 use base qw(Class::Tiny::Object);
   6 use Sub::Defer;
   7 use B 'perlstring';
   8
   9 sub register_attribute_specs {
  10   my ($self, %spec) = @_;
  11   @{$self->{attribute_specs}||={}}{keys %spec} = values %spec;
  12   $self;
  13 }
  14
  15 sub all_attribute_specs {
  16   $_[0]->{attribute_specs}
  17 }
  18
  19 sub accessor_generator {
  20   $_[0]->{accessor_generator}
  21 }
  22
  23 sub install_delayed {
  24   my ($self) = @_;
  25   my $package = $self->{package};
  26   defer_sub "${package}::new" => sub {
  27     unquote_sub $self->generate_method(
  28       $package, 'new', $self->{attribute_specs}, { no_install => 1 }
  29     )
  30   };
  31   $self;
  32 }
  33
  34 sub generate_method {
  35   my ($self, $into, $name, $spec, $quote_opts) = @_;
  36   foreach my $no_init (grep !exists($spec->{$_}{init_arg}), keys %$spec) {
  37     $spec->{$no_init}{init_arg} = $no_init;
  38   }
  39   local $self->{captures} = {};
  40   my $body = '    my $class = shift;'."\n";
  41   $body .= $self->_generate_args;
  42   $body .= $self->_check_required($spec);
  43   $body .= $self->_check_isa($spec);
  44   $body .= '    my $new = bless({}, $class);'."\n";
  45   $body .= $self->_assign_new($spec);
  46   $body .= $self->_fire_triggers($spec);
  47   $body .= '    return $new;'."\n";
  48   quote_sub
  49     "${into}::${name}" => $body,
  50     $self->{captures}, $quote_opts||{}
  51   ;
  52 }
  53
  54 sub _generate_args {
  55   my ($self) = @_;
  56   q{    my $args = ref($_[0]) eq 'HASH' ? $_[0] : { @_ };}."\n";
  57 }
  58
  59 sub _assign_new {
  60   my ($self, $spec) = @_;
  61   my (@init, @slots);
  62   NAME: foreach my $name (keys %$spec) {
  63     my $attr_spec = $spec->{$name};
  64     push @init, do {
  65       next NAME unless defined(my $i = $attr_spec->{init_arg});
  66       $i;
  67     };
  68     push @slots, $name;
  69   }
  70   return '' unless @init;
  71   '    @{$new}{qw('.join(' ',@slots).')} = @{$args}{qw('.join(' ',@init).')};'
  72     ."\n";
  73 }
  74
  75 sub _check_required {
  76   my ($self, $spec) = @_;
  77   my @required_init =
  78     map $spec->{$_}{init_arg},
  79       grep $spec->{$_}{required},
  80         sort keys %$spec;
  81   return '' unless @required_init;
  82   '    if (my @missing = grep !exists $args->{$_}, qw('
  83     .join(' ',@required_init).')) {'."\n"
  84     .q{      die "Missing required arguments: ".join(', ', sort @missing);}."\n"
  85     ."    }\n";
  86 }
  87
  88 sub _check_isa {
  89   my ($self, $spec) = @_;
  90   my $acc = $self->accessor_generator;
  91   my $captures = $self->{captures};
  92   my $check = '';
  93   foreach my $name (sort keys %$spec) {
  94     my ($init, $isa) = @{$spec->{$name}}{qw(init_arg isa)};
  95     next unless $init and $isa;
  96     my $init_str = perlstring($init);
  97     my ($code, $add_captures) = $acc->generate_isa_check(
  98       $name, "\$args->{${init_str}}", $isa
  99     );
 100     @{$captures}{keys %$add_captures} = values %$add_captures;
 101     $check .= "    ${code} if exists \$args->{${init_str}};\n";
 102   }
 103   return $check;
 104 }
 105
 106 sub _fire_triggers {
 107   my ($self, $spec) = @_;
 108   my $acc = $self->accessor_generator;
 109   my $captures = $self->{captures};
 110   my $fire = '';
 111   foreach my $name (sort keys %$spec) {
 112     my ($init, $trigger) = @{$spec->{$name}}{qw(init_arg trigger)};
 113     next unless $init && $trigger;
 114     my ($code, $add_captures) = $acc->generate_trigger(
 115       $name, '$new', $acc->generate_simple_get('$new', $name), $trigger
 116     );
 117     @{$captures}{keys %$add_captures} = values %$add_captures;
 118     $fire .= "    ${code} if exists \$args->{${\perlstring $init}};\n";
 119   }
 120   return $fire;
 121 }
 122
 123 1;