EntryTypeSourcesPlugin.php

<?php
namespace Craft;

class EntryTypeSourcesPlugin extends BasePlugin
{
	public function getName()
	{
		return Craft::t('Entry Type Sources');
	}

	public function getVersion()
	{
		return '1.1';
	}

	public function getDeveloper()
	{
		return 'Mike Pepper';
	}

	public function getDeveloperUrl()
	{
		return 'http://twitter.com/mdcpepper';
	}

	protected function defineSettings()
	{
		return array(
			'entryTypes' => array(AttributeType::Mixed, 'default' => array()),
		);
	}

	public function getSettingsHtml()
	{
		$entryTypes = $this->_getEntryTypes()->queryAll();

		$selectedEntryTypes = array();

		foreach ($this->getSettings()->entryTypes as $entryType)
		{
			$selectedEntryTypes[] = $entryType['value'];
		}

		return craft()->templates->renderMacro('_includes/forms', 'checkboxSelectField', array(
			array(
				'first' => true,
				'label' => 'Entry Types',
				'instructions' => 'Choose which entry types should be included in entry source lists.',
				'name' => 'entryTypes',
				'options' => $entryTypes,
				'values' => $selectedEntryTypes,
			)
		));
	}

	public function prepSettings($settings)
	{
		if ('*' == $settings['entryTypes'])
		{
			$settings['entryTypes'] = $this->_getEntryTypes()->queryAll();
		}
		else
		{
			$settings['entryTypes'] = $this->_getEntryTypes()
			    ->where(array('in', 'entrytypes.id', $settings['entryTypes']))
			    ->queryAll();
		}

		return $settings;
	}

	public function modifyEntrySources(&$sources, $context)
	{
		$settings = $this->getSettings();
		$entryTypesBySectionId = array();

		foreach ($settings->entryTypes as $entryType)
		{
			$entryTypesBySectionId[$entryType['sectionId']][] = $entryType;
		}

		foreach ($sources as $sourceKey => $source)
		{
			if (strpos($sourceKey, 'section:') === 0)
			{
				list($section, $sectionId) = explode(':', $sourceKey);

				if (array_key_exists($sectionId, $entryTypesBySectionId))
				{
					if (!isset($sources[$sourceKey]['nested']))
					{
						$sources[$sourceKey]['nested'] = array();
					}

					foreach($entryTypesBySectionId[$sectionId] as $entryType)
					{
						$sources[$sourceKey]['nested']['entrytype:'.$entryType['value']] = array(
							'label'    => $entryType['label'],
							'criteria' => array('type' => $entryType['value'], 'editable' => true),
						);
					}
				}
			}
		}
	}

	private function _getEntryTypes()
	{
		return craft()->db->createCommand()
		    ->select('entrytypes.id as value, entrytypes.name as label, entrytypes.sectionId')
		    ->from('entrytypes entrytypes')
		    ->join('sections sections', 'sections.id = entrytypes.sectionId')
		    ->where("sections.type != 'single'");
	}
}