source: branches/stable/cgi-bin/CustIDCK.pm@ 191

# ipdb/cgi-bin/CustIDCK.pm
# External Customer ID checker stub
###
# SVN revision info
# $Date: 2005-02-07 17:00:47 +0000 (Mon, 07 Feb 2005) $
# SVN revision $Rev: 156 $
# Last update by $Author: kdeugau $
###

package CustIDCK;

use strict;
use warnings;
use Exporter;
use vars qw($VERSION @ISA @EXPORT @EXPORT_OK %EXPORT_TAGS);
use DBI;

$VERSION        = 1.00;
@ISA            = qw(Exporter);
@EXPORT         = ();
@EXPORT_OK      = qw ( &custid_exist );

# this is really an example stub, and should be replaced by
# the local admin on installation
sub custid_exist {
  my $custid = shift;

  return 1 if $custid =~ /^STAFF$/;
  return 1 if $custid =~ /^6750400$/;  # just in case some later change might block this
  return 1 if $custid =~ /^\d{7}$/;
  return 1 if $custid =~ /^\d{10}$/;

# some example code for a database check
  # Try to catch failures to connect.  If the remote server is up but
  # not responding (this has HAPPENED) we need to break out rather than hanging.
  my $dbh;
  eval {
    local $SIG{ALRM} = sub { die "failed connection to newbilling!!" };
    alarm 3;    # 3-second timeout.  This may be too aggressive.

    eval {
      $dbh = DBI->connect ("DBI:Pg:host=newbilling;dbname=custids", "cidcheck", "c1dch4ck");
      die "failed connection to newbilling!!" if !$dbh;
# Not certain if this is needed here.  It doesn't seem to be.
#      $dbh->ping;      # Gotta do this to "force" a "failure".  NRGH.
    };
    alarm 0;    # cancel the alarm
    $dbh->ping; # Gotta do this to "force" a "failure".  NRGH.
  };
  alarm 0;      # avoid race conditions.  May not be needed here.  (Hah!)
  if ($@ && $@ !~ /failed connection to newbilling!!/) {
    $CustIDCK::Error = 1;
    $CustIDCK::ErrMsg = "Failed connection to newbilling DB host!  Unable to verify CustIDs.";
    return 0;
  }

  # We should have a valid DB connection by now.
  my $sth = $dbh->prepare("SELECT custid FROM custid WHERE custid = '$custid'");
  $sth->execute;
  if ($dbh->err) {
    $CustIDCK::Error = 1;
    $CustIDCK::ErrMsg = $dbh->errstr();
    $sth->finish;
    $dbh->disconnect;
    return 0;
  }
  my $hr = $sth->fetchrow_hashref();
  my $status = 0;
  $status = 1 if ( $hr->{custid} );
  $sth->finish;
  $dbh->disconnect;
  return $status;

  return 0;
  # Stubs for error messages
  $CustIDCK::Error = 1 if 1 == 0;
  $CustIDCK::ErrMsg = "bad foo-frob: 1 == 0";
}