Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F61522702
PhrictionDocumentQuery.php
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Tue, May 7, 05:01
Size
10 KB
Mime Type
text/x-php
Expires
Thu, May 9, 05:01 (2 d)
Engine
blob
Format
Raw Data
Handle
17523053
Attached To
rPH Phabricator
PhrictionDocumentQuery.php
View Options
<?php
final
class
PhrictionDocumentQuery
extends
PhabricatorCursorPagedPolicyAwareQuery
{
private
$ids
;
private
$phids
;
private
$slugs
;
private
$depths
;
private
$slugPrefix
;
private
$slugsPrefix
;
// c4science custo
private
$statuses
;
private
$parentPaths
;
private
$ancestorPaths
;
private
$needContent
;
const
ORDER_HIERARCHY
=
'hierarchy'
;
public
function
withIDs
(
array
$ids
)
{
$this
->
ids
=
$ids
;
return
$this
;
}
public
function
withPHIDs
(
array
$phids
)
{
$this
->
phids
=
$phids
;
return
$this
;
}
public
function
withSlugs
(
array
$slugs
)
{
$this
->
slugs
=
$slugs
;
return
$this
;
}
public
function
withDepths
(
array
$depths
)
{
$this
->
depths
=
$depths
;
return
$this
;
}
public
function
withSlugPrefix
(
$slug_prefix
)
{
$this
->
slugPrefix
=
$slug_prefix
;
return
$this
;
}
// c4science customization
public
function
withSlugsPrefix
(
$slugs_prefix
)
{
$this
->
slugsPrefix
=
$slugs_prefix
;
return
$this
;
}
public
function
withStatuses
(
array
$statuses
)
{
$this
->
statuses
=
$statuses
;
return
$this
;
}
public
function
withParentPaths
(
array
$paths
)
{
$this
->
parentPaths
=
$paths
;
return
$this
;
}
public
function
withAncestorPaths
(
array
$paths
)
{
$this
->
ancestorPaths
=
$paths
;
return
$this
;
}
public
function
needContent
(
$need_content
)
{
$this
->
needContent
=
$need_content
;
return
$this
;
}
protected
function
loadPage
()
{
return
$this
->
loadStandardPage
(
$this
->
newResultObject
());
}
public
function
newResultObject
()
{
return
new
PhrictionDocument
();
}
protected
function
willFilterPage
(
array
$documents
)
{
if
(
$documents
)
{
$ancestor_slugs
=
array
();
foreach
(
$documents
as
$key
=>
$document
)
{
$document_slug
=
$document
->
getSlug
();
foreach
(
PhabricatorSlug
::
getAncestry
(
$document_slug
)
as
$ancestor
)
{
$ancestor_slugs
[
$ancestor
][]
=
$key
;
}
}
if
(
$ancestor_slugs
)
{
$table
=
new
PhrictionDocument
();
$conn_r
=
$table
->
establishConnection
(
'r'
);
$ancestors
=
queryfx_all
(
$conn_r
,
'SELECT * FROM %T WHERE slug IN (%Ls)'
,
$document
->
getTableName
(),
array_keys
(
$ancestor_slugs
));
$ancestors
=
$table
->
loadAllFromArray
(
$ancestors
);
$ancestors
=
mpull
(
$ancestors
,
null
,
'getSlug'
);
foreach
(
$ancestor_slugs
as
$ancestor_slug
=>
$document_keys
)
{
$ancestor
=
idx
(
$ancestors
,
$ancestor_slug
);
foreach
(
$document_keys
as
$document_key
)
{
$documents
[
$document_key
]->
attachAncestor
(
$ancestor_slug
,
$ancestor
);
}
}
}
}
// To view a Phriction document, you must also be able to view all of the
// ancestor documents. Filter out documents which have ancestors that are
// not visible.
$document_map
=
array
();
foreach
(
$documents
as
$document
)
{
$document_map
[
$document
->
getSlug
()]
=
$document
;
foreach
(
$document
->
getAncestors
()
as
$key
=>
$ancestor
)
{
if
(
$ancestor
)
{
$document_map
[
$key
]
=
$ancestor
;
}
}
}
$filtered_map
=
$this
->
applyPolicyFilter
(
$document_map
,
array
(
PhabricatorPolicyCapability
::
CAN_VIEW
));
// Filter all of the documents where a parent is not visible.
foreach
(
$documents
as
$document_key
=>
$document
)
{
// If the document itself is not visible, filter it.
if
(!
isset
(
$filtered_map
[
$document
->
getSlug
()]))
{
$this
->
didRejectResult
(
$documents
[
$document_key
]);
unset
(
$documents
[
$document_key
]);
continue
;
}
// If an ancestor exists but is not visible, filter the document.
foreach
(
$document
->
getAncestors
()
as
$ancestor_key
=>
$ancestor
)
{
if
(!
$ancestor
)
{
continue
;
}
if
(!
isset
(
$filtered_map
[
$ancestor_key
]))
{
$this
->
didRejectResult
(
$documents
[
$document_key
]);
unset
(
$documents
[
$document_key
]);
break
;
}
}
}
if
(!
$documents
)
{
return
$documents
;
}
if
(
$this
->
needContent
)
{
$contents
=
id
(
new
PhrictionContentQuery
())
->
setViewer
(
$this
->
getViewer
())
->
setParentQuery
(
$this
)
->
withPHIDs
(
mpull
(
$documents
,
'getContentPHID'
))
->
execute
();
$contents
=
mpull
(
$contents
,
null
,
'getPHID'
);
foreach
(
$documents
as
$key
=>
$document
)
{
$content_phid
=
$document
->
getContentPHID
();
if
(
empty
(
$contents
[
$content_phid
]))
{
unset
(
$documents
[
$key
]);
continue
;
}
$document
->
attachContent
(
$contents
[
$content_phid
]);
}
}
return
$documents
;
}
protected
function
buildSelectClauseParts
(
AphrontDatabaseConnection
$conn
)
{
$select
=
parent
::
buildSelectClauseParts
(
$conn
);
if
(
$this
->
shouldJoinContentTable
())
{
$select
[]
=
qsprintf
(
$conn
,
'c.title'
);
}
return
$select
;
}
protected
function
buildJoinClauseParts
(
AphrontDatabaseConnection
$conn
)
{
$joins
=
parent
::
buildJoinClauseParts
(
$conn
);
if
(
$this
->
shouldJoinContentTable
())
{
$content_dao
=
new
PhrictionContent
();
$joins
[]
=
qsprintf
(
$conn
,
'JOIN %T c ON d.contentPHID = c.phid'
,
$content_dao
->
getTableName
());
}
return
$joins
;
}
private
function
shouldJoinContentTable
()
{
if
(
$this
->
getOrderVector
()->
containsKey
(
'title'
))
{
return
true
;
}
return
false
;
}
protected
function
buildWhereClauseParts
(
AphrontDatabaseConnection
$conn
)
{
$where
=
parent
::
buildWhereClauseParts
(
$conn
);
if
(
$this
->
ids
!==
null
)
{
$where
[]
=
qsprintf
(
$conn
,
'd.id IN (%Ld)'
,
$this
->
ids
);
}
if
(
$this
->
phids
!==
null
)
{
$where
[]
=
qsprintf
(
$conn
,
'd.phid IN (%Ls)'
,
$this
->
phids
);
}
if
(
$this
->
slugs
!==
null
)
{
$where
[]
=
qsprintf
(
$conn
,
'd.slug IN (%Ls)'
,
$this
->
slugs
);
}
if
(
$this
->
statuses
!==
null
)
{
$where
[]
=
qsprintf
(
$conn
,
'd.status IN (%Ls)'
,
$this
->
statuses
);
}
if
(
$this
->
slugPrefix
!==
null
)
{
$where
[]
=
qsprintf
(
$conn
,
'd.slug LIKE %>'
,
$this
->
slugPrefix
);
}
// c4science customization
if
(
$this
->
slugsPrefix
!==
null
)
{
foreach
(
$this
->
slugsPrefix
as
$prefix
)
{
$where
[]
=
qsprintf
(
$conn
,
'd.slug LIKE %>'
,
$prefix
);
}
}
if
(
$this
->
depths
!==
null
)
{
$where
[]
=
qsprintf
(
$conn
,
'd.depth IN (%Ld)'
,
$this
->
depths
);
}
if
(
$this
->
parentPaths
!==
null
||
$this
->
ancestorPaths
!==
null
)
{
$sets
=
array
(
array
(
'paths'
=>
$this
->
parentPaths
,
'parents'
=>
true
,
),
array
(
'paths'
=>
$this
->
ancestorPaths
,
'parents'
=>
false
,
),
);
$paths
=
array
();
foreach
(
$sets
as
$set
)
{
$set_paths
=
$set
[
'paths'
];
if
(
$set_paths
===
null
)
{
continue
;
}
if
(!
$set_paths
)
{
throw
new
PhabricatorEmptyQueryException
(
pht
(
'No parent/ancestor paths specified.'
));
}
$is_parents
=
$set
[
'parents'
];
foreach
(
$set_paths
as
$path
)
{
$path_normal
=
PhabricatorSlug
::
normalize
(
$path
);
if
(
$path
!==
$path_normal
)
{
throw
new
Exception
(
pht
(
'Document path "%s" is not a valid path. The normalized '
.
'form of this path is "%s".'
,
$path
,
$path_normal
));
}
$depth
=
PhabricatorSlug
::
getDepth
(
$path_normal
);
if
(
$is_parents
)
{
$min_depth
=
$depth
+
1
;
$max_depth
=
$depth
+
1
;
}
else
{
$min_depth
=
$depth
+
1
;
$max_depth
=
null
;
}
$paths
[]
=
array
(
$path_normal
,
$min_depth
,
$max_depth
,
);
}
}
$path_clauses
=
array
();
foreach
(
$paths
as
$path
)
{
$parts
=
array
();
list
(
$prefix
,
$min
,
$max
)
=
$path
;
// If we're getting children or ancestors of the root document, they
// aren't actually stored with the leading "/" in the database, so
// just skip this part of the clause.
if
(
$prefix
!==
'/'
)
{
$parts
[]
=
qsprintf
(
$conn
,
'd.slug LIKE %>'
,
$prefix
);
}
if
(
$min
!==
null
)
{
$parts
[]
=
qsprintf
(
$conn
,
'd.depth >= %d'
,
$min
);
}
if
(
$max
!==
null
)
{
$parts
[]
=
qsprintf
(
$conn
,
'd.depth <= %d'
,
$max
);
}
if
(
$parts
)
{
$path_clauses
[]
=
qsprintf
(
$conn
,
'%LA'
,
$parts
);
}
}
if
(
$path_clauses
)
{
$where
[]
=
qsprintf
(
$conn
,
'%LO'
,
$path_clauses
);
}
}
return
$where
;
}
public
function
getBuiltinOrders
()
{
return
parent
::
getBuiltinOrders
()
+
array
(
self
::
ORDER_HIERARCHY
=>
array
(
'vector'
=>
array
(
'depth'
,
'title'
,
'updated'
,
'id'
),
'name'
=>
pht
(
'Hierarchy'
),
),
);
}
public
function
getOrderableColumns
()
{
return
parent
::
getOrderableColumns
()
+
array
(
'depth'
=>
array
(
'table'
=>
'd'
,
'column'
=>
'depth'
,
'reverse'
=>
true
,
'type'
=>
'int'
,
),
'title'
=>
array
(
'table'
=>
'c'
,
'column'
=>
'title'
,
'reverse'
=>
true
,
'type'
=>
'string'
,
),
'updated'
=>
array
(
'table'
=>
'd'
,
'column'
=>
'editedEpoch'
,
'type'
=>
'int'
,
'unique'
=>
false
,
),
);
}
protected
function
newPagingMapFromCursorObject
(
PhabricatorQueryCursor
$cursor
,
array
$keys
)
{
$document
=
$cursor
->
getObject
();
$map
=
array
(
'id'
=>
(
int
)
$document
->
getID
(),
'depth'
=>
$document
->
getDepth
(),
'updated'
=>
(
int
)
$document
->
getEditedEpoch
(),
);
if
(
isset
(
$keys
[
'title'
]))
{
$map
[
'title'
]
=
$cursor
->
getRawRowProperty
(
'title'
);
}
return
$map
;
}
protected
function
getPrimaryTableAlias
()
{
return
'd'
;
}
public
function
getQueryApplicationClass
()
{
return
'PhabricatorPhrictionApplication'
;
}
}
Event Timeline
Log In to Comment