Poll.php 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979
  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['start'] = (int) $_REQUEST['start'];
  272. $context['is_edit'] = isset($_REQUEST['add']) ? 0 : 1;
  273. // Check if a poll currently exists on this topic, and get the id, question and starter.
  274. $request = $smcFunc['db_query']('', '
  275. SELECT
  276. t.id_member_started, p.id_poll, p.question, p.hide_results, p.expire_time, p.max_votes, p.change_vote,
  277. m.subject, p.guest_vote, p.id_member AS poll_starter
  278. FROM {db_prefix}topics AS t
  279. INNER JOIN {db_prefix}messages AS m ON (m.id_msg = t.id_first_msg)
  280. LEFT JOIN {db_prefix}polls AS p ON (p.id_poll = t.id_poll)
  281. WHERE t.id_topic = {int:current_topic}
  282. LIMIT 1',
  283. array(
  284. 'current_topic' => $topic,
  285. )
  286. );
  287. // Assume the the topic exists, right?
  288. if ($smcFunc['db_num_rows']($request) == 0)
  289. fatal_lang_error('no_board');
  290. // Get the poll information.
  291. $pollinfo = $smcFunc['db_fetch_assoc']($request);
  292. $smcFunc['db_free_result']($request);
  293. // If we are adding a new poll - make sure that there isn't already a poll there.
  294. if (!$context['is_edit'] && !empty($pollinfo['id_poll']))
  295. fatal_lang_error('poll_already_exists');
  296. // Otherwise, if we're editing it, it does exist I assume?
  297. elseif ($context['is_edit'] && empty($pollinfo['id_poll']))
  298. fatal_lang_error('poll_not_found');
  299. // Can you do this?
  300. if ($context['is_edit'] && !allowedTo('poll_edit_any'))
  301. isAllowedTo('poll_edit_' . ($user_info['id'] == $pollinfo['id_member_started'] || ($pollinfo['poll_starter'] != 0 && $user_info['id'] == $pollinfo['poll_starter']) ? 'own' : 'any'));
  302. elseif (!$context['is_edit'] && !allowedTo('poll_add_any'))
  303. isAllowedTo('poll_add_' . ($user_info['id'] == $pollinfo['id_member_started'] ? 'own' : 'any'));
  304. $context['can_moderate_poll'] = isset($_REQUEST['add']) ? true : allowedTo('poll_edit_' . ($user_info['id'] == $pollinfo['id_member_started'] || ($pollinfo['poll_starter'] != 0 && $user_info['id'] == $pollinfo['poll_starter']) ? '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. $context['last_choice_id'] = $last_id;
  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']) || !$context['can_moderate_poll'] ? '' : 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. $context['last_choice_id'] = $last_id;
  470. }
  471. // New poll?
  472. else
  473. {
  474. // Setup the default poll options.
  475. $context['poll'] = array(
  476. 'id' => 0,
  477. 'question' => '',
  478. 'hide_results' => 0,
  479. 'max_votes' => 1,
  480. 'change_vote' => 0,
  481. 'guest_vote' => 0,
  482. 'guest_vote_allowed' => in_array(-1, $groupsAllowedVote['allowed']),
  483. 'expiration' => '',
  484. );
  485. // Make all five poll choices empty.
  486. $context['choices'] = array(
  487. array('id' => 0, 'number' => 1, 'votes' => -1, 'label' => '', 'is_last' => false),
  488. array('id' => 1, 'number' => 2, 'votes' => -1, 'label' => '', 'is_last' => false),
  489. array('id' => 2, 'number' => 3, 'votes' => -1, 'label' => '', 'is_last' => false),
  490. array('id' => 3, 'number' => 4, 'votes' => -1, 'label' => '', 'is_last' => false),
  491. array('id' => 4, 'number' => 5, 'votes' => -1, 'label' => '', 'is_last' => true)
  492. );
  493. $context['last_choice_id'] = 4;
  494. }
  495. }
  496. $context['page_title'] = $context['is_edit'] ? $txt['poll_edit'] : $txt['add_poll'];
  497. // Build the link tree.
  498. censorText($pollinfo['subject']);
  499. $context['linktree'][] = array(
  500. 'url' => $scripturl . '?topic=' . $topic . '.0',
  501. 'name' => $pollinfo['subject'],
  502. );
  503. $context['linktree'][] = array(
  504. 'name' => $context['page_title'],
  505. );
  506. // Register this form in the session variables.
  507. checkSubmitOnce('register');
  508. }
  509. /**
  510. * Update the settings for a poll, or add a new one.
  511. * Must be called with a topic specified in the URL.
  512. * The user must have poll_edit_any/poll_add_any permission
  513. * for the relevant action. Otherwise they must be poll starter
  514. * with poll_edit_own permission for editing, or be topic starter
  515. * with poll_add_any permission for adding.
  516. * In the case of an error, this function will redirect back to
  517. * EditPoll and display the relevant error message.
  518. * Upon successful completion of action will direct user back to topic.
  519. * Accessed via ?action=editpoll2.
  520. */
  521. function EditPoll2()
  522. {
  523. global $txt, $topic, $board, $context;
  524. global $modSettings, $user_info, $smcFunc, $sourcedir;
  525. // Sneaking off, are we?
  526. if (empty($_POST))
  527. redirectexit('action=editpoll;topic=' . $topic . '.0');
  528. if (checkSession('post', '', false) != '')
  529. $poll_errors[] = 'session_timeout';
  530. if (isset($_POST['preview']))
  531. return EditPoll();
  532. // HACKERS (!!) can't edit :P.
  533. if (empty($topic))
  534. fatal_lang_error('no_access', false);
  535. // Is this a new poll, or editing an existing?
  536. $isEdit = isset($_REQUEST['add']) ? 0 : 1;
  537. // Get the starter and the poll's ID - if it's an edit.
  538. $request = $smcFunc['db_query']('', '
  539. SELECT t.id_member_started, t.id_poll, p.id_member AS poll_starter, p.expire_time
  540. FROM {db_prefix}topics AS t
  541. LEFT JOIN {db_prefix}polls AS p ON (p.id_poll = t.id_poll)
  542. WHERE t.id_topic = {int:current_topic}
  543. LIMIT 1',
  544. array(
  545. 'current_topic' => $topic,
  546. )
  547. );
  548. if ($smcFunc['db_num_rows']($request) == 0)
  549. fatal_lang_error('no_board');
  550. $bcinfo = $smcFunc['db_fetch_assoc']($request);
  551. $smcFunc['db_free_result']($request);
  552. // Check their adding/editing is valid.
  553. if (!$isEdit && !empty($bcinfo['id_poll']))
  554. fatal_lang_error('poll_already_exists');
  555. // Are we editing a poll which doesn't exist?
  556. elseif ($isEdit && empty($bcinfo['id_poll']))
  557. fatal_lang_error('poll_not_found');
  558. // Check if they have the power to add or edit the poll.
  559. if ($isEdit && !allowedTo('poll_edit_any'))
  560. isAllowedTo('poll_edit_' . ($user_info['id'] == $bcinfo['id_member_started'] || ($bcinfo['poll_starter'] != 0 && $user_info['id'] == $bcinfo['poll_starter']) ? 'own' : 'any'));
  561. elseif (!$isEdit && !allowedTo('poll_add_any'))
  562. isAllowedTo('poll_add_' . ($user_info['id'] == $bcinfo['id_member_started'] ? 'own' : 'any'));
  563. $optionCount = 0;
  564. $idCount = 0;
  565. // Ensure the user is leaving a valid amount of options - there must be at least two.
  566. foreach ($_POST['options'] as $k => $option)
  567. {
  568. if (trim($option) != '')
  569. {
  570. $optionCount++;
  571. $idCount = max($idCount, $k);
  572. }
  573. }
  574. if ($optionCount < 2)
  575. $poll_errors[] = 'poll_few';
  576. elseif ($optionCount > 256 || $idCount > 255)
  577. $poll_errors[] = 'poll_many';
  578. // Also - ensure they are not removing the question.
  579. if (trim($_POST['question']) == '')
  580. $poll_errors[] = 'no_question';
  581. // Got any errors to report?
  582. if (!empty($poll_errors))
  583. {
  584. loadLanguage('Errors');
  585. // Previewing.
  586. $_POST['preview'] = true;
  587. $context['poll_error'] = array('messages' => array());
  588. foreach ($poll_errors as $poll_error)
  589. {
  590. $context['poll_error'][$poll_error] = true;
  591. $context['poll_error']['messages'][] = $txt['error_' . $poll_error];
  592. }
  593. return EditPoll();
  594. }
  595. // Prevent double submission of this form.
  596. checkSubmitOnce('check');
  597. // Now we've done all our error checking, let's get the core poll information cleaned... question first.
  598. $_POST['question'] = $smcFunc['htmlspecialchars']($_POST['question']);
  599. $_POST['question'] = $smcFunc['truncate']($_POST['question'], 255);
  600. $_POST['poll_hide'] = (int) $_POST['poll_hide'];
  601. $_POST['poll_expire'] = isset($_POST['poll_expire']) ? (int) $_POST['poll_expire'] : 0;
  602. $_POST['poll_change_vote'] = isset($_POST['poll_change_vote']) ? 1 : 0;
  603. $_POST['poll_guest_vote'] = isset($_POST['poll_guest_vote']) ? 1 : 0;
  604. // Make sure guests are actually allowed to vote generally.
  605. if ($_POST['poll_guest_vote'])
  606. {
  607. require_once($sourcedir . '/Subs-Members.php');
  608. $allowedGroups = groupsAllowedTo('poll_vote', $board);
  609. if (!in_array(-1, $allowedGroups['allowed']))
  610. $_POST['poll_guest_vote'] = 0;
  611. }
  612. // Ensure that the number options allowed makes sense, and the expiration date is valid.
  613. if (!$isEdit || allowedTo('moderate_board'))
  614. {
  615. $_POST['poll_expire'] = $_POST['poll_expire'] > 9999 ? 9999 : ($_POST['poll_expire'] < 0 ? 0 : $_POST['poll_expire']);
  616. if (empty($_POST['poll_expire']) && $_POST['poll_hide'] == 2)
  617. $_POST['poll_hide'] = 1;
  618. elseif (!$isEdit || $_POST['poll_expire'] != ceil($bcinfo['expire_time'] <= time() ? -1 : ($bcinfo['expire_time'] - time()) / (3600 * 24)))
  619. $_POST['poll_expire'] = empty($_POST['poll_expire']) ? '0' : time() + $_POST['poll_expire'] * 3600 * 24;
  620. else
  621. $_POST['poll_expire'] = $bcinfo['expire_time'];
  622. if (empty($_POST['poll_max_votes']) || $_POST['poll_max_votes'] <= 0)
  623. $_POST['poll_max_votes'] = 1;
  624. else
  625. $_POST['poll_max_votes'] = (int) $_POST['poll_max_votes'];
  626. }
  627. // If we're editing, let's commit the changes.
  628. if ($isEdit)
  629. {
  630. $smcFunc['db_query']('', '
  631. UPDATE {db_prefix}polls
  632. SET question = {string:question}, change_vote = {int:change_vote},' . (allowedTo('moderate_board') ? '
  633. hide_results = {int:hide_results}, expire_time = {int:expire_time}, max_votes = {int:max_votes},
  634. guest_vote = {int:guest_vote}' : '
  635. hide_results = CASE WHEN expire_time = {int:expire_time_zero} AND {int:hide_results} = 2 THEN 1 ELSE {int:hide_results} END') . '
  636. WHERE id_poll = {int:id_poll}',
  637. array(
  638. 'change_vote' => $_POST['poll_change_vote'],
  639. 'hide_results' => $_POST['poll_hide'],
  640. 'expire_time' => !empty($_POST['poll_expire']) ? $_POST['poll_expire'] : 0,
  641. 'max_votes' => !empty($_POST['poll_max_votes']) ? $_POST['poll_max_votes'] : 0,
  642. 'guest_vote' => $_POST['poll_guest_vote'],
  643. 'expire_time_zero' => 0,
  644. 'id_poll' => $bcinfo['id_poll'],
  645. 'question' => $_POST['question'],
  646. )
  647. );
  648. }
  649. // Otherwise, let's get our poll going!
  650. else
  651. {
  652. // Create the poll.
  653. $smcFunc['db_insert']('',
  654. '{db_prefix}polls',
  655. array(
  656. 'question' => 'string-255', 'hide_results' => 'int', 'max_votes' => 'int', 'expire_time' => 'int', 'id_member' => 'int',
  657. 'poster_name' => 'string-255', 'change_vote' => 'int', 'guest_vote' => 'int'
  658. ),
  659. array(
  660. $_POST['question'], $_POST['poll_hide'], $_POST['poll_max_votes'], $_POST['poll_expire'], $user_info['id'],
  661. $user_info['username'], $_POST['poll_change_vote'], $_POST['poll_guest_vote'],
  662. ),
  663. array('id_poll')
  664. );
  665. // Set the poll ID.
  666. $bcinfo['id_poll'] = $smcFunc['db_insert_id']('{db_prefix}polls', 'id_poll');
  667. // Link the poll to the topic
  668. $smcFunc['db_query']('', '
  669. UPDATE {db_prefix}topics
  670. SET id_poll = {int:id_poll}
  671. WHERE id_topic = {int:current_topic}',
  672. array(
  673. 'current_topic' => $topic,
  674. 'id_poll' => $bcinfo['id_poll'],
  675. )
  676. );
  677. }
  678. // Get all the choices. (no better way to remove all emptied and add previously non-existent ones.)
  679. $request = $smcFunc['db_query']('', '
  680. SELECT id_choice
  681. FROM {db_prefix}poll_choices
  682. WHERE id_poll = {int:id_poll}',
  683. array(
  684. 'id_poll' => $bcinfo['id_poll'],
  685. )
  686. );
  687. $choices = array();
  688. while ($row = $smcFunc['db_fetch_assoc']($request))
  689. $choices[] = $row['id_choice'];
  690. $smcFunc['db_free_result']($request);
  691. $delete_options = array();
  692. foreach ($_POST['options'] as $k => $option)
  693. {
  694. // Make sure the key is numeric for sanity's sake.
  695. $k = (int) $k;
  696. // They've cleared the box. Either they want it deleted, or it never existed.
  697. if (trim($option) == '')
  698. {
  699. // They want it deleted. Bye.
  700. if (in_array($k, $choices))
  701. $delete_options[] = $k;
  702. // Skip the rest...
  703. continue;
  704. }
  705. // Dress the option up for its big date with the database.
  706. $option = $smcFunc['htmlspecialchars']($option);
  707. // If it's already there, update it. If it's not... add it.
  708. if (in_array($k, $choices))
  709. $smcFunc['db_query']('', '
  710. UPDATE {db_prefix}poll_choices
  711. SET label = {string:option_name}
  712. WHERE id_poll = {int:id_poll}
  713. AND id_choice = {int:id_choice}',
  714. array(
  715. 'id_poll' => $bcinfo['id_poll'],
  716. 'id_choice' => $k,
  717. 'option_name' => $option,
  718. )
  719. );
  720. else
  721. $smcFunc['db_insert']('',
  722. '{db_prefix}poll_choices',
  723. array(
  724. 'id_poll' => 'int', 'id_choice' => 'int', 'label' => 'string-255', 'votes' => 'int',
  725. ),
  726. array(
  727. $bcinfo['id_poll'], $k, $option, 0,
  728. ),
  729. array()
  730. );
  731. }
  732. // I'm sorry, but... well, no one was choosing you. Poor options, I'll put you out of your misery.
  733. if (!empty($delete_options))
  734. {
  735. $smcFunc['db_query']('', '
  736. DELETE FROM {db_prefix}log_polls
  737. WHERE id_poll = {int:id_poll}
  738. AND id_choice IN ({array_int:delete_options})',
  739. array(
  740. 'delete_options' => $delete_options,
  741. 'id_poll' => $bcinfo['id_poll'],
  742. )
  743. );
  744. $smcFunc['db_query']('', '
  745. DELETE FROM {db_prefix}poll_choices
  746. WHERE id_poll = {int:id_poll}
  747. AND id_choice IN ({array_int:delete_options})',
  748. array(
  749. 'delete_options' => $delete_options,
  750. 'id_poll' => $bcinfo['id_poll'],
  751. )
  752. );
  753. }
  754. // Shall I reset the vote count, sir?
  755. if (isset($_POST['resetVoteCount']))
  756. {
  757. $smcFunc['db_query']('', '
  758. UPDATE {db_prefix}polls
  759. SET num_guest_voters = {int:no_votes}, reset_poll = {int:time}
  760. WHERE id_poll = {int:id_poll}',
  761. array(
  762. 'no_votes' => 0,
  763. 'id_poll' => $bcinfo['id_poll'],
  764. 'time' => time(),
  765. )
  766. );
  767. $smcFunc['db_query']('', '
  768. UPDATE {db_prefix}poll_choices
  769. SET votes = {int:no_votes}
  770. WHERE id_poll = {int:id_poll}',
  771. array(
  772. 'no_votes' => 0,
  773. 'id_poll' => $bcinfo['id_poll'],
  774. )
  775. );
  776. $smcFunc['db_query']('', '
  777. DELETE FROM {db_prefix}log_polls
  778. WHERE id_poll = {int:id_poll}',
  779. array(
  780. 'id_poll' => $bcinfo['id_poll'],
  781. )
  782. );
  783. }
  784. // Off we go.
  785. redirectexit('topic=' . $topic . '.' . $_REQUEST['start']);
  786. }
  787. /**
  788. * Remove a poll from a topic without removing the topic.
  789. * Must be called with a topic specified in the URL.
  790. * Requires poll_remove_any permission, unless it's the poll starter
  791. * with poll_remove_own permission.
  792. * Upon successful completion of action will direct user back to topic.
  793. * Accessed via ?action=removepoll.
  794. */
  795. function RemovePoll()
  796. {
  797. global $topic, $user_info, $smcFunc;
  798. // Make sure the topic is not empty.
  799. if (empty($topic))
  800. fatal_lang_error('no_access', false);
  801. // Verify the session.
  802. checkSession('get');
  803. // Check permissions.
  804. if (!allowedTo('poll_remove_any'))
  805. {
  806. $request = $smcFunc['db_query']('', '
  807. SELECT t.id_member_started, p.id_member AS poll_starter
  808. FROM {db_prefix}topics AS t
  809. INNER JOIN {db_prefix}polls AS p ON (p.id_poll = t.id_poll)
  810. WHERE t.id_topic = {int:current_topic}
  811. LIMIT 1',
  812. array(
  813. 'current_topic' => $topic,
  814. )
  815. );
  816. if ($smcFunc['db_num_rows']($request) == 0)
  817. fatal_lang_error('no_access', false);
  818. list ($topicStarter, $pollStarter) = $smcFunc['db_fetch_row']($request);
  819. $smcFunc['db_free_result']($request);
  820. isAllowedTo('poll_remove_' . ($topicStarter == $user_info['id'] || ($pollStarter != 0 && $user_info['id'] == $pollStarter) ? 'own' : 'any'));
  821. }
  822. // Retrieve the poll ID.
  823. $request = $smcFunc['db_query']('', '
  824. SELECT id_poll
  825. FROM {db_prefix}topics
  826. WHERE id_topic = {int:current_topic}
  827. LIMIT 1',
  828. array(
  829. 'current_topic' => $topic,
  830. )
  831. );
  832. list ($pollID) = $smcFunc['db_fetch_row']($request);
  833. $smcFunc['db_free_result']($request);
  834. // Remove all user logs for this poll.
  835. $smcFunc['db_query']('', '
  836. DELETE FROM {db_prefix}log_polls
  837. WHERE id_poll = {int:id_poll}',
  838. array(
  839. 'id_poll' => $pollID,
  840. )
  841. );
  842. // Remove all poll choices.
  843. $smcFunc['db_query']('', '
  844. DELETE FROM {db_prefix}poll_choices
  845. WHERE id_poll = {int:id_poll}',
  846. array(
  847. 'id_poll' => $pollID,
  848. )
  849. );
  850. // Remove the poll itself.
  851. $smcFunc['db_query']('', '
  852. DELETE FROM {db_prefix}polls
  853. WHERE id_poll = {int:id_poll}',
  854. array(
  855. 'id_poll' => $pollID,
  856. )
  857. );
  858. // Finally set the topic poll ID back to 0!
  859. $smcFunc['db_query']('', '
  860. UPDATE {db_prefix}topics
  861. SET id_poll = {int:no_poll}
  862. WHERE id_topic = {int:current_topic}',
  863. array(
  864. 'current_topic' => $topic,
  865. 'no_poll' => 0,
  866. )
  867. );
  868. // Take the moderator back to the topic.
  869. redirectexit('topic=' . $topic . '.' . $_REQUEST['start']);
  870. }
  871. ?>