ModIndexer/Program.cs
SmallJoker fefcc141a5 PUSH
2016-01-16 14:32:20 +01:00

521 lines
12 KiB
C#

using System;
using System.Net;
using System.Net.Sockets;
using System.Threading;
using System.Collections.Generic;
using HtmlAgilityPack;
using Newtonsoft.Json;
namespace ModIndexer
{
/*
Indev = 9,
Mods = 11,
Old = 13,
Games = 15
*/
class Program
{
static void Main(string[] args)
{
Console.Write("Select a mode:\n" +
"\n\t1) MR titles" +
"\n\t2) MR topics" +
"\n\t3) WIP titles" +
"\n\t4) WIP topics" +
"\n\t5) Game topics" +
"\n\t6) Old Mods titles" +
"\n\t7) Exit" +
"\n\nYour choice: ");
ConsoleKeyInfo k = new ConsoleKeyInfo();
while (k.Key != ConsoleKey.D1 &&
k.Key != ConsoleKey.D2 &&
k.Key != ConsoleKey.D3 &&
k.Key != ConsoleKey.D4 &&
k.Key != ConsoleKey.D5 &&
k.Key != ConsoleKey.D6) {
k = Console.ReadKey(true);
if (k.Key == ConsoleKey.D7) return;
}
Console.WriteLine(k.KeyChar + "\n");
byte forum = 11;
bool fetch_topics = false;
if (k.Key == ConsoleKey.D2)
fetch_topics = true;
if (k.Key == ConsoleKey.D3)
forum = 9;
if (k.Key == ConsoleKey.D4) {
forum = 9;
fetch_topics = true;
}
if (k.Key == ConsoleKey.D5) {
forum = 15;
fetch_topics = true;
}
if (k.Key == ConsoleKey.D6)
forum = 13;
Console.Write("Start page: ");
string start = Console.ReadLine();
Console.Write("End page: ");
string stop = Console.ReadLine();
new Engine(forum, start, stop, fetch_topics);
Console.WriteLine("=== DONE!\nPress any key to exit.");
Console.ReadKey(false);
}
}
/* Database types
1 mod
2 modpack
3 indev mod
4 indev modpack
5 old mod
6 game
*/
class ForumData
{
public int topicId, userId, type;
public string title, userName, link;
public ForumData(int _topicId, string _title, int _type, int _userId, string _userName, string _link)
{
topicId = _topicId;
title = _title;
type = _type;
userId = _userId;
userName = _userName;
link = _link;
}
}
class Engine
{
System.Text.Encoding enc = System.Text.Encoding.UTF8;
List<ForumData> update_data;
WebClient cli = new WebClient();
bool fetch_topics;
public Engine(int forum, string start, string stop, bool _fetch_topics)
{
update_data = new List<ForumData>();
ServicePointManager.ServerCertificateValidationCallback += ValidateRemoteCertificate;
fetch_topics = _fetch_topics;
int begin, end;
if (!int.TryParse(start, out begin))
begin = 1;
if (!int.TryParse(stop, out end))
end = 1;
for (int i = begin - 1; i < end; i++)
Start(i, forum);
byte[] answer = Config.Upload(ref update_data);
try {
int[] topics = JsonConvert.DeserializeObject<int[]>(enc.GetString(answer));
for (int i = 0; i < topics.Length; i++) {
foreach (ForumData d in update_data) {
if (d.topicId == topics[i]) {
Console.WriteLine("\t" + d.title);
break;
}
}
}
} catch (Exception e) {
Console.WriteLine(e.ToString());
Console.WriteLine(enc.GetString(answer));
}
Console.WriteLine("Done.");
}
bool ValidateRemoteCertificate(object sender, System.Security.Cryptography.X509Certificates.X509Certificate cert, System.Security.Cryptography.X509Certificates.X509Chain chain, System.Net.Security.SslPolicyErrors error)
{
// If the certificate is a valid, signed certificate, return true.
if (error == System.Net.Security.SslPolicyErrors.None ||
error == System.Net.Security.SslPolicyErrors.RemoteCertificateNameMismatch)
return true;
Console.WriteLine("X509Certificate [{0}] Policy Error: '{1}'",
cert.Subject,
error.ToString());
return false;
}
// Download page and convert to a HtmlNode object
HtmlNodeCollection OpenPage(string url, string nodes)
{
string text = enc.GetString(cli.DownloadData(url));
HtmlDocument htmlDoc = new HtmlAgilityPack.HtmlDocument();
htmlDoc.LoadHtml(text);
if (htmlDoc.ParseErrors != null) {
int count = 0;
foreach (HtmlParseError er in htmlDoc.ParseErrors) {
count++;
}
if (count > 0) {
Console.WriteLine("ParseErrors: " + count);
}
}
// If valid html page
if (htmlDoc.DocumentNode == null) {
Console.WriteLine("DocumentNode null");
return null;
}
HtmlNodeCollection bodyNode = htmlDoc.DocumentNode.SelectNodes(nodes);
if (bodyNode == null) {
Console.WriteLine("bodyNode null");
return null;
}
return bodyNode;
}
// Analyze the topic list in a forum
void Start(int page, int forum)
{
Console.WriteLine("=== Forum " + forum + " - Page " + (page + 1));
HtmlNodeCollection bodyNode = OpenPage(
"https://forum.minetest.net/viewforum.php?f=" + forum + "&start=" + (page * 30),
"//ul[@class='topiclist topics']//dt/a");
if (bodyNode == null) return;
bool canAdd = false;
string title = "",
author = "";
int topicId = 0,
authorId = 0;
foreach (HtmlNode dtNode in bodyNode) {
#region extract IDs
string link = dtNode.GetAttributeValue("href", "");
link = link.Replace("&amp;", "&");
string[] linkArgs = link.Split('&');
if (!canAdd) {
for (int i = 0; i < linkArgs.Length; i++) {
if (linkArgs[i].Length < 3) continue;
string first = linkArgs[i][0].ToString() + linkArgs[i][1];
if (first == "t=") {
topicId = int.Parse(linkArgs[i].Remove(0, 2));
}
}
title = dtNode.InnerText;
} else {
for (int i = 0; i < linkArgs.Length; i++) {
if (linkArgs[i].Length < 3) continue;
string first = linkArgs[i][0].ToString() + linkArgs[i][1];
if (first == "u=") {
authorId = int.Parse(linkArgs[i].Remove(0, 2));
}
}
author = dtNode.InnerText;
}
#endregion
if (canAdd && title.Length > 10) {
#region filter
byte type = 0; //not a mod
string lowTitle = title.ToLower();
if (lowTitle.StartsWith("[wip]")) {
bool end_space = (title[5] == ' ');
lowTitle = lowTitle.Remove(0, end_space ? 6 : 5);
title = title.Remove(0, end_space ? 6 : 5);
}
if (lowTitle.StartsWith("[mod]")) {
title = title.Remove(0, 5);
type = 1;
} else if (lowTitle.StartsWith("[modpack]")) {
title = title.Remove(0, 9);
type = 2;
} else if (lowTitle.StartsWith("[game]") &&
forum == 15) {
title = title.Remove(0, 6);
type = 6;
}
if (type == 0) {
canAdd = !canAdd;
continue;
}
switch (forum) {
case 9: type += 2; break; // make indev
case 13: type = 5; break; // is old
}
#endregion
title = specCharsToHex(removeTrash(title));
string download = "";
if (fetch_topics && type != 5) {
bool is_git = getLink(topicId, ref download);
if (!(type <= 2 || type == 6 || is_git))
download = null;
}
update_data.Add(new ForumData(topicId, title, type, authorId, author, download));
author = "";
title = "";
}
canAdd = !canAdd;
}
}
// Analyze topic contents and get link
bool getLink(int topicId, ref string link)
{
Console.WriteLine("=== Topic " + topicId);
Thread.Sleep(200);
link = "";
HtmlNodeCollection bodyNode = OpenPage(
"https://forum.minetest.net/viewtopic.php?t=" + topicId,
"//div[@class='content']");
if (bodyNode == null)
return false;
HtmlNodeCollection content = bodyNode[0].SelectNodes(".//a[@class='postlink']");
if (content == null)
return false;
string download = "", source = "";
foreach (HtmlNode dtNode in content) {
string url = dtNode.GetAttributeValue("href", "");
if (url.EndsWith(".git")) {
source = url;
continue;
}
if (url[url.Length - 1] == '/') {
url = url.Remove(url.Length - 1);
}
if (url.Contains(".zip") ||
url.Contains("/zipball/") ||
url.Contains("/tarball/") ||
url.Contains("/archive/") ||
url.Contains("mediafire.com/")) {
// Direct download link
if (url.Contains("://ompldr.org"))
continue;
bool contains_git = url.Contains("git");
if (contains_git) {
byte count = 0,
pos = 0;
for (byte i = 0; i < url.Length; i++) {
if (url[i] == '/') {
if (count == 4) {
pos = i;
}
count++;
}
}
if (count == 6) {
source = url.Substring(0, pos);
break;
}
}
if (download == "")
download = url;
} else if (url.Contains("://github.com/") ||
url.Contains("://notabug.org/")) {
if (url.Contains("/minetest/minetest") ||
url.Contains("/commits"))
continue;
byte count = 0,
pos = 0;
for (byte i = 0; i < url.Length; i++) {
if (url[i] == '/') {
if (count == 4) {
// If it's too long
pos = i;
}
count++;
}
}
if (count < 4 || count > 5)
continue;
// url.EndsWith("/tree") || url.EndsWith("/master")
// //github/user/proj/master
if (count == 5)
source = url.Substring(0, pos);
else
source = url;
break;
}
}
if (source == "" && download == "")
return false;
link = source != "" ? source : download;
return source != "";
}
// Remove common control characters and spaces
string FastTrim(string t, string rm)
{
int index = 0;
bool space_begin = true;
char[] ret = new char[t.Length];
for (int i = 0; i < t.Length; i++) {
char cur = t[i];
bool found = false;
if (space_begin) {
space_begin = (
cur == ' ' ||
cur == '\t' ||
cur == '\n' ||
cur == '\r'
);
found = space_begin;
}
for (int k = 0; k < rm.Length && !found; k++) {
if (rm[k] == cur) {
found = true;
break;
}
}
if (!found) {
ret[index] = cur;
index++;
}
}
if (index < t.Length)
Array.Resize(ref ret, index);
return new string(ret);
}
// Remove useless tags from the forum titles
string bad_chars = ". !?";
string removeTrash(string t)
{
string raw = t;
int pos = -1,
open_pos = 0;
bool opened = false,
delete = false;
for (int i = 0; i < t.Length; i++) {
pos++;
char cur = t[i];
if (cur == '[' || cur == '{') {
opened = true;
open_pos = pos;
continue;
}
if (cur == ']' || cur == '}') {
opened = false;
int len = pos - open_pos + 1;
string content = raw.Substring(open_pos + 1, len - 2).ToLower();
int num = 0xC0FFEE;
bool is_number = int.TryParse(content, out num);
if (delete || (num != 0xC0FFEE && is_number)
|| content == "git"
|| content == "github"
|| content == "wip") {
raw = raw.Remove(open_pos, len);
pos -= len;
}
delete = false;
continue;
}
if (opened &&
!delete &&
strContains(bad_chars, cur)) {
delete = true;
}
}
delete = true;
pos = 0;
char[] ret = new char[raw.Length];
for (int i = 0; i < raw.Length; i++) {
char cur = raw[i];
bool is_space = (cur == ' ');
if (delete && is_space)
continue;
if (is_space && i == raw.Length - 1)
continue;
delete = is_space;
ret[pos] = cur;
pos++;
}
if (pos < ret.Length)
Array.Resize(ref ret, pos);
return new string(ret);
}
// Convert special characters to HTML code
string specCharsToHex(string t)
{
char[] fromCr = { '"', '\'', '\\', '{', '}', '|', '%', ':', '<', '>' };
string[] toStr = { "&quot;", "&#39;", "&#92;", "&#123;", "&#125;", "&#124;", "&#37;", "&#58;", "&lt;", "&gt;" };
System.Text.StringBuilder sb = new System.Text.StringBuilder();
bool wasSpace = false;
for (int i = 0; i < t.Length; i++) {
bool isSpace = (t[i] == ' ');
if (wasSpace && isSpace)
continue;
// Cut off non-ASCII
if ((ushort)t[i] > 0xFF)
continue;
bool found = false;
for (int k = 0; k < fromCr.Length; k++) {
if (t[i] == fromCr[k]) {
sb.Append(toStr[k]);
found = true;
break;
}
}
wasSpace = isSpace;
if (!found)
sb.Append(t[i]);
}
return sb.ToString();
}
bool strContains(string t, char c)
{
for (int i = 0; i < t.Length; i++) {
if (t[i] == c)
return true;
}
return false;
}
}
}