Poll.php 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970
  1. <?php
  2. /**
  3. * This file contains the functions for voting, locking, removing and
  4. * editing polls. Note that that posting polls is done in Post.php.
  5. *
  6. * Simple Machines Forum (SMF)
  7. *
  8. * @package SMF
  9. * @author Simple Machines http://www.simplemachines.org
  10. * @copyright 2011 Simple Machines
  11. * @license http://www.simplemachines.org/about/smf/license.php BSD
  12. *
  13. * @version 2.1 Alpha 1
  14. */
  15. if (!defined('SMF'))
  16. die('Hacking attempt...');
  17. /**
  18. * Allow the user to vote.
  19. * It is called to register a vote in a poll.
  20. * Must be called with a topic and option specified.
  21. * Requires the poll_vote permission.
  22. * Upon successful completion of action will direct user back to topic.
  23. * Accessed via ?action=vote.
  24. *
  25. * @uses Post language file.
  26. */
  27. function Vote()
  28. {
  29. global $topic, $txt, $user_info, $smcFunc, $sourcedir, $modSettings;
  30. // Make sure you can vote.
  31. isAllowedTo('poll_vote');
  32. loadLanguage('Post');
  33. // Check if they have already voted, or voting is locked.
  34. $request = $smcFunc['db_query']('', '
  35. SELECT IFNULL(lp.id_choice, -1) AS selected, p.voting_locked, p.id_poll, p.expire_time, p.max_votes, p.change_vote,
  36. p.guest_vote, p.reset_poll, p.num_guest_voters
  37. FROM {db_prefix}topics AS t
  38. INNER JOIN {db_prefix}polls AS p ON (p.id_poll = t.id_poll)
  39. LEFT JOIN {db_prefix}log_polls AS lp ON (p.id_poll = lp.id_poll AND lp.id_member = {int:current_member} AND lp.id_member != {int:not_guest})
  40. WHERE t.id_topic = {int:current_topic}
  41. LIMIT 1',
  42. array(
  43. 'current_member' => $user_info['id'],
  44. 'current_topic' => $topic,
  45. 'not_guest' => 0,
  46. )
  47. );
  48. if ($smcFunc['db_num_rows']($request) == 0)
  49. fatal_lang_error('poll_error', false);
  50. $row = $smcFunc['db_fetch_assoc']($request);
  51. $smcFunc['db_free_result']($request);
  52. // If this is a guest can they vote?
  53. if ($user_info['is_guest'])
  54. {
  55. // Guest voting disabled?
  56. if (!$row['guest_vote'])
  57. fatal_lang_error('guest_vote_disabled');
  58. // Guest already voted?
  59. elseif (!empty($_COOKIE['guest_poll_vote']) && preg_match('~^[0-9,;]+$~', $_COOKIE['guest_poll_vote']) && strpos($_COOKIE['guest_poll_vote'], ';' . $row['id_poll'] . ',') !== false)
  60. {
  61. // ;id,timestamp,[vote,vote...]; etc
  62. $guestinfo = explode(';', $_COOKIE['guest_poll_vote']);
  63. // Find the poll we're after.
  64. foreach ($guestinfo as $i => $guestvoted)
  65. {
  66. $guestvoted = explode(',', $guestvoted);
  67. if ($guestvoted[0] == $row['id_poll'])
  68. break;
  69. }
  70. // Has the poll been reset since guest voted?
  71. if ($row['reset_poll'] > $guestvoted[1])
  72. {
  73. // Remove the poll info from the cookie to allow guest to vote again
  74. unset($guestinfo[$i]);
  75. if (!empty($guestinfo))
  76. $_COOKIE['guest_poll_vote'] = ';' . implode(';', $guestinfo);
  77. else
  78. unset($_COOKIE['guest_poll_vote']);
  79. }
  80. else
  81. fatal_lang_error('poll_error', false);
  82. unset($guestinfo, $guestvoted, $i);
  83. }
  84. }
  85. // Is voting locked or has it expired?
  86. if (!empty($row['voting_locked']) || (!empty($row['expire_time']) && time() > $row['expire_time']))
  87. fatal_lang_error('poll_error', false);
  88. // If they have already voted and aren't allowed to change their vote - hence they are outta here!
  89. if (!$user_info['is_guest'] && $row['selected'] != -1 && empty($row['change_vote']))
  90. fatal_lang_error('poll_error', false);
  91. // Otherwise if they can change their vote yet they haven't sent any options... remove their vote and redirect.
  92. elseif (!empty($row['change_vote']) && !$user_info['is_guest'] && empty($_POST['options']))
  93. {
  94. checkSession('request');
  95. $pollOptions = array();
  96. // Find out what they voted for before.
  97. $request = $smcFunc['db_query']('', '
  98. SELECT id_choice
  99. FROM {db_prefix}log_polls
  100. WHERE id_member = {int:current_member}
  101. AND id_poll = {int:id_poll}',
  102. array(
  103. 'current_member' => $user_info['id'],
  104. 'id_poll' => $row['id_poll'],
  105. )
  106. );
  107. while ($choice = $smcFunc['db_fetch_row']($request))
  108. $pollOptions[] = $choice[0];
  109. $smcFunc['db_free_result']($request);
  110. // Just skip it if they had voted for nothing before.
  111. if (!empty($pollOptions))
  112. {
  113. // Update the poll totals.
  114. $smcFunc['db_query']('', '
  115. UPDATE {db_prefix}poll_choices
  116. SET votes = votes - 1
  117. WHERE id_poll = {int:id_poll}
  118. AND id_choice IN ({array_int:poll_options})
  119. AND votes > {int:votes}',
  120. array(
  121. 'poll_options' => $pollOptions,
  122. 'id_poll' => $row['id_poll'],
  123. 'votes' => 0,
  124. )
  125. );
  126. // Delete off the log.
  127. $smcFunc['db_query']('', '
  128. DELETE FROM {db_prefix}log_polls
  129. WHERE id_member = {int:current_member}
  130. AND id_poll = {int:id_poll}',
  131. array(
  132. 'current_member' => $user_info['id'],
  133. 'id_poll' => $row['id_poll'],
  134. )
  135. );
  136. }
  137. // Redirect back to the topic so the user can vote again!
  138. if (empty($_POST['options']))
  139. redirectexit('topic=' . $topic . '.' . $_REQUEST['start']);
  140. }
  141. checkSession('request');
  142. // Make sure the option(s) are valid.
  143. if (empty($_POST['options']))
  144. fatal_lang_error('didnt_select_vote', false);
  145. // Too many options checked!
  146. if (count($_REQUEST['options']) > $row['max_votes'])
  147. fatal_lang_error('poll_too_many_votes', false, array($row['max_votes']));
  148. $pollOptions = array();
  149. $inserts = array();
  150. foreach ($_REQUEST['options'] as $id)
  151. {
  152. $id = (int) $id;
  153. $pollOptions[] = $id;
  154. $inserts[] = array($row['id_poll'], $user_info['id'], $id);
  155. }
  156. // Add their vote to the tally.
  157. $smcFunc['db_insert']('insert',
  158. '{db_prefix}log_polls',
  159. array('id_poll' => 'int', 'id_member' => 'int', 'id_choice' => 'int'),
  160. $inserts,
  161. array('id_poll', 'id_member', 'id_choice')
  162. );
  163. $smcFunc['db_query']('', '
  164. UPDATE {db_prefix}poll_choices
  165. SET votes = votes + 1
  166. WHERE id_poll = {int:id_poll}
  167. AND id_choice IN ({array_int:poll_options})',
  168. array(
  169. 'poll_options' => $pollOptions,
  170. 'id_poll' => $row['id_poll'],
  171. )
  172. );
  173. // If it's a guest don't let them vote again.
  174. if ($user_info['is_guest'] && count($pollOptions) > 0)
  175. {
  176. // Time is stored in case the poll is reset later, plus what they voted for.
  177. $_COOKIE['guest_poll_vote'] = empty($_COOKIE['guest_poll_vote']) ? '' : $_COOKIE['guest_poll_vote'];
  178. // ;id,timestamp,[vote,vote...]; etc
  179. $_COOKIE['guest_poll_vote'] .= ';' . $row['id_poll'] . ',' . time() . ',' . (count($pollOptions) > 1 ? explode(',' . $pollOptions) : $pollOptions[0]);
  180. // Increase num guest voters count by 1
  181. $smcFunc['db_query']('', '
  182. UPDATE {db_prefix}polls
  183. SET num_guest_voters = num_guest_voters + 1
  184. WHERE id_poll = {int:id_poll}',
  185. array(
  186. 'id_poll' => $row['id_poll'],
  187. )
  188. );
  189. require_once($sourcedir . '/Subs-Auth.php');
  190. $cookie_url = url_parts(!empty($modSettings['localCookies']), !empty($modSettings['globalCookies']));
  191. setcookie('guest_poll_vote', $_COOKIE['guest_poll_vote'], time() + 2500000, $cookie_url[1], $cookie_url[0], 0);
  192. }
  193. // Return to the post...
  194. redirectexit('topic=' . $topic . '.' . $_REQUEST['start']);
  195. }
  196. //
  197. /**
  198. * Lock the voting for a poll.
  199. * Must be called with a topic specified in the URL.
  200. * An admin always has over riding permission to lock a poll.
  201. * If not an admin must have poll_lock_any permission, otherwise must
  202. * be poll starter with poll_lock_own permission.
  203. * Upon successful completion of action will direct user back to topic.
  204. * Accessed via ?action=lockvoting.
  205. */
  206. function LockVoting()
  207. {
  208. global $topic, $user_info, $smcFunc;
  209. checkSession('get');
  210. // Get the poll starter, ID, and whether or not it is locked.
  211. $request = $smcFunc['db_query']('', '
  212. SELECT t.id_member_started, t.id_poll, p.voting_locked
  213. FROM {db_prefix}topics AS t
  214. INNER JOIN {db_prefix}polls AS p ON (p.id_poll = t.id_poll)
  215. WHERE t.id_topic = {int:current_topic}
  216. LIMIT 1',
  217. array(
  218. 'current_topic' => $topic,
  219. )
  220. );
  221. list ($memberID, $pollID, $voting_locked) = $smcFunc['db_fetch_row']($request);
  222. // If the user _can_ modify the poll....
  223. if (!allowedTo('poll_lock_any'))
  224. isAllowedTo('poll_lock_' . ($user_info['id'] == $memberID ? 'own' : 'any'));
  225. // It's been locked by a non-moderator.
  226. if ($voting_locked == '1')
  227. $voting_locked = '0';
  228. // Locked by a moderator, and this is a moderator.
  229. elseif ($voting_locked == '2' && allowedTo('moderate_board'))
  230. $voting_locked = '0';
  231. // Sorry, a moderator locked it.
  232. elseif ($voting_locked == '2' && !allowedTo('moderate_board'))
  233. fatal_lang_error('locked_by_admin', 'user');
  234. // A moderator *is* locking it.
  235. elseif ($voting_locked == '0' && allowedTo('moderate_board'))
  236. $voting_locked = '2';
  237. // Well, it's gonna be locked one way or another otherwise...
  238. else
  239. $voting_locked = '1';
  240. // Lock! *Poof* - no one can vote.
  241. $smcFunc['db_query']('', '
  242. UPDATE {db_prefix}polls
  243. SET voting_locked = {int:voting_locked}
  244. WHERE id_poll = {int:id_poll}',
  245. array(
  246. 'voting_locked' => $voting_locked,
  247. 'id_poll' => $pollID,
  248. )
  249. );
  250. redirectexit('topic=' . $topic . '.' . $_REQUEST['start']);
  251. }
  252. /**
  253. * Display screen for editing or adding a poll.
  254. * Must be called with a topic specified in the URL.
  255. * If the user is adding a poll to a topic, must contain the variable
  256. * 'add' in the url.
  257. * User must have poll_edit_any/poll_add_any permission for the
  258. * relevant action, otherwise must be poll starter with poll_edit_own
  259. * permission for editing, or be topic starter with poll_add_any permission for adding.
  260. * Accessed via ?action=editpoll.
  261. *
  262. * @uses Post language file.
  263. * @uses Poll template, main sub-template.
  264. */
  265. function EditPoll()
  266. {
  267. global $txt, $user_info, $context, $topic, $board, $smcFunc, $sourcedir, $scripturl;
  268. if (empty($topic))
  269. fatal_lang_error('no_access', false);
  270. loadLanguage('Post');
  271. loadTemplate('Poll');
  272. $context['can_moderate_poll'] = isset($_REQUEST['add']) ? 1 : allowedTo('moderate_board');
  273. $context['start'] = (int) $_REQUEST['start'];
  274. $context['is_edit'] = isset($_REQUEST['add']) ? 0 : 1;
  275. // Check if a poll currently exists on this topic, and get the id, question and starter.
  276. $request = $smcFunc['db_query']('', '
  277. SELECT
  278. t.id_member_started, p.id_poll, p.question, p.hide_results, p.expire_time, p.max_votes, p.change_vote,
  279. m.subject, p.guest_vote, p.id_member AS poll_starter
  280. FROM {db_prefix}topics AS t
  281. INNER JOIN {db_prefix}messages AS m ON (m.id_msg = t.id_first_msg)
  282. LEFT JOIN {db_prefix}polls AS p ON (p.id_poll = t.id_poll)
  283. WHERE t.id_topic = {int:current_topic}
  284. LIMIT 1',
  285. array(
  286. 'current_topic' => $topic,
  287. )
  288. );
  289. // Assume the the topic exists, right?
  290. if ($smcFunc['db_num_rows']($request) == 0)
  291. fatal_lang_error('no_board');
  292. // Get the poll information.
  293. $pollinfo = $smcFunc['db_fetch_assoc']($request);
  294. $smcFunc['db_free_result']($request);
  295. // If we are adding a new poll - make sure that there isn't already a poll there.
  296. if (!$context['is_edit'] && !empty($pollinfo['id_poll']))
  297. fatal_lang_error('poll_already_exists');
  298. // Otherwise, if we're editing it, it does exist I assume?
  299. elseif ($context['is_edit'] && empty($pollinfo['id_poll']))
  300. fatal_lang_error('poll_not_found');
  301. // Can you do this?
  302. if ($context['is_edit'] && !allowedTo('poll_edit_any'))
  303. isAllowedTo('poll_edit_' . ($user_info['id'] == $pollinfo['id_member_started'] || ($pollinfo['poll_starter'] != 0 && $user_info['id'] == $pollinfo['poll_starter']) ? 'own' : 'any'));
  304. elseif (!$context['is_edit'] && !allowedTo('poll_add_any'))
  305. isAllowedTo('poll_add_' . ($user_info['id'] == $pollinfo['id_member_started'] ? 'own' : 'any'));
  306. // Do we enable guest voting?
  307. require_once($sourcedir . '/Subs-Members.php');
  308. $groupsAllowedVote = groupsAllowedTo('poll_vote', $board);
  309. // Want to make sure before you actually submit? Must be a lot of options, or something.
  310. if (isset($_POST['preview']))
  311. {
  312. $question = $smcFunc['htmlspecialchars']($_POST['question']);
  313. // Basic theme info...
  314. $context['poll'] = array(
  315. 'id' => $pollinfo['id_poll'],
  316. 'question' => $question,
  317. 'hide_results' => empty($_POST['poll_hide']) ? 0 : $_POST['poll_hide'],
  318. 'change_vote' => isset($_POST['poll_change_vote']),
  319. 'guest_vote' => isset($_POST['poll_guest_vote']),
  320. 'guest_vote_allowed' => in_array(-1, $groupsAllowedVote['allowed']),
  321. 'max_votes' => empty($_POST['poll_max_votes']) ? '1' : max(1, $_POST['poll_max_votes']),
  322. );
  323. // Start at number one with no last id to speak of.
  324. $number = 1;
  325. $last_id = 0;
  326. // Get all the choices - if this is an edit.
  327. if ($context['is_edit'])
  328. {
  329. $request = $smcFunc['db_query']('', '
  330. SELECT label, votes, id_choice
  331. FROM {db_prefix}poll_choices
  332. WHERE id_poll = {int:id_poll}',
  333. array(
  334. 'id_poll' => $pollinfo['id_poll'],
  335. )
  336. );
  337. $context['choices'] = array();
  338. while ($row = $smcFunc['db_fetch_assoc']($request))
  339. {
  340. // Get the highest id so we can add more without reusing.
  341. if ($row['id_choice'] >= $last_id)
  342. $last_id = $row['id_choice'] + 1;
  343. // They cleared this by either omitting it or emptying it.
  344. if (!isset($_POST['options'][$row['id_choice']]) || $_POST['options'][$row['id_choice']] == '')
  345. continue;
  346. censorText($row['label']);
  347. // Add the choice!
  348. $context['choices'][$row['id_choice']] = array(
  349. 'id' => $row['id_choice'],
  350. 'number' => $number++,
  351. 'votes' => $row['votes'],
  352. 'label' => $row['label'],
  353. 'is_last' => false
  354. );
  355. }
  356. $smcFunc['db_free_result']($request);
  357. }
  358. // Work out how many options we have, so we get the 'is_last' field right...
  359. $totalPostOptions = 0;
  360. foreach ($_POST['options'] as $id => $label)
  361. if ($label != '')
  362. $totalPostOptions++;
  363. $count = 1;
  364. // If an option exists, update it. If it is new, add it - but don't reuse ids!
  365. foreach ($_POST['options'] as $id => $label)
  366. {
  367. $label = $smcFunc['htmlspecialchars']($label);
  368. censorText($label);
  369. if (isset($context['choices'][$id]))
  370. $context['choices'][$id]['label'] = $label;
  371. elseif ($label != '')
  372. $context['choices'][] = array(
  373. 'id' => $last_id++,
  374. 'number' => $number++,
  375. 'label' => $label,
  376. 'votes' => -1,
  377. 'is_last' => $count++ == $totalPostOptions && $totalPostOptions > 1 ? true : false,
  378. );
  379. }
  380. // Make sure we have two choices for sure!
  381. if ($totalPostOptions < 2)
  382. {
  383. // Need two?
  384. if ($totalPostOptions == 0)
  385. $context['choices'][] = array(
  386. 'id' => $last_id++,
  387. 'number' => $number++,
  388. 'label' => '',
  389. 'votes' => -1,
  390. 'is_last' => false
  391. );
  392. $poll_errors[] = 'poll_few';
  393. }
  394. // Always show one extra box...
  395. $context['choices'][] = array(
  396. 'id' => $last_id++,
  397. 'number' => $number++,
  398. 'label' => '',
  399. 'votes' => -1,
  400. 'is_last' => true
  401. );
  402. if ($context['can_moderate_poll'])
  403. $context['poll']['expiration'] = $_POST['poll_expire'];
  404. // Check the question/option count for errors.
  405. if (trim($_POST['question']) == '' && empty($context['poll_error']))
  406. $poll_errors[] = 'no_question';
  407. // No check is needed, since nothing is really posted.
  408. checkSubmitOnce('free');
  409. // Take a check for any errors... assuming we haven't already done so!
  410. if (!empty($poll_errors) && empty($context['poll_error']))
  411. {
  412. loadLanguage('Errors');
  413. $context['poll_error'] = array('messages' => array());
  414. foreach ($poll_errors as $poll_error)
  415. {
  416. $context['poll_error'][$poll_error] = true;
  417. $context['poll_error']['messages'][] = $txt['error_' . $poll_error];
  418. }
  419. }
  420. }
  421. else
  422. {
  423. // Basic theme info...
  424. $context['poll'] = array(
  425. 'id' => $pollinfo['id_poll'],
  426. 'question' => $pollinfo['question'],
  427. 'hide_results' => $pollinfo['hide_results'],
  428. 'max_votes' => $pollinfo['max_votes'],
  429. 'change_vote' => !empty($pollinfo['change_vote']),
  430. 'guest_vote' => !empty($pollinfo['guest_vote']),
  431. 'guest_vote_allowed' => in_array(-1, $groupsAllowedVote['allowed']),
  432. );
  433. // Poll expiration time?
  434. $context['poll']['expiration'] = empty($pollinfo['expire_time']) || !allowedTo('moderate_board') ? '' : ceil($pollinfo['expire_time'] <= time() ? -1 : ($pollinfo['expire_time'] - time()) / (3600 * 24));
  435. // Get all the choices - if this is an edit.
  436. if ($context['is_edit'])
  437. {
  438. $request = $smcFunc['db_query']('', '
  439. SELECT label, votes, id_choice
  440. FROM {db_prefix}poll_choices
  441. WHERE id_poll = {int:id_poll}',
  442. array(
  443. 'id_poll' => $pollinfo['id_poll'],
  444. )
  445. );
  446. $context['choices'] = array();
  447. $number = 1;
  448. while ($row = $smcFunc['db_fetch_assoc']($request))
  449. {
  450. censorText($row['label']);
  451. $context['choices'][$row['id_choice']] = array(
  452. 'id' => $row['id_choice'],
  453. 'number' => $number++,
  454. 'votes' => $row['votes'],
  455. 'label' => $row['label'],
  456. 'is_last' => false
  457. );
  458. }
  459. $smcFunc['db_free_result']($request);
  460. $last_id = max(array_keys($context['choices'])) + 1;
  461. // Add an extra choice...
  462. $context['choices'][] = array(
  463. 'id' => $last_id,
  464. 'number' => $number,
  465. 'votes' => -1,
  466. 'label' => '',
  467. 'is_last' => true
  468. );
  469. }
  470. // New poll?
  471. else
  472. {
  473. // Setup the default poll options.
  474. $context['poll'] = array(
  475. 'id' => 0,
  476. 'question' => '',
  477. 'hide_results' => 0,
  478. 'max_votes' => 1,
  479. 'change_vote' => 0,
  480. 'guest_vote' => 0,
  481. 'guest_vote_allowed' => in_array(-1, $groupsAllowedVote['allowed']),
  482. 'expiration' => '',
  483. );
  484. // Make all five poll choices empty.
  485. $context['choices'] = array(
  486. array('id' => 0, 'number' => 1, 'votes' => -1, 'label' => '', 'is_last' => false),
  487. array('id' => 1, 'number' => 2, 'votes' => -1, 'label' => '', 'is_last' => false),
  488. array('id' => 2, 'number' => 3, 'votes' => -1, 'label' => '', 'is_last' => false),
  489. array('id' => 3, 'number' => 4, 'votes' => -1, 'label' => '', 'is_last' => false),
  490. array('id' => 4, 'number' => 5, 'votes' => -1, 'label' => '', 'is_last' => true)
  491. );
  492. }
  493. }
  494. $context['page_title'] = $context['is_edit'] ? $txt['poll_edit'] : $txt['add_poll'];
  495. // Build the link tree.
  496. censorText($pollinfo['subject']);
  497. $context['linktree'][] = array(
  498. 'url' => $scripturl . '?topic=' . $topic . '.0',
  499. 'name' => $pollinfo['subject'],
  500. );
  501. $context['linktree'][] = array(
  502. 'name' => $context['page_title'],
  503. );
  504. // Register this form in the session variables.
  505. checkSubmitOnce('register');
  506. }
  507. /**
  508. * Update the settings for a poll, or add a new one.
  509. * Must be called with a topic specified in the URL.
  510. * The user must have poll_edit_any/poll_add_any permission
  511. * for the relevant action. Otherwise they must be poll starter
  512. * with poll_edit_own permission for editing, or be topic starter
  513. * with poll_add_any permission for adding.
  514. * In the case of an error, this function will redirect back to
  515. * EditPoll and display the relevant error message.
  516. * Upon successful completion of action will direct user back to topic.
  517. * Accessed via ?action=editpoll2.
  518. */
  519. function EditPoll2()
  520. {
  521. global $txt, $topic, $board, $context;
  522. global $modSettings, $user_info, $smcFunc, $sourcedir;
  523. // Sneaking off, are we?
  524. if (empty($_POST))
  525. redirectexit('action=editpoll;topic=' . $topic . '.0');
  526. if (checkSession('post', '', false) != '')
  527. $poll_errors[] = 'session_timeout';
  528. if (isset($_POST['preview']))
  529. return EditPoll();
  530. // HACKERS (!!) can't edit :P.
  531. if (empty($topic))
  532. fatal_lang_error('no_access', false);
  533. // Is this a new poll, or editing an existing?
  534. $isEdit = isset($_REQUEST['add']) ? 0 : 1;
  535. // Get the starter and the poll's ID - if it's an edit.
  536. $request = $smcFunc['db_query']('', '
  537. SELECT t.id_member_started, t.id_poll, p.id_member AS poll_starter, p.expire_time
  538. FROM {db_prefix}topics AS t
  539. LEFT JOIN {db_prefix}polls AS p ON (p.id_poll = t.id_poll)
  540. WHERE t.id_topic = {int:current_topic}
  541. LIMIT 1',
  542. array(
  543. 'current_topic' => $topic,
  544. )
  545. );
  546. if ($smcFunc['db_num_rows']($request) == 0)
  547. fatal_lang_error('no_board');
  548. $bcinfo = $smcFunc['db_fetch_assoc']($request);
  549. $smcFunc['db_free_result']($request);
  550. // Check their adding/editing is valid.
  551. if (!$isEdit && !empty($bcinfo['id_poll']))
  552. fatal_lang_error('poll_already_exists');
  553. // Are we editing a poll which doesn't exist?
  554. elseif ($isEdit && empty($bcinfo['id_poll']))
  555. fatal_lang_error('poll_not_found');
  556. // Check if they have the power to add or edit the poll.
  557. if ($isEdit && !allowedTo('poll_edit_any'))
  558. isAllowedTo('poll_edit_' . ($user_info['id'] == $bcinfo['id_member_started'] || ($bcinfo['poll_starter'] != 0 && $user_info['id'] == $bcinfo['poll_starter']) ? 'own' : 'any'));
  559. elseif (!$isEdit && !allowedTo('poll_add_any'))
  560. isAllowedTo('poll_add_' . ($user_info['id'] == $bcinfo['id_member_started'] ? 'own' : 'any'));
  561. $optionCount = 0;
  562. // Ensure the user is leaving a valid amount of options - there must be at least two.
  563. foreach ($_POST['options'] as $k => $option)
  564. {
  565. if (trim($option) != '')
  566. $optionCount++;
  567. }
  568. if ($optionCount < 2)
  569. $poll_errors[] = 'poll_few';
  570. // Also - ensure they are not removing the question.
  571. if (trim($_POST['question']) == '')
  572. $poll_errors[] = 'no_question';
  573. // Got any errors to report?
  574. if (!empty($poll_errors))
  575. {
  576. loadLanguage('Errors');
  577. // Previewing.
  578. $_POST['preview'] = true;
  579. $context['poll_error'] = array('messages' => array());
  580. foreach ($poll_errors as $poll_error)
  581. {
  582. $context['poll_error'][$poll_error] = true;
  583. $context['poll_error']['messages'][] = $txt['error_' . $poll_error];
  584. }
  585. return EditPoll();
  586. }
  587. // Prevent double submission of this form.
  588. checkSubmitOnce('check');
  589. // Now we've done all our error checking, let's get the core poll information cleaned... question first.
  590. $_POST['question'] = $smcFunc['htmlspecialchars']($_POST['question']);
  591. $_POST['question'] = $smcFunc['truncate']($_POST['question'], 255);
  592. $_POST['poll_hide'] = (int) $_POST['poll_hide'];
  593. $_POST['poll_expire'] = isset($_POST['poll_expire']) ? (int) $_POST['poll_expire'] : 0;
  594. $_POST['poll_change_vote'] = isset($_POST['poll_change_vote']) ? 1 : 0;
  595. $_POST['poll_guest_vote'] = isset($_POST['poll_guest_vote']) ? 1 : 0;
  596. // Make sure guests are actually allowed to vote generally.
  597. if ($_POST['poll_guest_vote'])
  598. {
  599. require_once($sourcedir . '/Subs-Members.php');
  600. $allowedGroups = groupsAllowedTo('poll_vote', $board);
  601. if (!in_array(-1, $allowedGroups['allowed']))
  602. $_POST['poll_guest_vote'] = 0;
  603. }
  604. // Ensure that the number options allowed makes sense, and the expiration date is valid.
  605. if (!$isEdit || allowedTo('moderate_board'))
  606. {
  607. $_POST['poll_expire'] = $_POST['poll_expire'] > 9999 ? 9999 : ($_POST['poll_expire'] < 0 ? 0 : $_POST['poll_expire']);
  608. if (empty($_POST['poll_expire']) && $_POST['poll_hide'] == 2)
  609. $_POST['poll_hide'] = 1;
  610. elseif (!$isEdit || $_POST['poll_expire'] != ceil($bcinfo['expire_time'] <= time() ? -1 : ($bcinfo['expire_time'] - time()) / (3600 * 24)))
  611. $_POST['poll_expire'] = empty($_POST['poll_expire']) ? '0' : time() + $_POST['poll_expire'] * 3600 * 24;
  612. else
  613. $_POST['poll_expire'] = $bcinfo['expire_time'];
  614. if (empty($_POST['poll_max_votes']) || $_POST['poll_max_votes'] <= 0)
  615. $_POST['poll_max_votes'] = 1;
  616. else
  617. $_POST['poll_max_votes'] = (int) $_POST['poll_max_votes'];
  618. }
  619. // If we're editing, let's commit the changes.
  620. if ($isEdit)
  621. {
  622. $smcFunc['db_query']('', '
  623. UPDATE {db_prefix}polls
  624. SET question = {string:question}, change_vote = {int:change_vote},' . (allowedTo('moderate_board') ? '
  625. hide_results = {int:hide_results}, expire_time = {int:expire_time}, max_votes = {int:max_votes},
  626. guest_vote = {int:guest_vote}' : '
  627. hide_results = CASE WHEN expire_time = {int:expire_time_zero} AND {int:hide_results} = 2 THEN 1 ELSE {int:hide_results} END') . '
  628. WHERE id_poll = {int:id_poll}',
  629. array(
  630. 'change_vote' => $_POST['poll_change_vote'],
  631. 'hide_results' => $_POST['poll_hide'],
  632. 'expire_time' => !empty($_POST['poll_expire']) ? $_POST['poll_expire'] : 0,
  633. 'max_votes' => !empty($_POST['poll_max_votes']) ? $_POST['poll_max_votes'] : 0,
  634. 'guest_vote' => $_POST['poll_guest_vote'],
  635. 'expire_time_zero' => 0,
  636. 'id_poll' => $bcinfo['id_poll'],
  637. 'question' => $_POST['question'],
  638. )
  639. );
  640. }
  641. // Otherwise, let's get our poll going!
  642. else
  643. {
  644. // Create the poll.
  645. $smcFunc['db_insert']('',
  646. '{db_prefix}polls',
  647. array(
  648. 'question' => 'string-255', 'hide_results' => 'int', 'max_votes' => 'int', 'expire_time' => 'int', 'id_member' => 'int',
  649. 'poster_name' => 'string-255', 'change_vote' => 'int', 'guest_vote' => 'int'
  650. ),
  651. array(
  652. $_POST['question'], $_POST['poll_hide'], $_POST['poll_max_votes'], $_POST['poll_expire'], $user_info['id'],
  653. $user_info['username'], $_POST['poll_change_vote'], $_POST['poll_guest_vote'],
  654. ),
  655. array('id_poll')
  656. );
  657. // Set the poll ID.
  658. $bcinfo['id_poll'] = $smcFunc['db_insert_id']('{db_prefix}polls', 'id_poll');
  659. // Link the poll to the topic
  660. $smcFunc['db_query']('', '
  661. UPDATE {db_prefix}topics
  662. SET id_poll = {int:id_poll}
  663. WHERE id_topic = {int:current_topic}',
  664. array(
  665. 'current_topic' => $topic,
  666. 'id_poll' => $bcinfo['id_poll'],
  667. )
  668. );
  669. }
  670. // Get all the choices. (no better way to remove all emptied and add previously non-existent ones.)
  671. $request = $smcFunc['db_query']('', '
  672. SELECT id_choice
  673. FROM {db_prefix}poll_choices
  674. WHERE id_poll = {int:id_poll}',
  675. array(
  676. 'id_poll' => $bcinfo['id_poll'],
  677. )
  678. );
  679. $choices = array();
  680. while ($row = $smcFunc['db_fetch_assoc']($request))
  681. $choices[] = $row['id_choice'];
  682. $smcFunc['db_free_result']($request);
  683. $delete_options = array();
  684. foreach ($_POST['options'] as $k => $option)
  685. {
  686. // Make sure the key is numeric for sanity's sake.
  687. $k = (int) $k;
  688. // They've cleared the box. Either they want it deleted, or it never existed.
  689. if (trim($option) == '')
  690. {
  691. // They want it deleted. Bye.
  692. if (in_array($k, $choices))
  693. $delete_options[] = $k;
  694. // Skip the rest...
  695. continue;
  696. }
  697. // Dress the option up for its big date with the database.
  698. $option = $smcFunc['htmlspecialchars']($option);
  699. // If it's already there, update it. If it's not... add it.
  700. if (in_array($k, $choices))
  701. $smcFunc['db_query']('', '
  702. UPDATE {db_prefix}poll_choices
  703. SET label = {string:option_name}
  704. WHERE id_poll = {int:id_poll}
  705. AND id_choice = {int:id_choice}',
  706. array(
  707. 'id_poll' => $bcinfo['id_poll'],
  708. 'id_choice' => $k,
  709. 'option_name' => $option,
  710. )
  711. );
  712. else
  713. $smcFunc['db_insert']('',
  714. '{db_prefix}poll_choices',
  715. array(
  716. 'id_poll' => 'int', 'id_choice' => 'int', 'label' => 'string-255', 'votes' => 'int',
  717. ),
  718. array(
  719. $bcinfo['id_poll'], $k, $option, 0,
  720. ),
  721. array()
  722. );
  723. }
  724. // I'm sorry, but... well, no one was choosing you. Poor options, I'll put you out of your misery.
  725. if (!empty($delete_options))
  726. {
  727. $smcFunc['db_query']('', '
  728. DELETE FROM {db_prefix}log_polls
  729. WHERE id_poll = {int:id_poll}
  730. AND id_choice IN ({array_int:delete_options})',
  731. array(
  732. 'delete_options' => $delete_options,
  733. 'id_poll' => $bcinfo['id_poll'],
  734. )
  735. );
  736. $smcFunc['db_query']('', '
  737. DELETE FROM {db_prefix}poll_choices
  738. WHERE id_poll = {int:id_poll}
  739. AND id_choice IN ({array_int:delete_options})',
  740. array(
  741. 'delete_options' => $delete_options,
  742. 'id_poll' => $bcinfo['id_poll'],
  743. )
  744. );
  745. }
  746. // Shall I reset the vote count, sir?
  747. if (isset($_POST['resetVoteCount']))
  748. {
  749. $smcFunc['db_query']('', '
  750. UPDATE {db_prefix}polls
  751. SET num_guest_voters = {int:no_votes}, reset_poll = {int:time}
  752. WHERE id_poll = {int:id_poll}',
  753. array(
  754. 'no_votes' => 0,
  755. 'id_poll' => $bcinfo['id_poll'],
  756. 'time' => time(),
  757. )
  758. );
  759. $smcFunc['db_query']('', '
  760. UPDATE {db_prefix}poll_choices
  761. SET votes = {int:no_votes}
  762. WHERE id_poll = {int:id_poll}',
  763. array(
  764. 'no_votes' => 0,
  765. 'id_poll' => $bcinfo['id_poll'],
  766. )
  767. );
  768. $smcFunc['db_query']('', '
  769. DELETE FROM {db_prefix}log_polls
  770. WHERE id_poll = {int:id_poll}',
  771. array(
  772. 'id_poll' => $bcinfo['id_poll'],
  773. )
  774. );
  775. }
  776. // Off we go.
  777. redirectexit('topic=' . $topic . '.' . $_REQUEST['start']);
  778. }
  779. /**
  780. * Remove a poll from a topic without removing the topic.
  781. * Must be called with a topic specified in the URL.
  782. * Requires poll_remove_any permission, unless it's the poll starter
  783. * with poll_remove_own permission.
  784. * Upon successful completion of action will direct user back to topic.
  785. * Accessed via ?action=removepoll.
  786. */
  787. function RemovePoll()
  788. {
  789. global $topic, $user_info, $smcFunc;
  790. // Make sure the topic is not empty.
  791. if (empty($topic))
  792. fatal_lang_error('no_access', false);
  793. // Verify the session.
  794. checkSession('get');
  795. // Check permissions.
  796. if (!allowedTo('poll_remove_any'))
  797. {
  798. $request = $smcFunc['db_query']('', '
  799. SELECT t.id_member_started, p.id_member AS poll_starter
  800. FROM {db_prefix}topics AS t
  801. INNER JOIN {db_prefix}polls AS p ON (p.id_poll = t.id_poll)
  802. WHERE t.id_topic = {int:current_topic}
  803. LIMIT 1',
  804. array(
  805. 'current_topic' => $topic,
  806. )
  807. );
  808. if ($smcFunc['db_num_rows']($request) == 0)
  809. fatal_lang_error('no_access', false);
  810. list ($topicStarter, $pollStarter) = $smcFunc['db_fetch_row']($request);
  811. $smcFunc['db_free_result']($request);
  812. isAllowedTo('poll_remove_' . ($topicStarter == $user_info['id'] || ($pollStarter != 0 && $user_info['id'] == $pollStarter) ? 'own' : 'any'));
  813. }
  814. // Retrieve the poll ID.
  815. $request = $smcFunc['db_query']('', '
  816. SELECT id_poll
  817. FROM {db_prefix}topics
  818. WHERE id_topic = {int:current_topic}
  819. LIMIT 1',
  820. array(
  821. 'current_topic' => $topic,
  822. )
  823. );
  824. list ($pollID) = $smcFunc['db_fetch_row']($request);
  825. $smcFunc['db_free_result']($request);
  826. // Remove all user logs for this poll.
  827. $smcFunc['db_query']('', '
  828. DELETE FROM {db_prefix}log_polls
  829. WHERE id_poll = {int:id_poll}',
  830. array(
  831. 'id_poll' => $pollID,
  832. )
  833. );
  834. // Remove all poll choices.
  835. $smcFunc['db_query']('', '
  836. DELETE FROM {db_prefix}poll_choices
  837. WHERE id_poll = {int:id_poll}',
  838. array(
  839. 'id_poll' => $pollID,
  840. )
  841. );
  842. // Remove the poll itself.
  843. $smcFunc['db_query']('', '
  844. DELETE FROM {db_prefix}polls
  845. WHERE id_poll = {int:id_poll}',
  846. array(
  847. 'id_poll' => $pollID,
  848. )
  849. );
  850. // Finally set the topic poll ID back to 0!
  851. $smcFunc['db_query']('', '
  852. UPDATE {db_prefix}topics
  853. SET id_poll = {int:no_poll}
  854. WHERE id_topic = {int:current_topic}',
  855. array(
  856. 'current_topic' => $topic,
  857. 'no_poll' => 0,
  858. )
  859. );
  860. // Take the moderator back to the topic.
  861. redirectexit('topic=' . $topic . '.' . $_REQUEST['start']);
  862. }
  863. ?>