Passed
Push — master ( 15d161...316d3b )
by Nicolaas
02:03
created

getAllVersionedDataClassesBase()   A

Complexity

Conditions 5
Paths 5

Size

Total Lines 20
Code Lines 11

Duplication

Lines 0
Ratio 0 %

Importance

Changes 1
Bugs 0 Features 0
Metric Value
cc 5
eloc 11
c 1
b 0
f 0
nc 5
nop 0
dl 0
loc 20
rs 9.6111
1
<?php
2
3
namespace Sunnysideup\VersionPruner\Api;
4
5
use SilverStripe\Core\ClassInfo;
6
use SilverStripe\Core\Config\Config;
7
use SilverStripe\Core\Injector\Injector;
8
use SilverStripe\Dev\BuildTask;
9
use SilverStripe\ORM\DataObject;
10
use SilverStripe\ORM\DB;
11
use SilverStripe\Versioned\Versioned;
0 ignored issues
show
Bug introduced by
The type SilverStripe\Versioned\Versioned was not found. Maybe you did not declare it correctly or list all dependencies?

The issue could also be caused by a filter entry in the build configuration. If the path has been excluded in your configuration, e.g. excluded_paths: ["lib/*"], you can move it to the dependency path list as follows:

filter:
    dependency_paths: ["lib/*"]

For further information see https://scrutinizer-ci.com/docs/tools/php/php-scrutinizer/#list-dependency-paths

Loading history...
12
13
class PruneAllVersionedRecordsReviewTemplates extends BuildTask
14
{
15
    /**
16
     * @var string
17
     */
18
    protected $title = 'Prune all versioned records - review templates for each dataobject';
19
20
    protected $description = 'Go through all dataobjects and shows the pruning schedule.';
21
22
    /**
23
     * @var string
24
     */
25
    private static $segment = 'prune-all-versioned-records-review-templates';
26
27
    /**
28
     * Prune all published DataObjects which are published according to config.
29
     *
30
     * @param mixed $request
31
     */
32
    public function run($request)
33
    {
34
        $allClasses = ClassInfo::subclassesFor(DataObject::class, false);
35
        $runner = RunForOneObject::inst();
36
        Versioned::set_stage(Versioned::DRAFT);
37
        foreach ($allClasses as $className) {
38
            $name = Injector::inst()->get($className)->i18n_singular_name();
39
            $count = $this->getObjectCountPerClassName($className);
40
            $versionCount = $this->getObjectCountPerClassName($className);
41
            if ($count) {
42
                $object = DataObject::get_one($className);
43
                if ($object) {
44
                    $array = $runner->getTemplatesDescription($object);
45
                    if (count($array)) {
46
                        DB::alteration_message('-----------------------------------');
47
                        DB::alteration_message($name . ' (' . $count . ' records, '.$versionCount.' version records) ' . $className);
48
                        DB::alteration_message('... ' . $className);
49
                        foreach ($array as $string) {
50
                            DB::alteration_message('... ... ' . $string);
51
                        }
52
                    }
53
                    $array = $runner->getTableSizes($object, true);
54
                    if(! empty($array)) {
55
                        DB::alteration_message('... Version Records');
56
                        foreach($array as $table => $size) {
57
                            DB::alteration_message('... ... ' . $table.': '. number_format($size));
58
                        }
59
                    }
60
                }
61
            }
62
        }
63
    }
64
65
    protected $objectCountPerClassNameCache = [];
66
67
    protected $objectCountForVersionsPerClassNameCache = [];
68
69
    protected function getObjectCountPerClassName(string $className): int
70
    {
71
        if(! isset($this->objectCountPerClassNameCache[$className])) {
72
            $this->objectCountPerClassNameCache[$className] = $className::get()->limit(100000)->count();
73
        }
74
        return $this->objectCountPerClassNameCache[$className];
75
    }
76
77
    protected function getObjectCountForVersionsPerClassName(string $className): int
78
    {
79
        if(! isset($this->objectCountForVersionsPerClassNameCache[$className])) {
80
            $tableName = Config::inst()->get($className, 'table_name');
81
            $this->objectCountForVersionsPerClassNameCache[$className] = (int) DB::query('SELECT COUNT("ID") FROM "'.$tableName.'_Versions";')->value();
82
        }
83
        return $this->objectCountForVersionsPerClassNameCache[$className];
84
    }
85
86
    /**
87
     * Get all versioned database classes.
88
     */
89
    protected function getAllVersionedDataClassesBase(): array
90
    {
91
        $allClasses = ClassInfo::subclassesFor(DataObject::class);
92
        $versionedClasses = [];
93
        foreach ($allClasses as $className) {
94
            if (DataObject::has_extension($className, Versioned::class)) {
95
                $ancestors = ClassInfo::ancestry($className);
96
                foreach ($ancestors as $classNameInner) {
97
                    if (DataObject::has_extension($classNameInner, Versioned::class)) {
98
                        $versionedClasses[$classNameInner] = $classNameInner;
99
100
                        continue 2;
101
                    }
102
                }
103
104
                $versionedClasses[$className] = $className;
105
            }
106
        }
107
108
        return $versionedClasses;
109
    }
110
}
111