Added source type.

wpsmith · May 24, 2012 · a801473 · a801473
1 parent db3f9f3
commit a801473
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tgm-plugin-activation/class-tgm-plugin-activation.php b/tgm-plugin-activation/class-tgm-plugin-activation.php
@@ -1196,7 +1196,11 @@ protected function _gather_plugin_data() {
 				if ( isset( $table_data[$i]['plugin'] ) && (array) $table_data[$i]['plugin'] )
 					$plugin['name'] = $table_data[$i]['plugin'];
 
-				if ( isset( $plugin['external_url'] ) ) {
+				if ( isset( $plugin['source_type'] ) ) {
+					/** The plugin is linked to an external source */
+					$table_data[$i]['source'] = $plugin['source_type'];
+				}
+				elseif ( isset( $plugin['external_url'] ) ) {
 					/** The plugin is linked to an external source */
 					$table_data[$i]['source'] = __( 'External Link', TGM_Plugin_Activation::$instance->domain );
 				}