Poll.php 30 KB

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