devel_generate.module 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522
  1. <?php
  2. /**
  3. * Implements hook_menu().
  4. */
  5. function devel_generate_menu() {
  6. $items = array();
  7. $items['admin/config/development/generate/user'] = array(
  8. 'title' => 'Generate users',
  9. 'description' => 'Generate a given number of users. Optionally delete current users.',
  10. 'page callback' => 'drupal_get_form',
  11. 'page arguments' => array('devel_generate_users_form'),
  12. 'access arguments' => array('administer users'),
  13. );
  14. $items['admin/config/development/generate/content'] = array(
  15. 'title' => 'Generate content',
  16. 'description' => 'Generate a given number of nodes and comments. Optionally delete current items.',
  17. 'page callback' => 'drupal_get_form',
  18. 'page arguments' => array('devel_generate_content_form'),
  19. 'access arguments' => array('administer nodes'),
  20. );
  21. if (module_exists('taxonomy')) {
  22. $items['admin/config/development/generate/taxonomy'] = array(
  23. 'title' => 'Generate terms',
  24. 'description' => 'Generate a given number of terms. Optionally delete current terms.',
  25. 'page callback' => 'drupal_get_form',
  26. 'page arguments' => array('devel_generate_term_form'),
  27. 'access arguments' => array('administer taxonomy'),
  28. );
  29. $items['admin/config/development/generate/vocabs'] = array(
  30. 'title' => 'Generate vocabularies',
  31. 'description' => 'Generate a given number of vocabularies. Optionally delete current vocabularies.',
  32. 'page callback' => 'drupal_get_form',
  33. 'page arguments' => array('devel_generate_vocab_form'),
  34. 'access arguments' => array('administer taxonomy'),
  35. );
  36. }
  37. $items['admin/config/development/generate/menu'] = array(
  38. 'title' => 'Generate menus',
  39. 'description' => 'Generate a given number of menus and menu links. Optionally delete current menus.',
  40. 'page callback' => 'drupal_get_form',
  41. 'page arguments' => array('devel_generate_menu_form'),
  42. 'access arguments' => array('administer menu'),
  43. );
  44. return $items;
  45. }
  46. /**
  47. * Generates users using FormAPI.
  48. */
  49. function devel_generate_users_form() {
  50. $form['num'] = array(
  51. '#type' => 'textfield',
  52. '#title' => t('How many users would you like to generate?'),
  53. '#default_value' => 50,
  54. '#size' => 10,
  55. );
  56. $form['kill_users'] = array(
  57. '#type' => 'checkbox',
  58. '#title' => t('Delete all users (except user id 1) before generating new users.'),
  59. '#default_value' => FALSE,
  60. );
  61. $options = user_roles(TRUE);
  62. unset($options[DRUPAL_AUTHENTICATED_RID]);
  63. $form['roles'] = array(
  64. '#type' => 'checkboxes',
  65. '#title' => t('Which roles should the users receive?'),
  66. '#description' => t('Users always receive the <em>authenticated user</em> role.'),
  67. '#options' => $options,
  68. );
  69. $form['pass'] = array(
  70. '#type' => 'textfield',
  71. '#title' => t('Password to be set'),
  72. '#default_value' => NULL,
  73. '#size' => 32,
  74. '#description' => t('Leave this field empty if you do not need to set a password.'),
  75. );
  76. $options = array(1 => t('Now'));
  77. foreach (array(3600, 86400, 604800, 2592000, 31536000) as $interval) {
  78. $options[$interval] = format_interval($interval, 1) . ' ' . t('ago');
  79. }
  80. $form['time_range'] = array(
  81. '#type' => 'select',
  82. '#title' => t('How old should user accounts be?'),
  83. '#description' => t('User ages will be distributed randomly from the current time, back to the selected time.'),
  84. '#options' => $options,
  85. '#default_value' => 604800,
  86. );
  87. $form['submit'] = array(
  88. '#type' => 'submit',
  89. '#value' => t('Generate'),
  90. );
  91. return $form;
  92. }
  93. /**
  94. * FormAPI submission to generate users.
  95. */
  96. function devel_generate_users_form_submit($form, &$form_state) {
  97. module_load_include('inc', 'devel_generate');
  98. $values = $form_state['values'];
  99. devel_create_users($values['num'], $values['kill_users'], $values['time_range'], $values['roles'], $values['pass']);
  100. }
  101. /**
  102. * Generates nodes using FormAPI.
  103. */
  104. function devel_generate_content_form() {
  105. $options = array();
  106. if (module_exists('content')) {
  107. $types = content_types();
  108. foreach ($types as $type) {
  109. $warn = '';
  110. if (count($type['fields'])) {
  111. $warn = t('. This type contains CCK fields which will only be populated by fields that implement the content_generate hook.');
  112. }
  113. $options[$type['type']] = t($type['name']). $warn;
  114. }
  115. }
  116. else {
  117. $types = node_type_get_types();
  118. foreach ($types as $type) {
  119. $options[$type->type] = array(
  120. 'type' => t($type->name),
  121. );
  122. if (module_exists('comment')) {
  123. $default = variable_get('comment_' . $type->type, COMMENT_NODE_OPEN);
  124. $map = array(t('Hidden'), t('Closed'), t('Open'));
  125. $options[$type->type]['comments'] = '<small>'. $map[$default]. '</small>';
  126. }
  127. }
  128. }
  129. // we cannot currently generate valid polls.
  130. unset($options['poll']);
  131. if (empty($options)) {
  132. drupal_set_message(t('You do not have any content types that can be generated. <a href="@create-type">Go create a new content type</a> already!</a>', array('@create-type' => url('admin/structure/types/add'))), 'error', FALSE);
  133. return;
  134. }
  135. $header = array(
  136. 'type' => t('Content type'),
  137. );
  138. if (module_exists('comment')) {
  139. $header['comments'] = t('Comments');
  140. }
  141. $form['node_types'] = array(
  142. '#type' => 'tableselect',
  143. '#header' => $header,
  144. '#options' => $options,
  145. '#required' => TRUE,
  146. );
  147. if (module_exists('checkall')) $form['node_types']['#checkall'] = TRUE;
  148. $form['kill_content'] = array(
  149. '#type' => 'checkbox',
  150. '#title' => t('<strong>Delete all content</strong> in these content types before generating new content.'),
  151. '#default_value' => FALSE,
  152. );
  153. $form['num_nodes'] = array(
  154. '#type' => 'textfield',
  155. '#title' => t('How many nodes would you like to generate?'),
  156. '#default_value' => 50,
  157. '#size' => 10,
  158. );
  159. $options = array(1 => t('Now'));
  160. foreach (array(3600, 86400, 604800, 2592000, 31536000) as $interval) {
  161. $options[$interval] = format_interval($interval, 1) . ' ' . t('ago');
  162. }
  163. $form['time_range'] = array(
  164. '#type' => 'select',
  165. '#title' => t('How far back in time should the nodes be dated?'),
  166. '#description' => t('Node creation dates will be distributed randomly from the current time, back to the selected time.'),
  167. '#options' => $options,
  168. '#default_value' => 604800,
  169. );
  170. $form['max_comments'] = array(
  171. '#type' => module_exists('comment') ? 'textfield' : 'value',
  172. '#title' => t('Maximum number of comments per node.'),
  173. '#description' => t('You must also enable comments for the content types you are generating. Note that some nodes will randomly receive zero comments. Some will receive the max.'),
  174. '#default_value' => 0,
  175. '#size' => 3,
  176. '#access' => module_exists('comment'),
  177. );
  178. $form['title_length'] = array(
  179. '#type' => 'textfield',
  180. '#title' => t('Maximum number of words in titles'),
  181. '#default_value' => 4,
  182. '#size' => 10,
  183. );
  184. $form['add_alias'] = array(
  185. '#type' => 'checkbox',
  186. '#disabled' => !module_exists('path'),
  187. '#description' => t('Requires path.module'),
  188. '#title' => t('Add an url alias for each node.'),
  189. '#default_value' => FALSE,
  190. );
  191. $form['add_statistics'] = array(
  192. '#type' => 'checkbox',
  193. '#title' => t('Add statistics for each node (node_counter table).'),
  194. '#default_value' => TRUE,
  195. '#access' => module_exists('statistics'),
  196. );
  197. unset($options);
  198. $options[LANGUAGE_NONE] = t('Language neutral');
  199. if (module_exists('locale')) {
  200. $options += locale_language_list();
  201. }
  202. $form['add_language'] = array(
  203. '#type' => 'select',
  204. '#title' => t('Set language on nodes'),
  205. '#multiple' => TRUE,
  206. '#disabled' => !module_exists('locale'),
  207. '#description' => t('Requires locale.module'),
  208. '#options' => $options,
  209. '#default_value' => array(LANGUAGE_NONE),
  210. );
  211. $form['submit'] = array(
  212. '#type' => 'submit',
  213. '#value' => t('Generate'),
  214. );
  215. $form['#redirect'] = FALSE;
  216. return $form;
  217. }
  218. /**
  219. * FormAPI validation before generate nodes.
  220. */
  221. function devel_generate_content_form_validate($form, &$form_state) {
  222. form_set_value($form['node_types'], array_filter($form_state['values']['node_types']) , $form_state);
  223. if (!empty($form_state['values']['kill_content']) && empty($form_state['values']['node_types'])) {
  224. form_set_error('', t('Please select at least one content type to delete the content.'));
  225. }
  226. }
  227. /**
  228. * FormAPI submission to generate nodes.
  229. */
  230. function devel_generate_content_form_submit($form, &$form_state) {
  231. module_load_include('inc', 'devel_generate', 'devel_generate');
  232. if ($form_state['values']['num_nodes'] <= 50 && $form_state['values']['max_comments'] <= 10) {
  233. module_load_include('inc', 'devel_generate');
  234. devel_generate_content($form_state);
  235. }
  236. else {
  237. module_load_include('inc', 'devel_generate', 'devel_generate_batch');
  238. devel_generate_batch_content($form_state);
  239. }
  240. }
  241. /**
  242. * Generates taxonomy terms using FormAPI.
  243. */
  244. function devel_generate_term_form() {
  245. $options = array();
  246. foreach (taxonomy_get_vocabularies() as $vid => $vocab) {
  247. $options[$vid] = $vocab->machine_name;
  248. }
  249. $form['vids'] = array(
  250. '#type' => 'select',
  251. '#multiple' => TRUE,
  252. '#title' => t('Vocabularies'),
  253. '#required' => TRUE,
  254. '#options' => $options,
  255. '#description' => t('Restrict terms to these vocabularies.'),
  256. );
  257. $form['num_terms'] = array(
  258. '#type' => 'textfield',
  259. '#title' => t('Number of terms?'),
  260. '#default_value' => 10,
  261. '#size' => 10,
  262. );
  263. $form['title_length'] = array(
  264. '#type' => 'textfield',
  265. '#title' => t('Maximum number of characters in term names'),
  266. '#default_value' => 12,
  267. '#size' => 10,
  268. );
  269. $form['kill_taxonomy'] = array(
  270. '#type' => 'checkbox',
  271. '#title' => t('Delete existing terms in specified vocabularies before generating new terms.'),
  272. '#default_value' => FALSE,
  273. );
  274. $form['submit'] = array(
  275. '#type' => 'submit',
  276. '#value' => t('Generate'),
  277. );
  278. return $form;
  279. }
  280. /**
  281. * Generates taxonomy vocabularies using FormAPI.
  282. */
  283. function devel_generate_vocab_form() {
  284. $form['num_vocabs'] = array(
  285. '#type' => 'textfield',
  286. '#title' => t('Number of vocabularies?'),
  287. '#default_value' => 1,
  288. '#size' => 10,
  289. );
  290. $form['title_length'] = array(
  291. '#type' => 'textfield',
  292. '#title' => t('Maximum number of characters in vocabulary names'),
  293. '#default_value' => 12,
  294. '#size' => 10,
  295. );
  296. $form['kill_taxonomy'] = array(
  297. '#type' => 'checkbox',
  298. '#title' => t('Delete existing vocabularies before generating new ones.'),
  299. '#default_value' => FALSE,
  300. );
  301. $form['submit'] = array(
  302. '#type' => 'submit',
  303. '#value' => t('Generate'),
  304. );
  305. return $form;
  306. }
  307. /**
  308. * FormAPI submission to generate taxonomy terms.
  309. */
  310. function devel_generate_term_form_submit($form, &$form_state) {
  311. $values = $form_state['values'];
  312. module_load_include('inc', 'devel_generate');
  313. if ($values['kill_taxonomy']) {
  314. foreach ($values['vids'] as $vid) {
  315. devel_generate_delete_vocabulary_terms($vid);
  316. }
  317. drupal_set_message(t('Deleted existing terms.'));
  318. }
  319. $vocabs = taxonomy_vocabulary_load_multiple($values['vids']);
  320. $new_terms = devel_generate_terms($values['num_terms'], $vocabs, $values['title_length']);
  321. if (!empty($new_terms)) {
  322. drupal_set_message(t('Created the following new terms: !terms', array('!terms' => implode(', ', $new_terms))));
  323. }
  324. }
  325. /**
  326. * FormAPI submission to generate taxonomy vocabularies.
  327. */
  328. function devel_generate_vocab_form_submit($form, &$form_state) {
  329. $values = $form_state['values'];
  330. module_load_include('inc', 'devel_generate');
  331. if ($values['kill_taxonomy']) {
  332. devel_generate_delete_vocabularies();
  333. drupal_set_message(t('Deleted existing vocabularies.'));
  334. }
  335. $new_vocs = devel_generate_vocabs($values['num_vocabs'], $values['title_length']);
  336. if (!empty($new_vocs)) {
  337. drupal_set_message(t('Created the following new vocabularies: !vocs', array('!vocs' => implode(', ', $new_vocs))));
  338. }
  339. }
  340. /**
  341. * Inserts nodes properly based on generation options.
  342. *
  343. * @param $node
  344. * The base node created on submit. Inspects $node->devel_generate.
  345. */
  346. function devel_generate_node_insert($node) {
  347. if (isset($node->devel_generate)) {
  348. $results = $node->devel_generate;
  349. if (!empty($results['max_comments']) && $node->comment >= COMMENT_NODE_OPEN) {
  350. devel_generate_add_comments($node, $results['users'], $results['max_comments'], $results['title_length']);
  351. }
  352. // Add an url alias. Cannot happen before save because we don't know the nid.
  353. if (!empty($results['add_alias'])) {
  354. $path = array(
  355. 'source' => 'node/' . $node->nid,
  356. 'alias' => 'node-' . $node->nid . '-' . $node->type,
  357. );
  358. path_save($path);
  359. }
  360. // Add node statistics.
  361. if (!empty($results['add_statistics']) && module_exists('statistics')) {
  362. devel_generate_add_statistics($node);
  363. }
  364. }
  365. }
  366. /**
  367. * Set a message for either drush or the web interface.
  368. *
  369. * @param $msg
  370. * The message to display.
  371. * @param $type
  372. * The message type, as defined by drupal_set_message().
  373. *
  374. * @return
  375. * Context-appropriate message output.
  376. */
  377. function devel_generate_set_message($msg, $type = 'status') {
  378. $function = function_exists('drush_log') ? 'drush_log' : 'drupal_set_message';
  379. $function($msg, $type);
  380. }
  381. /**
  382. * Generates menus using FormAPI.
  383. */
  384. function devel_generate_menu_form() {
  385. $menu_enabled = module_exists('menu');
  386. if ($menu_enabled) {
  387. $menus = array('__new-menu__' => t('Create new menu(s)')) + menu_get_menus();
  388. }
  389. else {
  390. $menus = menu_list_system_menus();
  391. }
  392. $form['existing_menus'] = array(
  393. '#type' => 'checkboxes',
  394. '#title' => t('Generate links for these menus'),
  395. '#options' => $menus,
  396. '#default_value' => array('__new-menu__'),
  397. '#required' => TRUE,
  398. );
  399. if ($menu_enabled) {
  400. $form['num_menus'] = array(
  401. '#type' => 'textfield',
  402. '#title' => t('Number of new menus to create'),
  403. '#default_value' => 2,
  404. '#size' => 10,
  405. '#states' => array(
  406. 'visible' => array(
  407. ':input[name=existing_menus[__new-menu__]]' => array('checked' => TRUE),
  408. ),
  409. ),
  410. );
  411. }
  412. $form['num_links'] = array(
  413. '#type' => 'textfield',
  414. '#title' => t('Number of links to generate'),
  415. '#default_value' => 50,
  416. '#size' => 10,
  417. '#required' => TRUE,
  418. );
  419. $form['title_length'] = array(
  420. '#type' => 'textfield',
  421. '#title' => t('Maximum number of characters in menu and menu link names'),
  422. '#description' => t("The minimum length is 2."),
  423. '#default_value' => 12,
  424. '#size' => 10,
  425. '#required' => TRUE,
  426. );
  427. $form['link_types'] = array(
  428. '#type' => 'checkboxes',
  429. '#title' => t('Types of links to generate'),
  430. '#options' => array(
  431. 'node' => t('Nodes'),
  432. 'front' => t('Front page'),
  433. 'external' => t('External'),
  434. ),
  435. '#default_value' => array('node', 'front', 'external'),
  436. '#required' => TRUE,
  437. );
  438. $form['max_depth'] = array(
  439. '#type' => 'select',
  440. '#title' => t('Maximum link depth'),
  441. '#options' => range(0, MENU_MAX_DEPTH),
  442. '#default_value' => floor(MENU_MAX_DEPTH / 2),
  443. '#required' => TRUE,
  444. );
  445. unset($form['max_depth']['#options'][0]);
  446. $form['max_width'] = array(
  447. '#type' => 'textfield',
  448. '#title' => t('Maximum menu width'),
  449. '#default_value' => 6,
  450. '#size' => 10,
  451. '#description' => t("Limit the width of the generated menu's first level of links to a certain number of items."),
  452. '#required' => TRUE,
  453. );
  454. $form['kill'] = array(
  455. '#type' => 'checkbox',
  456. '#title' => t('Delete existing custom generated menus and menu links before generating new ones.'),
  457. '#default_value' => FALSE,
  458. );
  459. $form['submit'] = array(
  460. '#type' => 'submit',
  461. '#value' => t('Generate'),
  462. );
  463. return $form;
  464. }
  465. /**
  466. * FormAPI submission to generate menus.
  467. */
  468. function devel_generate_menu_form_submit($form, &$form_state) {
  469. // If the create new menus checkbox is off, set the number of new menus to 0.
  470. if (!isset($form_state['values']['existing_menus']['__new-menu__']) || !$form_state['values']['existing_menus']['__new-menu__']) {
  471. $form_state['values']['num_menus'] = 0;
  472. }
  473. module_load_include('inc', 'devel_generate');
  474. // Delete custom menus.
  475. if ($form_state['values']['kill']) {
  476. devel_generate_delete_menus();
  477. drupal_set_message(t('Deleted existing menus and links.'));
  478. }
  479. // Generate new menus.
  480. $new_menus = devel_generate_menus($form_state['values']['num_menus'], $form_state['values']['title_length']);
  481. if (!empty($new_menus)) {
  482. drupal_set_message(t('Created the following new menus: !menus', array('!menus' => implode(', ', $new_menus))));
  483. }
  484. // Generate new menu links.
  485. $menus = $new_menus + $form_state['values']['existing_menus'];
  486. $new_links = devel_generate_links($form_state['values']['num_links'], $menus, $form_state['values']['title_length'], $form_state['values']['link_types'], $form_state['values']['max_depth'], $form_state['values']['max_width']);
  487. drupal_set_message(t('Created @count new menu links.', array('@count' => count($new_links))));
  488. }