File Coverage

blib/lib/File/Inplace.pm

Criterion	Covered	Total	%
statement	112	112	100.0
branch	34	44	77.2
condition	7	8	87.5
subroutine	23	23	100.0
pod	9	10	90.0
total	185	197	93.9

package File::Inplace;
use strict;

use Carp qw/carp croak/;
use File::Basename qw/dirname/;
use File::Temp qw/tempfile/;
use File::Copy;
use IO::File;
use IO::Handle;

our $VERSION = '0.20';

my @allowed_options = qw/chomp regex separator suffix file/;
my %allowed_options = map { $_ => 1 } @allowed_options;

sub new {
  my $class = shift;
  my %params = @_;

  for my $opt (keys %params) {
    croak "Invalid constructor option '$opt'" unless exists $allowed_options{$opt};
  }
  croak "Required parameter 'file' not specified in constructor"
    unless exists $params{file};

  my $self = bless \%params, $class;

  $params{chomp} = 1 unless exists $params{chomp};
  $params{regex} = $params{regex} || $params{separator} || qr/\s+/;
  $params{separator} ||= ' ';

  if ($self->{suffix}) {
    $self->{backup_name} = $self->{file} . $self->{suffix};
    copy($self->{file} => $self->{backup_name})
      or croak "error creating backup: $!";
  }

  $self->_open_input_file;
  $self->_open_output_file;

  $self->{current_line} = undef;

  return $self;
}

sub has_lines {
  my $self = shift;

  return 1 if not $self->{infh}->eof();
  return 0;
}

sub next_line {
  my $self = shift;

  $self->_write_current_line;

  $self->{current_line} = $self->_read_next_line();

  if (wantarray) {
    if (defined $self->{current_line}) {
      return ($self->{current_line});
    }
    else {
      return ();
    }
  }

  return $self->{current_line};
}

sub next_line_split {
  my $self = shift;

  my $line = $self->next_line;

  return split $self->{regex}, $line;
}

sub all_lines {
  my $self = shift;

  croak "cannot use all_lines after any lines have been read"
    if defined $self->{current_line};

  my @ret;
  while (1) {
    my $line = $self->_read_next_line;
    last unless defined $line;
    push @ret, $line;
  }

  return @ret;
}

sub replace_line {
  my $self = shift;

  if (@_ == 1) {
    $self->{current_line} = shift;
  }
  else {
    $self->{current_line} = join($self->{separator}, @_);
  }
}

sub replace_lines {
  my $self = shift;
  my @lines = @_;

  my $fh = $self->{outfh};
  for my $line (@lines) {
    $fh->print($line);
    if ($self->{chomp}) {
      $fh->print($/);
    }
  }
}

sub _open_input_file {
  my $self = shift;

  $self->{infh} = new IO::File("<$self->{file}");
  croak "open $self->{file}: $!" if not $self->{infh};
}

sub _open_output_file {
  my $self = shift;

  my $dir = dirname $self->{file};
  my ($tmpfh, $tmpname) = tempfile(DIR => $dir);
  $self->{outfh} = bless $tmpfh, "IO::Handle";
  $self->{tmpfile} = $tmpname;
}

sub _write_current_line {
  my $self = shift;

  my $fh = $self->{outfh};
  if (defined $self->{current_line}) {
    $fh->print($self->{current_line});
    if ($self->{chomp}) {
      $fh->print($/);
    }
  }
}

sub _read_next_line {
  my $self = shift;

  my $fh = $self->{infh};
  return undef unless $fh;
  my $line = $fh->getline;
  if (not defined $line) {
    $fh->close;
    delete $self->{infh};
  }

  if (defined $line and $self->{chomp}) {
    chomp $line;
  }

  return $line;
}

sub commit {
  my $self = shift;

  $self->_write_current_line;

  rename $self->{tmpfile} => $self->{file}
    or croak "Can't rename $self->{tmpname} => $self->{file}: $!";

  $self->_close_all();
}

sub commit_to_backup {
  my $self = shift;

  $self->_write_current_line;

  croak "cannot commit_to_backup if no backup file is in use"
    unless $self->{backup_name};

  rename $self->{tmpfile} => $self->{backup_name}
    or croak "Can't rename $self->{tmpname} => $self->{backup_name}: $!";

  $self->_close_all();
}

sub rollback {
  my $self = shift;

  $self->_close_all();
  unlink $self->{tmpfile};
}

sub DESTROY {
  my $self = shift;

  $self->_close_all();
  unlink $self->{tmpfile};
}

sub _close_all {
  my $self = shift;

  for my $handle (qw/infh outfh/) {
    $self->{$handle}->close()
      if $self->{$handle};
  }
}

1;
__END__