<?php

//Version $Revision: 1461 $ $Date: 2011-02-22 09:30:25 +0100 (Tue, 22 Feb 2011) $

/**
 * Realtime Register whois proxy
 * 
 * <code>
 * define('WP_USERNAME', 'username');
 * define('WP_PASSWORD', 'password');
 * define('WP_HOST', 'epp.12register.com');
 * define('WP_PORT', '2001');
 * 
 * require_once 'whoisproxy.php';
 * 
 * $wp = new whoisproxy();
 * $wp->check('domainname', 'com');
 * 
 * $result = $wp->result();
 * echo $result['domain'].' '.$result['result'];
 *
 * $wp->close();
 * </code>
 */

class WhoisProxy {
    /**
     * Connection
     *
     * @var     object
     */
    var $_fp;
    
    /**
     * Construct
     *
     * @return  void
     */
    function WhoisProxy() {
    
    }
    
    /**
     * Close
     * 
     * @return  void
     */
    function close() {
        $this->write('CLOSE');
        
        @fclose($this->_fp);
    }
    
    /**
     * Connect
     *
     * @return  bool    Connection established?
     */
    function connect() {
        $this->_fp = @fsockopen(WP_HOST, WP_PORT, $errno, $errstr, 10);
        if(!is_resource($this->_fp)) {
            return false;
        }
        
        /**
         * Login
         */
        return $this->login();
    }
    
    /**
     * Check
     *
     * @param   string  Domainname
     * @param   mixed   TLD(s)
     * @return  void
     */
    function check($domainname, $tlds) {
        if(!is_array($tlds)) {
            $tlds = (array) $tlds;
        }
        
        foreach($tlds as $tld) {
            $this->write('IS '.$domainname.'.'.$tld);
        }
    }
    
    /**
     * Result
     * 
     * @return  array   [ domain, result ]
     */
    function result() {
        $response = $this->read();
        if(preg_match('#^([\-\w.]+)\s(available|not\savailable|invalid\sdomain|error)#', $response, $match)) {
            return array('domain' => $match[1], 'result' => $match[2]);
        }
        
        return array('domain' => '-', 'result' => 'error');
    }
    
    /**
     * Is connected?
     *
     * @return  bool    Connection?
     */
    function is_connected() {
        return is_resource($this->_fp);
    }
    
    /**
     * Login
     *
     * @return  bool    Login successfull?
     */
    function login() {
        if(!$this->write('LOGIN '.WP_USERNAME.' '.WP_PASSWORD)) {
            return false;
        }
        
        $response = $this->read();
        if(preg_match('#^400\sLogin\sfailed#', $response)) {
            return false;
        }
        
        return preg_match('#^100\sLogin\sok#', $response);
    }
    
    /**
     * Read
     *
     * @return  string  Response
     */
    function read() {
        if(!$this->is_connected()) {
            $this->connect();
        }
        
        if(!$response = fgets($this->_fp, 1024)) {
            return false;
        }
        
        return trim($response);
    }
    
    /**
     * Write
     *
     * @param   string  Message
     * @return  bool    Writing successfull?
     */
    function write($message) {
        if(!$this->is_connected()) {
            $this->connect();
        }
        
        return @fputs($this->_fp, $message."\r\n");
    }
}

?>