<?php
$bot_api = require('api_key.php');
$chat_id = -1001180504638;
$mysql = require('mysql_credentials.php');
$conn = new mysqli($mysql['servername'], $mysql['username'], $mysql['password'], $mysql['database']);
if ($conn->connect_error) {
    die("Connection failed: " . $conn->connect_error);
}

# Sends the given text wrapped in triple backticks as Markdown.
function send_text($post_message) {
  global $bot_api;
  global $chat_id;
  $url = 'https://api.telegram.org/bot' . $bot_api . '/sendMessage';
  $post_msg = array('chat_id' => $chat_id, 'text' => $post_message );
  $options = array(
    'http' => array(
      'header' => "Content-type: application/x-www-form-urlencoded\r\n",
      'method' => 'POST',
      'content' => http_build_query($post_msg)
    )
  );
  $context = stream_context_create($options);
  $result = file_get_contents($url, false, $context);
  return $result;
}

# Pin Message
function pin_message($message_id) {
  global $bot_api;
  global $chat_id;
  $url = 'https://api.telegram.org/bot' . $bot_api . '/pinChatMessage';
  $post_msg = array('chat_id' => $chat_id, 'message_id' => $message_id, 'disable_notification' => 'true' );
  $options = array(
    'http' => array(
      'header' => "Content-type: application/x-www-form-urlencoded\r\n",
      'method' => 'POST',
      'content' => http_build_query($post_msg)
    )
  );
  $context = stream_context_create($options);
  $result = file_get_contents($url, false, $context);
  return $result;
}

# Takes a user array as input. Checks who follows him/her and adds
# him/her to $output array. Exits when a user is followed by no one.
function get_chain_from_user($user) {
  global $conn;
  $output = array($user);
  $last_user_id = $user['user_id'];
  while (true) {
    $query = "SELECT user_id, username from users where follows = $last_user_id ;";
    $result = $conn->query($query);
    if ($result->num_rows > 1) {
      $text = "Chain is unstable. The following users are pointing to ";
      $op = $conn->query("select username from users where user_id = $last_user_id;");
      $text .= $op->fetch_assoc()['username'] . "\n";
      while ($row = $result->fetch_assoc()) {
        $text .= '@' . $row['username'] . "\n";
      }
      send_text($text);
      mysql_data_seek($result, 0);
    }
    if ($result->num_rows > 0){
      # Code executed if this isn't the last user
      $details = $result->fetch_assoc();
      array_push($output, $details );
      $last_user_id = $details['user_id'];
    }
    else {
      break;
    }
  }
  return $output;
}

# First generates a list of people not following anyone (end_points)
# Then runs get_chain_from_user on all of them and stores them in
# $chains array. Then compares the arrays inside $chains array and
# returns the longest one.
function get_longest_chain() {
  global $conn;
  $query = "SELECT user_id, username FROM users WHERE follows = -1;";
  $end_points = $conn->query($query);
  if ($end_points->num_rows == 0) {
    return array();
  }
  $chains = array();
  while ($end_point = $end_points->fetch_assoc()){
    $chain = get_chain_from_user($end_point);
    array_push($chains, $chain);
  }
  $longest_chain_index = 0;
  for ($i = 0; $i < count($chains); $i++){
    if (count($chains[$i]) > count($chains[$longest_chain_index])) {
      $longest_chain_index = $i;
    }
  }
  return $chains[$longest_chain_index];
}

# Converts chain to string for sending
function chain_to_string($chain) {
  $string = "";
  for ($i = count($chain) - 1; $i >= 0 ; $i--) {
    $string .= $chain[$i]['username'];
    if ($i != 0) {
      $string .= "\n";
    }
  }
  return $string;
}

function update_user_by_username($username) {
  global $conn;
  $html = file_get_contents("https://t.me/" . $username);
  $dom = new domDocument;
  $dom->loadHTML($html);
  $dom->preserveWhiteSpace = false;
  $xpath = new \DOMXPath($dom);
  $changed = false;
  foreach ($xpath->query("descendant-or-self::div[@class and contains(concat(' ', normalize-space(@class), ' '), ' tgme_page_description ')]/a") as $node){
    $username_follows = preg_replace('/^@/', '', $node->nodeValue);
    $query = "SELECT user_id FROM users WHERE username = '" .  $username_follows . "';" ;
    $result = $conn->query($query);
    $query2 = "SELECT * FROM exceptions WHERE username = '" . $username  . "' AND points_to = '" . $username_follows . "';";
    $exceptions = $conn->query($query2);
    if ($result->num_rows > 0 && $exceptions->num_rows == 0) {
      $row = $result->fetch_assoc();
      $query = "UPDATE users SET follows = " . $row['user_id'] . " WHERE username =  '" . $username . "';" ;
      $conn->query($query);
      $changed = true;
      return;
    }
  }
  if (!$changed) {
      $query = "UPDATE users SET follows = -1 WHERE username =  '" . $username . "';" ;
      $conn->query($query);
  }
}
if ($argv[1] == "all") {
  $query = "SELECT user_id, username FROM users WHERE follows = -1;";
  $end_points = $conn->query($query);
  while ($end_point = $end_points->fetch_assoc()){
    $chain = get_chain_from_user($end_point);
    send_text(chain_to_string($chain));
  }
  exit();
}

# Update users
$query = "SELECT username FROM users;";
$users = $conn->query($query);
while ($user = $users->fetch_assoc()) {
  update_user_by_username($user['username']);
}

# Get longest chain and compare it to old chain and send it.
$chain = get_longest_chain();
$chain_string = chain_to_string($chain);
$saved_chain = include('chain.php');
if ($saved_chain != $chain_string) {
  $send_message = "Chain Length: " . count($chain) . "\n#Chain". count($chain) ."\n\n" . $chain_string;
  $reply = send_text($send_message);
  $json = json_decode($reply);
  pin_message($json->{'result'}->{"message_id"});
  $file = fopen('chain.php', 'w');
  $contents = "<?php return '". $chain_string . "'; ?>";
  fwrite($file, $contents);
  fclose($file);
  $file2 = fopen('lastmember.php', 'w');
  $contents = "<?php return '" . $chain[count($chain) - 1]["username"] . "'; ?>";
  fwrite($file2, $contents);
  fclose($file2);
}
elseif ($argv[1] == "send") {
  send_text("The pinned message is already the latest.");
}


$conn->close();
?>