dnadesign / silverstripe-elemental

code/extensions/ElementPublishChildren.php

<?php

/**
 * @package elemental
 */
class ElementPublishChildren extends DataExtension

PSR1 recommends that each class must be in a namespace of at least one level to avoid collisions.

{

    public function onBeforeVersionedPublish()
    {
        $staged = array();

        foreach ($this->owner->Elements() as $widget) {
            $staged[] = $widget->ID;

            $widget->publish('Stage', 'Live');
        }

        if($this->owner->ID) {

            // remove any elements that are on live but not in draft.
            $widgets = Versioned::get_by_stage('BaseElement', 'Live', sprintf(
                "ListID = '%s'", $this->owner->ID
            ));

            foreach ($widgets as $widget) {
                if (!in_array($widget->ID, $staged)) {
                    $widget->deleteFromStage('Live');
                }
            }
        }
    }
}