Minor cleanups
[squirrelmail.git] / functions / abook_local_file.php
index 8bc02405d6be3df8f7bf851e7501331864e74fbd..04c1eaa2137a80766b2b6741b6b1aea14161ee27 100644 (file)
 <?php
 
-  /**
-   **  abook_local_file.php
-   **
-   **  Backend for addressbook as a pipe separated file
-   **
-   **  An array with the following elements must be passed to
-   **  the class constructor (elements marked ? are optional):
-   **
-   **     filename  => path to addressbook file
-   **   ? create    => if true: file is created if it does not exist.
-   **   ? umask     => umask set before opening file.
-   **
-   **  NOTE. This class should not be used directly. Use the
-   **        "AddressBook" class instead.
-   **
-   ** $Id$
-   **/
-
-   class abook_local_file extends addressbook_backend {
-     var $btype = 'local';
-     var $bname = 'local_file';
-
-     var $filename   = '';
-     var $filehandle = 0;
-     var $create     = false;
-     var $umask;
-
-     // ========================== Private =======================
-
-     // Constructor
-     function abook_local_file($param) {
-       $this->sname = _("Personal address book");
-       $this->umask = Umask();
-
-       if(is_array($param)) {
-        if(empty($param['filename']))
-          return $this->set_error('Invalid parameters');
-        if(!is_string($param['filename']))
-          return $this->set_error($param['filename'] . ': '.
-                                  _("Not a file name"));
-
-        $this->filename = $param['filename'];
-
-        if($param['create'])
-          $this->create = true;
-        if(isset($param['umask'])) 
-          $this->umask = $param['umask'];
-
-        if(!empty($param['name']))
-          $this->sname = $param['name'];
-
-        $this->open(true);
-       } else {
-        $this->set_error('Invalid argument to constructor');
-       }
-     }
-
-     // Open the addressbook file and store the file pointer.
-     // Use $file as the file to open, or the class' own 
-     // filename property. If $param is empty and file is  
-     // open, do nothing.
-     function open($new = false) {
-       $this->error = '';
-       $file   = $this->filename;
-       $create = $this->create;
-
-       // Return true is file is open and $new is unset
-       if($this->filehandle && !$new)
-        return true;
-
-       // Check that new file exitsts
-       if((!(file_exists($file) && is_readable($file))) && !$create)
-        return $this->set_error("$file: " . _("No such file or directory"));
-
-       // Close old file, if any
-       if($this->filehandle) $this->close();
-       
-       // Open file. First try to open for reading and writing,
-       // but fall back to read only.
-       umask($this->umask);
-       $fh = @fopen($file, 'a+');
-       if($fh) {
-        $this->filehandle = &$fh;
-        $this->filename   = $file;
-        $this->writeable  = true;
-       } else {
-        $fh = @fopen($file, 'r');
-        if($fh) {
-          $this->filehandle = &$fh;
-          $this->filename   = $file;
-          $this->writeable  = false;
-        } else {
-          return $this->set_error("$file: " . _("Open failed"));
-        }
-       }
-
-       return true;
-     }
-
-     // Close the file and forget the filehandle
-     function close() {
-       @fclose($this->filehandle);
-       $this->filehandle = 0;
-       $this->filename   = '';
-       $this->writable   = false;
-     }
-
-     // Lock the datafile - try 20 times in 5 seconds
-     function lock() {
-       for($i = 0 ; $i < 20 ; $i++) {
-        if(flock($this->filehandle, 2 + 4)) 
-          return true;
-        else
-          usleep(250000);
-       }
-       return false;
-     }
-
-     // Lock the datafile
-     function unlock() {
-       return flock($this->filehandle, 3);
-     }
-
-     // Overwrite the file with data from $rows
-     // NOTE! Previous locks are broken by this function
-     function overwrite($rows) {
-       $newfh = @fopen($this->filename, 'w');
-       if(!$newfh)
-        return $this->set_error("$file: " . _("Open failed"));
-
-       for($i = 0 ; $i < sizeof($rows) ; $i++) {
-        if(is_array($rows[$i]))
-          fwrite($newfh, join('|', $rows[$i]) . "\n");
-       }       
-
-       fclose($newfh);
-       $this->unlock();
-       $this->open(true);
-       return true;
-     }
-     
-     // ========================== Public ========================
-     
-     // Search the file
-     function search($expr) {
-
-       // To be replaced by advanded search expression parsing
-       if(is_array($expr)) return;
-
-       // Make regexp from glob'ed expression 
-       $expr = ereg_replace('\?', '.', $expr);
-       $expr = ereg_replace('\*', '.*', $expr);
-
-       $res = array();
-       if(!$this->open())
-        return false;
-
-       @rewind($this->filehandle);
-       
-       while ($row = @fgetcsv($this->filehandle, 2048, '|')) {
-        $line = join(' ', $row);
-        if(eregi($expr, $line)) {
-          array_push($res, array('nickname'  => $row[0],
-                                 'name'      => $row[1] . ' ' . $row[2],
-                                 'firstname' => $row[1],
-                                 'lastname'  => $row[2],
-                                 'email'     => $row[3],
-                                 'label'     => $row[4],
-                                 'backend'   => $this->bnum,
-                                 'source'    => &$this->sname));
-        }
-       }
-       
-       return $res;
-     }
-     
-     // Lookup alias
-     function lookup($alias) {
-       if(empty($alias))
-        return array();
-
-       $alias = strtolower($alias);
-       
-       $this->open();
-       @rewind($this->filehandle);
-       
-       while ($row = @fgetcsv($this->filehandle, 2048, '|')) {
-        if(strtolower($row[0]) == $alias) {
-          return array('nickname'  => $row[0],
-                       'name'      => $row[1] . ' ' . $row[2],
-                       'firstname' => $row[1],
-                       'lastname'  => $row[2],
-                       'email'     => $row[3],
-                       'label'     => $row[4],
-                       'backend'   => $this->bnum,
-                       'source'    => &$this->sname);
-        }
-       }
-       
-       return array();
-     }
-
-     // List all addresses
-     function list_addr() {
-       $res = array();
-       $this->open();
-       @rewind($this->filehandle);
-       
-       while ($row = @fgetcsv($this->filehandle, 2048, '|')) {
-        array_push($res, array('nickname'  => $row[0],
-                               'name'      => $row[1] . ' ' . $row[2],
-                               'firstname' => $row[1],
-                               'lastname'  => $row[2],
-                               'email'     => $row[3],
-                               'label'     => $row[4],
-                               'backend'   => $this->bnum,
-                               'source'    => &$this->sname));
-       }
-       return $res;
-     }
-
-     // Add address
-     function add($userdata) {
-       if(!$this->writeable) 
-        return $this->set_error(_("Addressbook is read-only"));
-
-       // See if user exist already
-       $ret = $this->lookup($userdata['nickname']);
-       if(!empty($ret))
-        return $this->set_error(sprintf(_("User '%s' already exist"),
-                                        $ret['nickname']));
-
-       // Here is the data to write
-       $data = $userdata['nickname'] . '|' . $userdata['firstname'] . '|' .
-               $userdata['lastname'] . '|' . $userdata['email'] . '|' .
-               $userdata['label'];
-       // Strip linefeeds
-       $data = ereg_replace("[\r\n]", ' ', $data);
-       // Add linefeed at end
-       $data = $data . "\n";
-
-       // Reopen file, just to be sure
-       $this->open(true);
-       if(!$this->writeable) 
-        return $this->set_error(_("Addressbook is read-only"));
-
-       // Lock the file
-       if(!$this->lock())
-        return $this->set_error(_("Could not lock datafile"));
-
-       // Write
-       $r = fwrite($this->filehandle, $data);
-
-       // Unlock file
-       $this->unlock();
-
-       // Test write result and exit if OK
-       if($r > 0) return true;
-
-       // Fail
-       $this->set_error(_("Write to addressbook failed"));
-       return false;
-     }
-
-     // Delete address
-     function remove($alias) {
-       if(!$this->writeable) 
-        return $this->set_error(_("Addressbook is read-only"));
-
-       // Lock the file to make sure we're the only process working
-       // on it.
-       if(!$this->lock())
-        return $this->set_error(_("Could not lock datafile"));
-
-       // Read file into memory, ignoring nicknames to delete
-       $this->open();
-       @rewind($this->filehandle);
-       $i = 0;
-       $rows = array();
-       while($row = @fgetcsv($this->filehandle, 2048, '|')) {
-        if(!in_array($row[0], $alias))
-          $rows[$i++] = $row;
-       }
-
-       // Write data back
-       if(!$this->overwrite(&$rows)) {
-        $this->unlock();
-        return false;
-       }
-
-       $this->unlock();
-       return true;
-     }
-
-     // Modify address
-     function modify($alias, $userdata) {
-       if(!$this->writeable) 
-        return $this->set_error(_("Addressbook is read-only"));
-
-       // See if user exist
-       $ret = $this->lookup($alias);
-       if(empty($ret))
-        return $this->set_error(sprintf(_("User '%s' does not exist"),
-                                        $alias));
-
-       // Lock the file to make sure we're the only process working
-       // on it.
-       if(!$this->lock())
-        return $this->set_error(_("Could not lock datafile"));
-
-       // Read file into memory, modifying the data for the 
-       // user identifyed by $alias
-       $this->open();
-       @rewind($this->filehandle);
-       $i = 0;
-       $rows = array();
-       while($row = @fgetcsv($this->filehandle, 2048, '|')) {
-        if(strtolower($row[0]) != strtolower($alias)) {
-          $rows[$i++] = $row;
-        } else {
-          $rows[$i++] = array(0 => $userdata['nickname'],
-                              1 => $userdata['firstname'],
-                              2 => $userdata['lastname'],
-                              3 => $userdata['email'], 
-                              4 => $userdata['label']);
-        }
-       }
-
-       // Write data back
-       if(!$this->overwrite(&$rows)) {
-        $this->unlock();
-        return false;
-       }
-
-       $this->unlock();
-       return true;
-     }
-     
-   } // End of class abook_local_file
-?>
+/**
+ * abook_local_file.php
+ *
+ * Copyright (c) 1999-2004 The SquirrelMail Project Team
+ * Licensed under the GNU GPL. For full terms see the file COPYING.
+ *
+ * Backend for addressbook as a pipe separated file
+ *
+ * An array with the following elements must be passed to
+ * the class constructor (elements marked ? are optional):
+ *
+ *    filename  => path to addressbook file
+ *  ? create    => if true: file is created if it does not exist.
+ *  ? umask     => umask set before opening file.
+ *
+ * NOTE. This class should not be used directly. Use the
+ *       "AddressBook" class instead.
+ *
+ * @version $Id$
+ * @package squirrelmail
+ * @subpackage addressbook
+ */
+
+/**
+ * Store the addressbook in a local file
+ * @package squirrelmail
+ */
+class abook_local_file extends addressbook_backend {
+    var $btype = 'local';
+    var $bname = 'local_file';
+
+    var $filename   = '';
+    var $filehandle = 0;
+    var $create     = false;
+    var $umask;
+
+    /* ========================== Private ======================= */
+
+    /* Constructor */
+    function abook_local_file($param) {
+        $this->sname = _("Personal address book");
+        $this->umask = Umask();
+
+        if(is_array($param)) {
+            if(empty($param['filename'])) {
+                return $this->set_error('Invalid parameters');
+            }
+            if(!is_string($param['filename'])) {
+                return $this->set_error($param['filename'] . ': '.
+                     _("Not a file name"));
+            }
+
+            $this->filename = $param['filename'];
+
+            if($param['create']) {
+                $this->create = true;
+            }
+            if(isset($param['umask'])) {
+                $this->umask = $param['umask'];
+            }
+            if(!empty($param['name'])) {
+                $this->sname = $param['name'];
+            }
+
+            $this->open(true);
+        } else {
+            $this->set_error('Invalid argument to constructor');
+        }
+    }
+
+    /* Open the addressbook file and store the file pointer.
+     * Use $file as the file to open, or the class' own
+     * filename property. If $param is empty and file is
+     * open, do nothing. */
+    function open($new = false) {
+        $this->error = '';
+        $file   = $this->filename;
+        $create = $this->create;
+
+        /* Return true is file is open and $new is unset */
+        if($this->filehandle && !$new) {
+            return true;
+        }
+
+        /* Check that new file exitsts */
+        if((!(file_exists($file) && is_readable($file))) && !$create) {
+            return $this->set_error("$file: " . _("No such file or directory"));
+        }
+
+        /* Close old file, if any */
+        if($this->filehandle) { $this->close(); }
+
+        /* Open file. First try to open for reading and writing,
+         * but fall back to read only. */
+        umask($this->umask);
+        $fh = @fopen($file, 'a+');
+        if($fh) {
+            $this->filehandle = &$fh;
+            $this->filename   = $file;
+            $this->writeable  = true;
+        } else {
+            $fh = @fopen($file, 'r');
+            if($fh) {
+                $this->filehandle = &$fh;
+                $this->filename   = $file;
+                $this->writeable  = false;
+            } else {
+                return $this->set_error("$file: " . _("Open failed"));
+            }
+        }
+        return true;
+    }
+
+    /* Close the file and forget the filehandle */
+    function close() {
+        @fclose($this->filehandle);
+        $this->filehandle = 0;
+        $this->filename   = '';
+        $this->writable   = false;
+    }
+
+    /* Lock the datafile - try 20 times in 5 seconds */
+    function lock() {
+        for($i = 0 ; $i < 20 ; $i++) {
+            if(flock($this->filehandle, 2 + 4))
+                return true;
+            else
+                usleep(250000);
+        }
+        return false;
+    }
+
+    /* Lock the datafile */
+    function unlock() {
+        return flock($this->filehandle, 3);
+    }
+
+    /* Overwrite the file with data from $rows
+     * NOTE! Previous locks are broken by this function */
+    function overwrite(&$rows) {
+        $this->unlock();
+        $newfh = @fopen($this->filename.'.tmp', 'w');
+
+        if(!$newfh) {
+            return $this->set_error($this->filename. '.tmp:' . _("Open failed"));
+        }
+
+        for($i = 0, $cnt=sizeof($rows) ; $i < $cnt ; $i++) {
+            if(is_array($rows[$i])) {
+                for($j = 0, $cnt_part=count($rows[$i]) ; $j < $cnt_part ; $j++) {
+                    $rows[$i][$j] = $this->quotevalue($rows[$i][$j]);
+                }
+                $tmpwrite = sq_fwrite($newfh, join('|', $rows[$i]) . "\n");
+                if ($tmpwrite === FALSE) {
+                    return $this->set_error($this->filename . '.tmp:' . _("Write failed"));
+                }
+            }
+        }
+
+        fclose($newfh);
+        if (!@copy($this->filename . '.tmp' , $this->filename)) {
+          return $this->set_error($this->filename . ':' . _("Unable to update"));
+        }
+        @unlink($this->filename . '.tmp');
+        $this->unlock();
+        $this->open(true);
+        return true;
+    }
+
+    /* ========================== Public ======================== */
+
+    /* Search the file */
+    function search($expr) {
+
+        /* To be replaced by advanded search expression parsing */
+        if(is_array($expr)) { return; }
+
+        /* Make regexp from glob'ed expression
+         * May want to quote other special characters like (, ), -, [, ], etc. */
+        $expr = str_replace('?', '.', $expr);
+        $expr = str_replace('*', '.*', $expr);
+
+        $res = array();
+        if(!$this->open()) {
+            return false;
+        }
+        @rewind($this->filehandle);
+
+        while ($row = @fgetcsv($this->filehandle, 2048, '|')) {
+            $line = join(' ', $row);
+            if(eregi($expr, $line)) {
+                array_push($res, array('nickname'  => $row[0],
+                    'name'      => $row[1] . ' ' . $row[2],
+                    'firstname' => $row[1],
+                    'lastname'  => $row[2],
+                    'email'     => $row[3],
+                    'label'     => $row[4],
+                    'backend'   => $this->bnum,
+                    'source'    => &$this->sname));
+            }
+        }
+
+        return $res;
+    }
+
+    /* Lookup alias */
+    function lookup($alias) {
+        if(empty($alias)) {
+            return array();
+        }
+
+        $alias = strtolower($alias);
+
+        $this->open();
+        @rewind($this->filehandle);
+
+        while ($row = @fgetcsv($this->filehandle, 2048, '|')) {
+            if(strtolower($row[0]) == $alias) {
+                return array('nickname'  => $row[0],
+                  'name'      => $row[1] . ' ' . $row[2],
+                  'firstname' => $row[1],
+                  'lastname'  => $row[2],
+                  'email'     => $row[3],
+                  'label'     => $row[4],
+                  'backend'   => $this->bnum,
+                  'source'    => &$this->sname);
+            }
+        }
+
+        return array();
+    }
+
+    /* List all addresses */
+    function list_addr() {
+        $res = array();
+        $this->open();
+        @rewind($this->filehandle);
+
+        while ($row = @fgetcsv($this->filehandle, 2048, '|')) {
+            array_push($res, array('nickname'  => $row[0],
+                'name'      => $row[1] . ' ' . $row[2],
+                'firstname' => $row[1],
+                'lastname'  => $row[2],
+                'email'     => $row[3],
+                'label'     => $row[4],
+                'backend'   => $this->bnum,
+                'source'    => &$this->sname));
+        }
+        return $res;
+    }
+
+    /* Add address */
+    function add($userdata) {
+        if(!$this->writeable) {
+            return $this->set_error(_("Addressbook is read-only"));
+        }
+        /* See if user exists already */
+        $ret = $this->lookup($userdata['nickname']);
+        if(!empty($ret)) {
+            return $this->set_error(sprintf(_("User '%s' already exist"),
+                   $ret['nickname']));
+        }
+
+        /* Here is the data to write */
+        $data = $this->quotevalue($userdata['nickname']) . '|' .
+                $this->quotevalue($userdata['firstname']) . '|' .
+                $this->quotevalue($userdata['lastname']) . '|' .
+                $this->quotevalue($userdata['email']) . '|' .
+                $this->quotevalue($userdata['label']);
+
+        /* Strip linefeeds */
+        $data = ereg_replace("[\r\n]", ' ', $data);
+        /* Add linefeed at end */
+        $data = $data . "\n";
+
+        /* Reopen file, just to be sure */
+        $this->open(true);
+        if(!$this->writeable) {
+            return $this->set_error(_("Addressbook is read-only"));
+        }
+
+        /* Lock the file */
+        if(!$this->lock()) {
+            return $this->set_error(_("Could not lock datafile"));
+        }
+
+        /* Write */
+        $r = sq_fwrite($this->filehandle, $data);
+
+        /* Unlock file */
+        $this->unlock();
+
+        /* Test write result */
+        if($r === FALSE) {
+               /* Fail */
+               $this->set_error(_("Write to addressbook failed"));
+               return FALSE;
+       }
+
+        return TRUE;
+    }
+
+    /* Delete address */
+    function remove($alias) {
+        if(!$this->writeable) {
+            return $this->set_error(_("Addressbook is read-only"));
+        }
+
+        /* Lock the file to make sure we're the only process working
+         * on it. */
+        if(!$this->lock()) {
+            return $this->set_error(_("Could not lock datafile"));
+        }
+
+        /* Read file into memory, ignoring nicknames to delete */
+        @rewind($this->filehandle);
+        $i = 0;
+        $rows = array();
+        while($row = @fgetcsv($this->filehandle, 2048, '|')) {
+            if(!in_array($row[0], $alias)) {
+                $rows[$i++] = $row;
+            }
+        }
+
+        /* Write data back */
+        if(!$this->overwrite($rows)) {
+            $this->unlock();
+            return false;
+        }
+
+        $this->unlock();
+        return true;
+    }
+
+    /* Modify address */
+    function modify($alias, $userdata) {
+        if(!$this->writeable) {
+            return $this->set_error(_("Addressbook is read-only"));
+        }
+
+        /* See if user exists */
+        $ret = $this->lookup($alias);
+        if(empty($ret)) {
+            return $this->set_error(sprintf(_("User '%s' does not exist"),
+                $alias));
+        }
+
+        /* Lock the file to make sure we're the only process working
+         * on it. */
+        if(!$this->lock()) {
+            return $this->set_error(_("Could not lock datafile"));
+        }
+
+        /* Read file into memory, modifying the data for the
+         * user identified by $alias */
+        $this->open(true);
+        @rewind($this->filehandle);
+        $i = 0;
+        $rows = array();
+        while($row = @fgetcsv($this->filehandle, 2048, '|')) {
+            if(strtolower($row[0]) != strtolower($alias)) {
+                $rows[$i++] = $row;
+            } else {
+                $rows[$i++] = array(0 => $userdata['nickname'],
+                                    1 => $userdata['firstname'],
+                                    2 => $userdata['lastname'],
+                                    3 => $userdata['email'],
+                                    4 => $userdata['label']);
+            }
+        }
+
+        /* Write data back */
+        if(!$this->overwrite($rows)) {
+            $this->unlock();
+            return false;
+        }
+
+        $this->unlock();
+        return true;
+    }
+
+    /* Function for quoting values before saving */
+    function quotevalue($value) {
+        /* Quote the field if it contains | or ". Double quotes need to
+         * be replaced with "" */
+        if(ereg("[|\"]", $value)) {
+            $value = '"' . str_replace('"', '""', $value) . '"';
+        }
+        return $value;
+    }
+
+} /* End of class abook_local_file */
+?>
\ No newline at end of file