2023-01-23 11:03:31 +01:00

405 lines
14 KiB
PHP

<?php
/**
* Piwik - free/libre analytics platform
*
* @link http://piwik.org
* @license http://www.gnu.org/licenses/gpl-3.0.html GPL v3 or later
*
*/
namespace Piwik;
use Exception;
use Piwik\ArchiveProcessor\Rules;
use Piwik\Container\StaticContainer;
use Piwik\DataAccess\LogQueryBuilder;
use Piwik\Plugins\API\API;
use Piwik\Segment\SegmentExpression;
/**
* Limits the set of visits Piwik uses when aggregating analytics data.
*
* A segment is a condition used to filter visits. They can, for example,
* select visits that have a specific browser or come from a specific
* country, or both.
*
* Plugins that aggregate data stored in Piwik can support segments by
* using this class when generating aggregation SQL queries.
*
* ### Examples
*
* **Basic usage**
*
* $idSites = array(1,2,3);
* $segmentStr = "browserCode==ff;countryCode==CA";
* $segment = new Segment($segmentStr, $idSites);
*
* $query = $segment->getSelectQuery(
* $select = "table.col1, table2.col2",
* $from = array("table", "table2"),
* $where = "table.col3 = ?",
* $bind = array(5),
* $orderBy = "table.col1 DESC",
* $groupBy = "table2.col2"
* );
*
* Db::fetchAll($query['sql'], $query['bind']);
*
* **Creating a _null_ segment**
*
* $idSites = array(1,2,3);
* $segment = new Segment('', $idSites);
* // $segment->getSelectQuery will return a query that selects all visits
*
* @api
*/
class Segment
{
/**
* @var SegmentExpression
*/
protected $segmentExpression = null;
/**
* @var string
*/
protected $string = null;
/**
* @var array
*/
protected $idSites = null;
/**
* @var LogQueryBuilder
*/
private $segmentQueryBuilder;
/**
* Truncate the Segments to 8k
*/
const SEGMENT_TRUNCATE_LIMIT = 8192;
/**
* Constructor.
*
* @param string $segmentCondition The segment condition, eg, `'browserCode=ff;countryCode=CA'`.
* @param array $idSites The list of sites the segment will be used with. Some segments are
* dependent on the site, such as goal segments.
* @throws
*/
public function __construct($segmentCondition, $idSites)
{
$this->segmentQueryBuilder = StaticContainer::get('Piwik\DataAccess\LogQueryBuilder');
$segmentCondition = trim($segmentCondition);
if (!SettingsPiwik::isSegmentationEnabled()
&& !empty($segmentCondition)
) {
throw new Exception("The Super User has disabled the Segmentation feature.");
}
// First try with url decoded value. If that fails, try with raw value.
// If that also fails, it will throw the exception
try {
$this->initializeSegment(urldecode($segmentCondition), $idSites);
} catch (Exception $e) {
$this->initializeSegment($segmentCondition, $idSites);
}
}
/**
* Returns the segment expression.
* @return SegmentExpression
* @api since Piwik 3.2.0
*/
public function getSegmentExpression()
{
return $this->segmentExpression;
}
private function getAvailableSegments()
{
// segment metadata
if (empty($this->availableSegments)) {
$this->availableSegments = API::getInstance()->getSegmentsMetadata($this->idSites, $_hideImplementationData = false);
}
return $this->availableSegments;
}
private function getSegmentByName($name)
{
$segments = $this->getAvailableSegments();
foreach ($segments as $segment) {
if ($segment['segment'] == $name && !empty($name)) {
// check permission
if (isset($segment['permission']) && $segment['permission'] != 1) {
throw new NoAccessException("You do not have enough permission to access the segment " . $name);
}
return $segment;
}
}
throw new Exception("Segment '$name' is not a supported segment.");
}
/**
* @param $string
* @param $idSites
* @throws Exception
*/
protected function initializeSegment($string, $idSites)
{
// As a preventive measure, we restrict the filter size to a safe limit
$string = substr($string, 0, self::SEGMENT_TRUNCATE_LIMIT);
$this->string = $string;
$this->idSites = $idSites;
$segment = new SegmentExpression($string);
$this->segmentExpression = $segment;
// parse segments
$expressions = $segment->parseSubExpressions();
$expressions = $this->getExpressionsWithUnionsResolved($expressions);
// convert segments name to sql segment
// check that user is allowed to view this segment
// and apply a filter to the value to match if necessary (to map DB fields format)
$cleanedExpressions = array();
foreach ($expressions as $expression) {
$operand = $expression[SegmentExpression::INDEX_OPERAND];
$cleanedExpression = $this->getCleanedExpression($operand);
$expression[SegmentExpression::INDEX_OPERAND] = $cleanedExpression;
$cleanedExpressions[] = $expression;
}
$segment->setSubExpressionsAfterCleanup($cleanedExpressions);
}
private function getExpressionsWithUnionsResolved($expressions)
{
$expressionsWithUnions = array();
foreach ($expressions as $expression) {
$operand = $expression[SegmentExpression::INDEX_OPERAND];
$name = $operand[SegmentExpression::INDEX_OPERAND_NAME];
$availableSegment = $this->getSegmentByName($name);
if (!empty($availableSegment['unionOfSegments'])) {
$count = 0;
foreach ($availableSegment['unionOfSegments'] as $segmentNameOfUnion) {
$count++;
$operator = SegmentExpression::BOOL_OPERATOR_OR; // we connect all segments within that union via OR
if ($count === count($availableSegment['unionOfSegments'])) {
$operator = $expression[SegmentExpression::INDEX_BOOL_OPERATOR];
}
$operand[SegmentExpression::INDEX_OPERAND_NAME] = $segmentNameOfUnion;
$expressionsWithUnions[] = array(
SegmentExpression::INDEX_BOOL_OPERATOR => $operator,
SegmentExpression::INDEX_OPERAND => $operand
);
}
} else {
$expressionsWithUnions[] = array(
SegmentExpression::INDEX_BOOL_OPERATOR => $expression[SegmentExpression::INDEX_BOOL_OPERATOR],
SegmentExpression::INDEX_OPERAND => $operand
);
}
}
return $expressionsWithUnions;
}
/**
* Returns `true` if the segment is empty, `false` if otherwise.
*/
public function isEmpty()
{
return $this->segmentExpression->isEmpty();
}
/**
* Detects whether the Piwik instance is configured to be able to archive this segment. It checks whether the segment
* will be either archived via browser or cli archiving. It does not check if the segment has been archived. If you
* want to know whether the segment has been archived, the actual report data needs to be requested.
*
* This method does not take any date/period into consideration. Meaning a Piwik instance might be able to archive
* this segment in general, but not for a certain period if eg the archiving of range dates is disabled.
*
* @return bool
*/
public function willBeArchived()
{
if ($this->isEmpty()) {
return true;
}
$idSites = $this->idSites;
if (!is_array($idSites)) {
$idSites = array($this->idSites);
}
return Rules::isRequestAuthorizedToArchive()
|| Rules::isBrowserArchivingAvailableForSegments()
|| Rules::isSegmentPreProcessed($idSites, $this);
}
protected $availableSegments = array();
protected function getCleanedExpression($expression)
{
$name = $expression[SegmentExpression::INDEX_OPERAND_NAME];
$matchType = $expression[SegmentExpression::INDEX_OPERAND_OPERATOR];
$value = $expression[SegmentExpression::INDEX_OPERAND_VALUE];
$segment = $this->getSegmentByName($name);
$sqlName = $segment['sqlSegment'];
if ($matchType != SegmentExpression::MATCH_IS_NOT_NULL_NOR_EMPTY
&& $matchType != SegmentExpression::MATCH_IS_NULL_OR_EMPTY) {
if (isset($segment['sqlFilterValue'])) {
$value = call_user_func($segment['sqlFilterValue'], $value, $segment['sqlSegment']);
}
// apply presentation filter
if (isset($segment['sqlFilter'])) {
$value = call_user_func($segment['sqlFilter'], $value, $segment['sqlSegment'], $matchType, $name);
if(is_null($value)) { // null is returned in TableLogAction::getIdActionFromSegment()
return array(null, $matchType, null);
}
// sqlFilter-callbacks might return arrays for more complex cases
// e.g. see TableLogAction::getIdActionFromSegment()
if (is_array($value) && isset($value['SQL'])) {
// Special case: returned value is a sub sql expression!
$matchType = SegmentExpression::MATCH_ACTIONS_CONTAINS;
}
}
}
return array($sqlName, $matchType, $value);
}
/**
* Returns the segment condition.
*
* @return string
*/
public function getString()
{
return $this->string;
}
/**
* Returns a hash of the segment condition, or the empty string if the segment
* condition is empty.
*
* @return string
*/
public function getHash()
{
if (empty($this->string)) {
return '';
}
// normalize the string as browsers may send slightly different payloads for the same archive
$normalizedSegmentString = urldecode($this->string);
return md5($normalizedSegmentString);
}
/**
* Extend an SQL query that aggregates data over one of the 'log_' tables with segment expressions.
*
* @param string $select The select clause. Should NOT include the **SELECT** just the columns, eg,
* `'t1.col1 as col1, t2.col2 as col2'`.
* @param array $from Array of table names (without prefix), eg, `array('log_visit', 'log_conversion')`.
* @param false|string $where (optional) Where clause, eg, `'t1.col1 = ? AND t2.col2 = ?'`.
* @param array|string $bind (optional) Bind parameters, eg, `array($col1Value, $col2Value)`.
* @param false|string $orderBy (optional) Order by clause, eg, `"t1.col1 ASC"`.
* @param false|string $groupBy (optional) Group by clause, eg, `"t2.col2"`.
* @param int $limit Limit number of result to $limit
* @param int $offset Specified the offset of the first row to return
* @param int If set to value >= 1 then the Select query (and All inner queries) will be LIMIT'ed by this value.
* Use only when you're not aggregating or it will sample the data.
* @return string The entire select query.
*/
public function getSelectQuery($select, $from, $where = false, $bind = array(), $orderBy = false, $groupBy = false, $limit = 0, $offset = 0)
{
$segmentExpression = $this->segmentExpression;
$limitAndOffset = null;
if($limit > 0) {
$limitAndOffset = (int) $offset . ', ' . (int) $limit;
}
return $this->segmentQueryBuilder->getSelectQueryString($segmentExpression, $select, $from, $where, $bind,
$groupBy, $orderBy, $limitAndOffset);
}
/**
* Returns the segment string.
*
* @return string
*/
public function __toString()
{
return (string) $this->getString();
}
/**
* Combines this segment with another segment condition, if the segment condition is not already
* in the segment.
*
* The combination is naive in that it does not take order of operations into account.
*
* @param string $segment
* @param string $operator The operator to use. Should be either SegmentExpression::AND_DELIMITER
* or SegmentExpression::OR_DELIMITER.
* @param string $segmentCondition The segment condition to add.
* @return string
* @throws Exception
*/
public static function combine($segment, $operator, $segmentCondition)
{
if (empty($segment)) {
return $segmentCondition;
}
if (empty($segmentCondition)
|| self::containsCondition($segment, $operator, $segmentCondition)
) {
return $segment;
}
return $segment . $operator . $segmentCondition;
}
private static function containsCondition($segment, $operator, $segmentCondition)
{
// check when segment/condition are of same encoding
return strpos($segment, $operator . $segmentCondition) !== false
|| strpos($segment, $segmentCondition . $operator) !== false
// check when both operator & condition are urlencoded in $segment
|| strpos($segment, urlencode($operator . $segmentCondition)) !== false
|| strpos($segment, urlencode($segmentCondition . $operator)) !== false
// check when operator is not urlencoded, but condition is in $segment
|| strpos($segment, $operator . urlencode($segmentCondition)) !== false
|| strpos($segment, urlencode($segmentCondition) . $operator) !== false
// check when segment condition is urlencoded & $segment isn't
|| strpos($segment, $operator . urldecode($segmentCondition)) !== false
|| strpos($segment, urldecode($segmentCondition) . $operator) !== false
|| $segment === $segmentCondition
|| $segment === urlencode($segmentCondition)
|| $segment === urldecode($segmentCondition);
}
}