date.chapril.org-framadate/app/classes/Framadate/Services/PurgeService.php

<?php
namespace Framadate\Services;
use Framadate\FramaDB;
use Framadate\Repositories\RepositoryFactory;

/**
 * This service helps to purge data.
 *
 * @package Framadate\Services
 */
class PurgeService {
    private $logService;
    private $pollRepository;
    private $slotRepository;
    private $voteRepository;
    private $commentRepository;

    function __construct(FramaDB $connect, LogService $logService) {
        $this->logService = $logService;
        $this->pollRepository = RepositoryFactory::pollRepository();
        $this->slotRepository = RepositoryFactory::slotRepository();
        $this->voteRepository = RepositoryFactory::voteRepository();
        $this->commentRepository = RepositoryFactory::commentRepository();
    }

    public function repeatedCleanings() {
    	$this->purgeOldPolls();
    	$this->cleanDemoPoll();
    }
    
    /**
     * This methode purges all old polls (the ones with end_date in past).
     *
     * @return bool true is action succeeded
     */
    function purgeOldPolls() {
        $oldPolls = $this->pollRepository->findOldPolls();
        $count = count($oldPolls);

        if ($count > 0) {
            $this->logService->log('EXPIRATION', 'Going to purge ' . $count . ' poll(s)...');

            foreach ($oldPolls as $poll) {
                if ($this->purgePollById($poll->id)) {
                    $this->logService->log('EXPIRATION_SUCCESS', 'id: ' . $poll->id . ', title:' . $poll->title . ', format: ' . $poll->format . ', admin: ' . $poll->admin_name);
                } else {
                    $this->logService->log('EXPIRATION_FAILED', 'id: ' . $poll->id . ', title:' . $poll->title . ', format: ' . $poll->format . ', admin: ' . $poll->admin_name);
                }
            }
        }

        return $count;
    }
    
    public function cleanDemoPoll() {
    	if (!defined("DEMO_POLL_ID") || !defined("DEMO_POLL_NUMBER_VOTES")) {
    		return;
    	}
    	
    	$this->voteRepository->beginTransaction();
    	
    	$demoVotes = $this->voteRepository->allUserVotesByPollId(DEMO_POLL_ID);
    	$votesToDelete = count($demoVotes) - DEMO_POLL_NUMBER_VOTES;
    	
    	if ($votesToDelete > 0) {
    		$this->voteRepository->deleteOldVotesByPollId(DEMO_POLL_ID, $votesToDelete);
    	}
    	
    	$this->voteRepository->commit();
    }
    
    /**
     * This methode delete all data about a poll.
     *
     * @param $poll_id int The ID of the poll
     * @return bool true is action succeeded
     */
    function purgePollById($poll_id) {
        $done = true;

        $this->pollRepository->beginTransaction();
        $done &= $this->commentRepository->deleteByPollId($poll_id);
        $done &= $this->voteRepository->deleteByPollId($poll_id);
        $done &= $this->slotRepository->deleteByPollId($poll_id);
        $done &= $this->pollRepository->deleteById($poll_id);

        if ($done) {
            $this->pollRepository->commit();
        } else {
            $this->pollRepository->rollback();
        }

        return $done;
    }
}