fixes #326 Signed-off-by: Marcel Klehr <mklehr@gmx.net>pull/26813/head
Marcel Klehr
5 years ago
parent
4f55ba2a88
commit
f755ee0868
@ -0,0 +1,217 @@ |
||||
<?php |
||||
|
||||
declare(strict_types=1); |
||||
|
||||
/** |
||||
* @copyright 2020 Christoph Wurst <christoph@winzerhof-wurst.at> |
||||
* |
||||
* @author Christoph Wurst <christoph@winzerhof-wurst.at> |
||||
* @author Joas Schilling <coding@schilljs.com> |
||||
* @author John Molakvoæ <skjnldsv@protonmail.com> |
||||
* @author Robin Appelman <robin@icewind.nl> |
||||
* @author Roeland Jago Douma <roeland@famdouma.nl> |
||||
* @author Marcel Klehr <mklehr@gmx.net> |
||||
* |
||||
* @license GNU AGPL version 3 or any later version |
||||
* |
||||
* This program is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU Affero General Public License as |
||||
* published by the Free Software Foundation, either version 3 of the |
||||
* License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU Affero General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Affero General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>. |
||||
* |
||||
*/ |
||||
namespace OCA\SystemTags\Search; |
||||
|
||||
use OC\Files\Search\SearchBinaryOperator; |
||||
use OC\Files\Search\SearchComparison; |
||||
use OC\Files\Search\SearchOrder; |
||||
use OC\Files\Search\SearchQuery; |
||||
use OCP\SystemTag\ISystemTag; |
||||
use OCP\SystemTag\ISystemTagManager; |
||||
use OCP\SystemTag\ISystemTagObjectMapper; |
||||
use OCP\Files\FileInfo; |
||||
use OCP\Files\IMimeTypeDetector; |
||||
use OCP\Files\IRootFolder; |
||||
use OCP\Files\Search\ISearchComparison; |
||||
use OCP\Files\Node; |
||||
use OCP\Files\Search\ISearchOrder; |
||||
use OCP\IL10N; |
||||
use OCP\IURLGenerator; |
||||
use OCP\IUser; |
||||
use OCP\Search\IProvider; |
||||
use OCP\Search\ISearchQuery; |
||||
use OCP\Search\SearchResult; |
||||
use OCP\Search\SearchResultEntry; |
||||
use RecursiveArrayIterator; |
||||
use RecursiveIteratorIterator; |
||||
|
||||
class TagSearchProvider implements IProvider { |
||||
|
||||
/** @var IL10N */ |
||||
private $l10n; |
||||
|
||||
/** @var IURLGenerator */ |
||||
private $urlGenerator; |
||||
|
||||
/** @var IMimeTypeDetector */ |
||||
private $mimeTypeDetector; |
||||
|
||||
/** @var IRootFolder */ |
||||
private $rootFolder; |
||||
private ISystemTagObjectMapper $objectMapper; |
||||
private ISystemTagManager $tagManager; |
||||
|
||||
public function __construct( |
||||
IL10N $l10n, |
||||
IURLGenerator $urlGenerator, |
||||
IMimeTypeDetector $mimeTypeDetector, |
||||
IRootFolder $rootFolder, |
||||
ISystemTagObjectMapper $objectMapper, |
||||
ISystemTagManager $tagManager |
||||
) { |
||||
$this->l10n = $l10n; |
||||
$this->urlGenerator = $urlGenerator; |
||||
$this->mimeTypeDetector = $mimeTypeDetector; |
||||
$this->rootFolder = $rootFolder; |
||||
$this->objectMapper = $objectMapper; |
||||
$this->tagManager = $tagManager; |
||||
} |
||||
|
||||
/** |
||||
* @inheritDoc |
||||
*/ |
||||
public function getId(): string { |
||||
return 'systemtags'; |
||||
} |
||||
|
||||
/** |
||||
* @inheritDoc |
||||
*/ |
||||
public function getName(): string { |
||||
return $this->l10n->t('Tags'); |
||||
} |
||||
|
||||
/** |
||||
* @inheritDoc |
||||
*/ |
||||
public function getOrder(string $route, array $routeParameters): int { |
||||
if ($route === 'files.View.index') { |
||||
return -4; |
||||
} |
||||
return 6; |
||||
} |
||||
|
||||
/** |
||||
* @inheritDoc |
||||
*/ |
||||
public function search(IUser $user, ISearchQuery $query): SearchResult { |
||||
$userFolder = $this->rootFolder->getUserFolder($user->getUID()); |
||||
$fileQuery = new SearchQuery( |
||||
new SearchBinaryOperator(SearchBinaryOperator::OPERATOR_OR, [ |
||||
new SearchComparison(ISearchComparison::COMPARE_LIKE, 'tagname', '%' . $query->getTerm() . '%'), |
||||
new SearchComparison(ISearchComparison::COMPARE_LIKE, 'systemtag', '%' . $query->getTerm() . '%'), |
||||
]), |
||||
$query->getLimit(), |
||||
(int)$query->getCursor(), |
||||
$query->getSortOrder() === ISearchQuery::SORT_DATE_DESC ? [ |
||||
new SearchOrder(ISearchOrder::DIRECTION_DESCENDING, 'mtime'), |
||||
] : [], |
||||
$user |
||||
); |
||||
|
||||
// do search |
||||
$searchResults = $userFolder->search($fileQuery); |
||||
$resultIds = array_map(function(Node $node) { |
||||
return $node->getId(); |
||||
}, $searchResults); |
||||
$matchedTags = $this->objectMapper->getTagIdsForObjects($resultIds, 'files'); |
||||
$relevantTags = $this->tagManager->getTagsByIds(array_unique($this->flattenArray($matchedTags))); |
||||
|
||||
// prepare direct tag results |
||||
$tagResults = array_map(function(ISystemTag $tag) { |
||||
$thumbnailUrl = ''; |
||||
$link = $this->urlGenerator->linkToRoute( |
||||
'files.view.index' |
||||
) . '?view=systemtagsfilter&tags='.$tag->getId(); |
||||
$searchResultEntry = new SearchResultEntry( |
||||
$thumbnailUrl, |
||||
$this->l10n->t('All tagged %s …', [$tag->getName()]), |
||||
'', |
||||
$this->urlGenerator->getAbsoluteURL($link), |
||||
'icon-tag' |
||||
); |
||||
return $searchResultEntry; |
||||
}, array_filter($relevantTags, function($tag) use ($query) { |
||||
return $tag->isUserVisible() && strpos($tag->getName(), $query->getTerm()) !== false; |
||||
})); |
||||
|
||||
// prepare files results |
||||
return SearchResult::paginated( |
||||
$this->l10n->t('Tags'), |
||||
array_map(function (Node $result) use ($userFolder, $matchedTags, $query) { |
||||
// Generate thumbnail url |
||||
$thumbnailUrl = $this->urlGenerator->linkToRouteAbsolute('core.Preview.getPreviewByFileId', ['x' => 32, 'y' => 32, 'fileId' => $result->getId()]); |
||||
$path = $userFolder->getRelativePath($result->getPath()); |
||||
|
||||
// Use shortened link to centralize the various |
||||
// files/folder url redirection in files.View.showFile |
||||
$link = $this->urlGenerator->linkToRoute( |
||||
'files.View.showFile', |
||||
['fileid' => $result->getId()] |
||||
); |
||||
|
||||
$searchResultEntry = new SearchResultEntry( |
||||
$thumbnailUrl, |
||||
$result->getName(), |
||||
$this->formatSubline($query, $matchedTags[$result->getId()]), |
||||
$this->urlGenerator->getAbsoluteURL($link), |
||||
$result->getMimetype() === FileInfo::MIMETYPE_FOLDER ? 'icon-folder' : $this->mimeTypeDetector->mimeTypeIcon($result->getMimetype()) |
||||
); |
||||
$searchResultEntry->addAttribute('fileId', (string)$result->getId()); |
||||
$searchResultEntry->addAttribute('path', $path); |
||||
return $searchResultEntry; |
||||
}, $searchResults) |
||||
+ $tagResults, |
||||
$query->getCursor() + $query->getLimit() |
||||
); |
||||
} |
||||
|
||||
/** |
||||
* Format subline for tagged files: Show the first 3 tags |
||||
* |
||||
* @param $query |
||||
* @param array $tagInfo |
||||
* @return string |
||||
*/ |
||||
private function formatSubline(ISearchQuery $query, array $tagInfo): string { |
||||
/** |
||||
* @var ISystemTag[] |
||||
*/ |
||||
$tags = $this->tagManager->getTagsByIds($tagInfo); |
||||
$tagNames = array_map(function($tag) { |
||||
return $tag->getName(); |
||||
}, array_filter($tags, function($tag) { |
||||
return $tag->isUserVisible(); |
||||
})); |
||||
|
||||
// show the tag that you have searched for first |
||||
usort($tagNames, function($tagName) use($query) { |
||||
return strpos($tagName, $query->getTerm()) !== false? -1 : 1; |
||||
}); |
||||
|
||||
return $this->l10n->t('tagged %s', [implode(', ', array_slice($tagNames, 0, 3))]); |
||||
} |
||||
|
||||
private function flattenArray($array) { |
||||
$it = new RecursiveIteratorIterator(new RecursiveArrayIterator($array)); |
||||
return iterator_to_array($it, true); |
||||
} |
||||
} |
||||
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
Loading…
Reference in new issue