<?php
class BookmarkService {
  var $db;

  function &getInstance(&$db) {
    static $instance;
    if (!isset($instance)) {
      $instance = new BookmarkService($db);
    }
    return $instance;
  }

  function BookmarkService(&$db) {
    $this->db =& $db;
  }

  function _getbookmark($fieldname, $value, $all = false) {
      if (!$all) {
          $userservice = & ServiceFactory :: getServiceInstance('UserService');
          $sId = $userservice->getCurrentUserId();
          $range = ' AND uId = '. $sId;
      }

      $query = 'SELECT * FROM '. $GLOBALS['tableprefix'] .'bookmarks WHERE '. $fieldname .' = "'. $this->db->sql_escape($value) .'"'. $range;

      if (!($dbresult = & $this->db->sql_query_limit($query, 1, 0))) {
          message_die(GENERAL_ERROR, 'Could not get bookmark', '', __LINE__, __FILE__, $query, $this->db);
          return false;
      }

      if ($row =& $this->db->sql_fetchrow($dbresult)) {
          return $row;
      } else {
          return false;
      }
  }

  function & getBookmark($bid, $include_tags = false) {
      if (!is_numeric($bid))
          return;

      $sql = 'SELECT * FROM '. $GLOBALS['tableprefix'] .'bookmarks WHERE bId = '. $this->db->sql_escape($bid);

      if (!($dbresult = & $this->db->sql_query($sql)))
          message_die(GENERAL_ERROR, 'Could not get vars', '', __LINE__, __FILE__, $sql, $this->db);

      if ($row = & $this->db->sql_fetchrow($dbresult)) {
          if ($include_tags) {
              $tagservice = & ServiceFactory :: getServiceInstance('TagService');
              $row['tags'] = $tagservice->getTagsForBookmark($bid);
          }
          return $row;
      } else {
          return false;
      }
  }

  function getBookmarkByAddress($address) {
      $hash = md5($address);
      return $this->getBookmarkByHash($hash);
  }

  function getBookmarkByHash($hash) {
      return $this->_getbookmark('bHash', $hash, true);
  }

  function editAllowed($bookmark) {
      if (!is_numeric($bookmark) && (!is_array($bookmark) || !is_numeric($bookmark['bId'])))
          return false;

      if (!is_array($bookmark))
          if (!($bookmark = $this->getBookmark($bookmark)))
              return false;

      $userservice = & ServiceFactory :: getServiceInstance('UserService');
      $userid = $userservice->getCurrentUserId();
      if ($userservice->isAdmin($userid))
          return true;
      else
          return ($bookmark['uId'] == $userid);
  }

  function bookmarkExists($address = false, $uid = NULL) {
      if (!$address) {
          return;
      }

      // If address doesn't contain ":", add "http://" as the default protocol
      if (strpos($address, ':') === false) {
          $address = 'http://'. $address;
      }

      $crit = array ('bHash' => md5($address));
      if (isset ($uid)) {
          $crit['uId'] = $uid;
      }

      $sql = 'SELECT COUNT(*) FROM '. $GLOBALS['tableprefix'] .'bookmarks WHERE '. $this->db->sql_build_array('SELECT', $crit);
      if (!($dbresult = & $this->db->sql_query($sql))) {
          message_die(GENERAL_ERROR, 'Could not get vars', '', __LINE__, __FILE__, $sql, $this->db);
      }
      return ($this->db->sql_fetchfield(0, 0) > 0);
  }

  // Adds a bookmark to the database.
  // Note that date is expected to be a string that's interpretable by strtotime().
  function addBookmark($address, $title, $description, $status, $categories, $date = NULL, $fromApi = false, $fromImport = false) {
      $userservice = & ServiceFactory :: getServiceInstance('UserService');
      $sId = $userservice->getCurrentUserId();

      // If bookmark address doesn't contain ":", add "http://" to the start as a default protocol
      if (strpos($address, ':') === false) {
          $address = 'http://'. $address;
      }

      // Get the client's IP address and the date; note that the date is in GMT.
      if (getenv('HTTP_CLIENT_IP'))
          $ip = getenv('HTTP_CLIENT_IP');
      else
          if (getenv('REMOTE_ADDR'))
              $ip = getenv('REMOTE_ADDR');
          else
              $ip = getenv('HTTP_X_FORWARDED_FOR');

      // Note that if date is NULL, then it's added with a date and time of now, and if it's present,
      // it's expected to be a string that's interpretable by strtotime().
      if (is_null($date))
          $time = time();
      else
          $time = strtotime($date);
      $datetime = gmdate('Y-m-d H:i:s', $time);

      // Set up the SQL insert statement and execute it.
      $values = array('uId' => intval($sId), 'bIp' => $ip, 'bDatetime' => $datetime, 'bModified' => $datetime, 'bTitle' => $title, 'bAddress' => $address, 'bDescription' => $description, 'bStatus' => intval($status), 'bHash' => md5($address));
      $sql = 'INSERT INTO '. $GLOBALS['tableprefix'] .'bookmarks '. $this->db->sql_build_array('INSERT', $values);
      $this->db->sql_transaction('begin');
      if (!($dbresult = & $this->db->sql_query($sql))) {
          $this->db->sql_transaction('rollback');
          message_die(GENERAL_ERROR, 'Could not insert bookmark', '', __LINE__, __FILE__, $sql, $this->db);
          return false;
      }
      // Get the resultant row ID for the bookmark.
      $bId = $this->db->sql_nextid($dbresult);
      if (!isset($bId) || !is_int($bId)) {
          $this->db->sql_transaction('rollback');
          message_die(GENERAL_ERROR, 'Could not insert bookmark', '', __LINE__, __FILE__, $sql, $this->db);
          return false;
      }

      $uriparts = explode('.', $address);
      $extension = end($uriparts);
      unset($uriparts);

      $tagservice = & ServiceFactory :: getServiceInstance('TagService');
      if (!$tagservice->attachTags($bId, $categories, $fromApi, $extension, false, $fromImport)) {
          $this->db->sql_transaction('rollback');
          message_die(GENERAL_ERROR, 'Could not insert bookmark', '', __LINE__, __FILE__, $sql, $this->db);
          return false;
      }
      $this->db->sql_transaction('commit');
      // Everything worked out, so return the new bookmark's bId.
      return $bId;
  }

  function updateBookmark($bId, $address, $title, $description, $status, $categories, $date = NULL, $fromApi = false) {
      if (!is_numeric($bId))
          return false;

      // Get the client's IP address and the date; note that the date is in GMT.
      if (getenv('HTTP_CLIENT_IP'))
          $ip = getenv('HTTP_CLIENT_IP');
      else
          if (getenv('REMOTE_ADDR'))
              $ip = getenv('REMOTE_ADDR');
          else
              $ip = getenv('HTTP_X_FORWARDED_FOR');

      $moddatetime = gmdate('Y-m-d H:i:s', time());

      // Set up the SQL update statement and execute it.
      $updates = array('bModified' => $moddatetime, 'bTitle' => $title, 'bAddress' => $address, 'bDescription' => $description, 'bStatus' => $status, 'bHash' => md5($address));

      if (!is_null($date)) {
          $updates['bDateTime'] = gmdate('Y-m-d H:i:s', strtotime($date));
      }

      $sql = 'UPDATE '. $GLOBALS['tableprefix'] .'bookmarks SET '. $this->db->sql_build_array('UPDATE', $updates) .' WHERE bId = '. intval($bId);
      $this->db->sql_transaction('begin');

      if (!($dbresult = & $this->db->sql_query($sql))) {
          $this->db->sql_transaction('rollback');
          message_die(GENERAL_ERROR, 'Could not update bookmark', '', __LINE__, __FILE__, $sql, $this->db);
          return false;
      }

      $uriparts = explode('.', $address);
      $extension = end($uriparts);
      unset($uriparts);

      $tagservice = & ServiceFactory :: getServiceInstance('TagService');
      if (!$tagservice->attachTags($bId, $categories, $fromApi, $extension)) {
          $this->db->sql_transaction('rollback');
          message_die(GENERAL_ERROR, 'Could not update bookmark', '', __LINE__, __FILE__, $sql, $this->db);
          return false;
      }

      $this->db->sql_transaction('commit');
      // Everything worked out, so return true.
      return true;
  }

  function & getBookmarks($start = 0, $perpage = NULL, $user = NULL, $tags = NULL, $terms = NULL, $sortOrder = NULL, $watched = NULL, $startdate = NULL, $enddate = NULL, $hash = NULL) {
      // Only get the bookmarks that are visible to the current user.  Our rules:
      //  - if the $user is NULL, that means get bookmarks from ALL users, so we need to make
      //    sure to check the logged-in user's watchlist and get the contacts-only bookmarks from
      //    those users. If the user isn't logged-in, just get the public bookmarks.
      //  - if the $user is set and isn't the logged-in user, then get that user's bookmarks, and
      //    if that user is on the logged-in user's watchlist, get the public AND contacts-only
      //    bookmarks; otherwise, just get the public bookmarks.
      //  - if the $user is set and IS the logged-in user, then get all bookmarks.
      $userservice =& ServiceFactory::getServiceInstance('UserService');
      $tagservice =& ServiceFactory::getServiceInstance('TagService');
      $sId = $userservice->getCurrentUserId();

      if ($userservice->isLoggedOn()) {
          // All public bookmarks, user's own bookmarks and any shared with user
          $privacy = ' AND ((B.bStatus = 0) OR (B.uId = '. $sId .')';
          $watchnames = $userservice->getWatchNames($sId, true);
          foreach($watchnames as $watchuser) {
              $privacy .= ' OR (U.username = "'. $watchuser .'" AND B.bStatus = 1)'; 
          }
          $privacy .= ')';
      } else {
          // Just public bookmarks
          $privacy = ' AND B.bStatus = 0';
      }

      // Set up the tags, if need be.
      if (!is_array($tags) && !is_null($tags)) {
          $tags = explode('+', trim($tags));
      }

      $tagcount = count($tags);
      for ($i = 0; $i < $tagcount; $i ++) {
          $tags[$i] = trim($tags[$i]);
      }

      // Set up the SQL query.
      $query_1 = 'SELECT DISTINCT ';
      if (SQL_LAYER == 'mysql4') {
          $query_1 .= 'SQL_CALC_FOUND_ROWS ';
      }
      $query_1 .= 'B.*, U.'. $userservice->getFieldName('username');

      $query_2 = ' FROM '. $userservice->getTableName() .' AS U, '. $GLOBALS['tableprefix'] .'bookmarks AS B';

      $query_3 = ' WHERE B.uId = U.'. $userservice->getFieldName('primary') . $privacy;
      if (is_null($watched)) {
          if (!is_null($user)) {
              $query_3 .= ' AND B.uId = '. $user;
          }
      } else {
          $arrWatch = $userservice->getWatchlist($user);
          if (count($arrWatch) > 0) {
              foreach($arrWatch as $row) {
                  $query_3_1 .= 'B.uId = '. intval($row) .' OR ';
              }
              $query_3_1 = substr($query_3_1, 0, -3);
          } else {
              $query_3_1 = 'B.uId = -1';
          }
          $query_3 .= ' AND ('. $query_3_1 .') AND B.bStatus IN (0, 1)';
      }

      switch($sortOrder) {
          case 'date_asc':
              $query_5 = ' ORDER BY B.bDatetime ASC ';
              break;
          case 'mod_date_desc':
              $query_5 = ' ORDER BY B.bModified DESC ';
              break;
          case 'mod_date_asc':
              $query_5 = ' ORDER BY B.bModified ASC ';
              break;
          case 'title_desc':
              $query_5 = ' ORDER BY B.bTitle DESC ';
              break;
          case 'title_asc':
              $query_5 = ' ORDER BY B.bTitle ASC ';
              break;
          case 'url_desc':
              $query_5 = ' ORDER BY B.bAddress DESC ';
              break;
          case 'url_asc':
              $query_5 = ' ORDER BY B.bAddress ASC ';
              break;
          default:
              $query_5 = ' ORDER BY B.bDatetime DESC ';
      }

      // Handle the parts of the query that depend on any tags that are present.
      $query_4 = '';
      for ($i = 0; $i < $tagcount; $i ++) {
          $query_2 .= ', '. $GLOBALS['tableprefix'] .'tags AS T'. $i;
          $query_4 .= ' AND T'. $i .'.tag = "'. $this->db->sql_escape($tags[$i]) .'" AND T'. $i .'.bId = B.bId';
      }

      // Search terms
      if ($terms) {
          // Multiple search terms okay
          $aTerms = explode(' ', $terms);
          $aTerms = array_map('trim', $aTerms);

          // Search terms in tags as well when none given
          if (!count($tags)) {
              $query_2 .= ' LEFT JOIN '. $GLOBALS['tableprefix'] .'tags AS T ON B.bId = T.bId';
              $dotags = true;
          } else {
              $dotags = false;
          }

          $query_4 = '';
          for ($i = 0; $i < count($aTerms); $i++) {
              $query_4 .= ' AND (B.bTitle LIKE "%'. $this->db->sql_escape($aTerms[$i]) .'%"';
              $query_4 .= ' OR B.bDescription LIKE "%'. $this->db->sql_escape($aTerms[$i]) .'%"';
              if ($dotags) {
                  $query_4 .= ' OR T.tag = "'. $this->db->sql_escape($aTerms[$i]) .'"';
              }
              $query_4 .= ')';
          }
      }

      // Start and end dates
      if ($startdate) {
          $query_4 .= ' AND B.bDatetime > "'. $startdate .'"';
      }
      if ($enddate) {
          $query_4 .= ' AND B.bDatetime < "'. $enddate .'"';
      }

      // Hash
      if ($hash) {
          $query_4 .= ' AND B.bHash = "'. $hash .'"';
      }

      $query = $query_1 . $query_2 . $query_3 . $query_4 . $query_5;
      if (!($dbresult = & $this->db->sql_query_limit($query, intval($perpage), intval($start)))) {
          message_die(GENERAL_ERROR, 'Could not get bookmarks', '', __LINE__, __FILE__, $query, $this->db);
          return false;
      }

      if (SQL_LAYER == 'mysql4') {
          $totalquery = 'SELECT FOUND_ROWS() AS total';
      } else {
          $totalquery = 'SELECT COUNT(*) AS total'. $query_2 . $query_3 . $query_4;
      }

      if (!($totalresult = & $this->db->sql_query($totalquery)) || (!($row = & $this->db->sql_fetchrow($totalresult)))) {
          message_die(GENERAL_ERROR, 'Could not get total bookmarks', '', __LINE__, __FILE__, $totalquery, $this->db);
          return false;
      }

      $total = $row['total'];

      $bookmarks = array();
      while ($row = & $this->db->sql_fetchrow($dbresult)) {
          $row['tags'] = $tagservice->getTagsForBookmark(intval($row['bId']));
          $bookmarks[] = $row;
      }
      return array ('bookmarks' => $bookmarks, 'total' => $total);
  }

  function deleteBookmark($bookmarkid) {
      $query = 'DELETE FROM '. $GLOBALS['tableprefix'] .'bookmarks WHERE bId = '. intval($bookmarkid);
      $this->db->sql_transaction('begin');
      if (!($dbresult = & $this->db->sql_query($query))) {
          $this->db->sql_transaction('rollback');
          message_die(GENERAL_ERROR, 'Could not delete bookmarks', '', __LINE__, __FILE__, $query, $this->db);
          return false;
      }

      $query = 'DELETE FROM '. $GLOBALS['tableprefix'] .'tags WHERE bId = '. intval($bookmarkid);
      $this->db->sql_transaction('begin');
      if (!($dbresult = & $this->db->sql_query($query))) {
          $this->db->sql_transaction('rollback');
          message_die(GENERAL_ERROR, 'Could not delete bookmarks', '', __LINE__, __FILE__, $query, $this->db);
          return false;
      }

      $this->db->sql_transaction('commit');
      return true;
  }

  function countOthers($address) {
      if (!$address) {
          return false;
      }

      $userservice = & ServiceFactory :: getServiceInstance('UserService');
      $sId = $userservice->getCurrentUserId();

      if ($userservice->isLoggedOn()) {
          // All public bookmarks, user's own bookmarks and any shared with user
          $privacy = ' AND ((B.bStatus = 0) OR (B.uId = '. $sId .')';
          $watchnames = $userservice->getWatchNames($sId, true);
          foreach($watchnames as $watchuser) {
              $privacy .= ' OR (U.username = "'. $watchuser .'" AND B.bStatus = 1)'; 
          }
          $privacy .= ')';
      } else {
          // Just public bookmarks
          $privacy = ' AND B.bStatus = 0';
      }

      $sql = 'SELECT COUNT(*) FROM '. $userservice->getTableName() .' AS U, '. $GLOBALS['tableprefix'] .'bookmarks AS B WHERE U.'. $userservice->getFieldName('primary') .' = B.uId AND B.bHash = "'. md5($address) .'"'. $privacy;
      if (!($dbresult = & $this->db->sql_query($sql))) {
          message_die(GENERAL_ERROR, 'Could not get vars', '', __LINE__, __FILE__, $sql, $this->db);
      }
      return $this->db->sql_fetchfield(0, 0) - 1;
  }
}