Blame view

sources/apps/files/command/scan.php 2.29 KB
31b7f2792   Kload   Upgrade to ownclo...
1
2
3
4
5
6
7
8
9
10
  <?php
  /**
   * Copyright (c) 2013 Thomas Müller <thomas.mueller@tmit.eu>
   * Copyright (c) 2013 Bart Visscher <bartv@thisnet.nl>
   * This file is licensed under the Affero General Public License version 3 or
   * later.
   * See the COPYING-README file.
   */
  
  namespace OCA\Files\Command;
6d9380f96   Cédric Dupont   Update sources OC...
11
  use OC\ForbiddenException;
31b7f2792   Kload   Upgrade to ownclo...
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
  use Symfony\Component\Console\Command\Command;
  use Symfony\Component\Console\Input\InputArgument;
  use Symfony\Component\Console\Input\InputInterface;
  use Symfony\Component\Console\Input\InputOption;
  use Symfony\Component\Console\Output\OutputInterface;
  
  class Scan extends Command {
  
  	/**
  	 * @var \OC\User\Manager $userManager
  	 */
  	private $userManager;
  
  	public function __construct(\OC\User\Manager $userManager) {
  		$this->userManager = $userManager;
  		parent::__construct();
  	}
  
  	protected function configure() {
  		$this
  			->setName('files:scan')
  			->setDescription('rescan filesystem')
  			->addArgument(
6d9380f96   Cédric Dupont   Update sources OC...
35
36
37
38
  				'user_id',
  				InputArgument::OPTIONAL | InputArgument::IS_ARRAY,
  				'will rescan all files of the given user(s)'
  			)
31b7f2792   Kload   Upgrade to ownclo...
39
  			->addOption(
6d9380f96   Cédric Dupont   Update sources OC...
40
41
42
43
44
  				'all',
  				null,
  				InputOption::VALUE_NONE,
  				'will rescan all files of all known users'
  			);
31b7f2792   Kload   Upgrade to ownclo...
45
46
47
48
  	}
  
  	protected function scanFiles($user, OutputInterface $output) {
  		$scanner = new \OC\Files\Utils\Scanner($user);
6d9380f96   Cédric Dupont   Update sources OC...
49
  		$scanner->listen('\OC\Files\Utils\Scanner', 'scanFile', function ($path) use ($output) {
31b7f2792   Kload   Upgrade to ownclo...
50
51
  			$output->writeln("Scanning <info>$path</info>");
  		});
6d9380f96   Cédric Dupont   Update sources OC...
52
  		$scanner->listen('\OC\Files\Utils\Scanner', 'scanFolder', function ($path) use ($output) {
31b7f2792   Kload   Upgrade to ownclo...
53
54
  			$output->writeln("Scanning <info>$path</info>");
  		});
6d9380f96   Cédric Dupont   Update sources OC...
55
56
57
58
59
60
  		try {
  			$scanner->scan('');
  		} catch (ForbiddenException $e) {
  			$output->writeln("<error>Home storage for user $user not writable</error>");
  			$output->writeln("Make sure you're running the scan command only as the user the web server runs as");
  		}
31b7f2792   Kload   Upgrade to ownclo...
61
62
63
64
65
66
67
68
  	}
  
  	protected function execute(InputInterface $input, OutputInterface $output) {
  		if ($input->getOption('all')) {
  			$users = $this->userManager->search('');
  		} else {
  			$users = $input->getArgument('user_id');
  		}
6d9380f96   Cédric Dupont   Update sources OC...
69
70
71
72
  		if (count($users) === 0) {
  			$output->writeln("<error>Please specify the user id to scan or \"--all\" to scan for all users</error>");
  			return;
  		}
31b7f2792   Kload   Upgrade to ownclo...
73
74
75
76
77
78
79
80
  		foreach ($users as $user) {
  			if (is_object($user)) {
  				$user = $user->getUID();
  			}
  			$this->scanFiles($user, $output);
  		}
  	}
  }