From a022377f08680f09fd53cc10685038b8a17abe97 Mon Sep 17 00:00:00 2001
From: Yessiest <yessiest@memeware.net>
Date: Mon, 27 Feb 2023 08:45:00 +0400
Subject: [PATCH] Prototype

---
 markdown.rb | 54 ++++++++++++++++++++++++++++++++++++++++-------------
 test.rb     | 30 +++++++++++++++++++++++++++--
 2 files changed, 69 insertions(+), 15 deletions(-)

diff --git a/markdown.rb b/markdown.rb
index 3be5015..cb3d380 100644
--- a/markdown.rb
+++ b/markdown.rb
@@ -79,7 +79,7 @@ module Markdown
         x.gsub(/^(?<!\\)(\#{1,4})([^\n\r]*)/) { 
           level,content = Regexp.last_match[1..2]
           "<h#{level.length}>"+content+"</h#{level.length}>"
-        }
+        }.gsub(/^\-{3,}/,"<hr>")
       end.join("\n")
       super
     end
@@ -93,10 +93,10 @@ module Markdown
       super()
     end
     def to_html
-      @output = @input.gsub(/(?<=\n)(?<!\\)```([\w_-]*)(.*?)```/) {
+      @output = @input.gsub(/(?:\n|^)```([\w_-]*)([\s\S]+?)```/) {
         language,code = Regexp.last_match[1..2]
-        code = Markdown::html_highlighter.call(language,code)
-        "<code>#{code}</code>"
+        code = Markdown::html_highlighter.call(language,code) if Markdown::html_highlighter
+        "<pre><code>#{code.gsub /[|#*`~_!\[]/,"\\\\\\0"}</code></pre>"
       }
       super()
     end
@@ -162,28 +162,56 @@ module Markdown
       tables = []
       cur_table = []
       lines.each_with_index { |line,index|
-        if (line.start_with? / *\|/) and (line.match /^ *\|.*\|/) then
-            table_start = index
-            table_column_count = line.count "|"
-            cur_table.push (line.split "|")
-        end
-        if (table_start != -1) and (line.match /^ *\|([^\|]*\|){#{table_column_count-1}}$/) then
+        if (table_start != -1) and (line.match /^\s*\|([^\|]*\|){#{table_column_count-1}}$/) then
           if (table_testline == -1) then
-            if (line.match /^ *\|(\-*\|){#{table_column_count-1}}$/) then 
+            if (line.match /^\s*\|(\-*\|){#{table_column_count-1}}$/) then
               table_testline = 1
             else
               table_start = -1
               cur_table = []
             end
           else 
-            cur_table.push (line.split "|") 
+            cur_table.push (line.split("|").filter_map { |x| x.strip if x.match /\S+/ }) 
           end
+        elsif (table_start != -1) then
+          obj = {table: cur_table, start: table_start, end: index}
+          tables.push(obj)
+          table_start = -1
+          cur_table = []
+          table_testline = -1
+          table_column_count = 0
+        end
+        if (table_start == -1) and (line.start_with? /\s*\|/ ) and (line.match /^\s*\|.*\|/) then
+          table_start = index
+          table_column_count = line.count "|"
+          cur_table.push (line.split("|").filter_map { |x| x.strip if x.match /\S+/ })
         end
-        puts cur_table
       }
+      if cur_table != [] then
+        obj = {table: cur_table, start:table_start, end: lines.count-1}
+        tables.push(obj)
+      end
+      tables.reverse.each { |x|
+        lines[x[:start]..x[:end]] = (x[:table].map do |a2d|
+          (a2d.map { |x| (x.start_with? "#") ? "    <th>"+x.sub(/^#\s+/,"")+"</th>" : "    <td>"+x+"</td>"}).prepend("  <tr>").append("  </tr>")
+        end).flatten.prepend("<table>").append("</table>")
+      }
+      @output = lines.join("\n")
       super()
     end
   end
+  
+  # Backslash cleaner
+  # Cleans excessive backslashes after the translation
+  class BackslashTranslator < AbstractTranslator
+    def initialize(text)
+      @input = text
+      @output = text
+    end
+    def to_html
+      @output = @input.gsub(/\\(.)/,"\\1")
+    end
+  end
 end
 
 
diff --git a/test.rb b/test.rb
index d7cd45e..48cbeb5 100644
--- a/test.rb
+++ b/test.rb
@@ -62,12 +62,31 @@ piss__
 CODE
                                   ).to_html
 
-puts (Markdown::QuoteTranslator.new(<<TEXT
+test = (Markdown::CodeBlockTranslator.new(<<TEXT
+# Markdown garbage gallery
+## Header level 2
+### Header level 3
+#### Header level 4
+__[Underlined Link](https://google.com)__
+__**unreal shitworks**__   
+
+split   
+---
+
+![Fucking image idk](https://external-content.duckduckgo.com/iu/?u=https%3A%2F%2Ftse3.explicit.bing.net%2Fth%3Fid%3DOIP.qX1HmpFNHyaTfXv-SLnAJgHaDD%26pid%3DApi&f=1&ipt=dc0e92fdd701395eda76714338060dcf91c7ff9e228f108d8af6e1ba3decd1c2&ipo=images)
 > Here's a bunch of shit i guess lmao idk
 ```markdown 
 test 
 test
 test
+|1|2|3|
+|-|-|-|
+|a|b|c|
+
+| uneven rows | test | yes |
+|-|-|-|
+| sosiska | dinozavri | suda pihaem |
+| sosiska 2 | vitalya 2 | brat 2 |
 *** test ***
 piss 
 cock
@@ -87,6 +106,13 @@ __cock__
 > __cum__
 __mashup__
 
+| # sosiska | sosiska | suda pihaem |
+|-|-|-|
+| # 2 | chuvak ya ukral tvayu sardelku ))0)))0))))))) | __blya ((9((9((9)__ |
+| # azazaz lalka sasI | test | test |
 TEXT
-    )+Markdown::QuoteTranslator+Markdown::LeftmostTagTranslator+Markdown::LinearTagTranslator)
+    )+Markdown::QuoteTranslator+Markdown::LeftmostTagTranslator+Markdown::LinearTagTranslator+Markdown::TableTranslator+Markdown::BackslashTranslator)
     .to_html
+write = File.new("/tmp/test.html","w")
+write.write(test)
+write.close