Add files via upload

Author: technomancy7

nt.nim

@@ -0,0 +1,220 @@
+import os, strutils, db_sqlite, parseopt
+
+const VERSION = "0.12"
+
+var
+    db: DbConn
+
+proc ntcreate(group:string) =
+    db.exec(sql"create table ? (name text, snippet text, language text)", group)
+
+proc ntadd(group, name, language, snippet: string) =
+    if not db.tryExec(sql"insert into ? (name, snippet, language) values (?, ?, ?)", group, name, snippet, language):
+        db.exec(sql"create table ? (name text, language text, snippet text)", group)
+        if not db.tryExec(sql"insert into ? (name, snippet, language) values (?, ?, ?)", group, name, snippet, language):
+            echo "Failed to insert... "
+            dbError(db)
+    
+proc ntdel(group, line: string) =
+    db.exec(sql"delete from ? where name = ?", group, line)
+        
+proc ntget(group, line: string) =
+    var q = "SELECT * FROM "&group&" WHERE name like '"&line&"'"
+    for x in db.fastRows(sql(q)):
+        echo x[0]&" ("&x[1]&")"
+        echo x[2]
+        echo "----"
+        
+proc ntshow(group, line: string) =
+    var q = "SELECT * FROM "&group&" WHERE name = '"&line&"'"
+    for x in db.fastRows(sql(q)):
+        echo x[2]
+        break
+        
+proc ntgetl(group, line: string) =
+    var q = "SELECT * FROM "&group&" where language = '"&line&"'"
+    for x in db.fastRows(sql(q)):
+        echo x
+
+proc ntgetc(group, line: string) =
+    var q = "SELECT * FROM "&group&" where snippet like '%"&line&"%'"
+    for x in db.fastRows(sql(q)):
+        echo x
+        
+proc ntedit(group, name, key, line: string) =
+    db.exec(sql"update ? set ? = ? where name = ?", group, key, line, name)
+    
+proc ntls(group: string) =
+    var q = "SELECT * FROM "&group
+    for x in db.fastRows(sql(q)):
+        echo x[0]&" ("&x[1]&")"
+        echo x[2]
+        echo "----"
+        
+proc ntls() =
+    var fq = "SELECT * from sqlite_master"
+    for x in db.fastRows(sql(fq)):
+        var q = "SELECT * FROM "&x[1]
+        for t in db.fastRows(sql(q)):
+            echo t[0]&" ("&t[1]&")"
+            echo t[2]
+            echo "----"
+            
+proc ntlst() =
+    var fq = "SELECT * from sqlite_master"
+    for x in db.fastRows(sql(fq)):
+        echo x[1]
+            
+proc ntdrop(group: string) =
+    db.exec(sql"drop table ?", group)
+    
+proc ntexec(line: string, catch: bool = false) =
+    if catch:
+        if not db.tryExec(sql(line)):
+            dbError(db)
+    else:
+        for x in db.fastRows(sql(line)):
+            echo x
+                
+proc spinLine() =
+    var
+        p = initOptParser()
+        command, table, lang, code, name: string
+        args = newSeq[string]()
+            
+    while true:
+        p.next()
+        case p.kind
+        of cmdEnd: break
+        of cmdShortOption, cmdLongOption:
+            if p.key == "version" or p.key == "v":
+                echo "NT "&VERSION
+                return
+                
+            if p.val != "":
+                if p.key == "table" or p.key == "t":
+                    table = p.val
+                if p.key == "lang" or p.key=="l":
+                    lang = p.val
+                if p.key == "code" or p.key == "c":
+                    code = p.val
+                if p.key == "name" or p.key == "n":
+                    name = p.val
+        of cmdArgument:
+            if command == "":
+                command = p.key
+            else:
+                args.add(p.key)
+                
+    if table == "":
+        table = "nt"
+
+    case command:
+        of "help":
+            echo "-------"
+            echo "NT: Nim Text storage "&VERSION
+            echo ""
+            echo "Variables: table/t, lang/l, code/c, name/n"
+            echo "Table is given a default if not supplied."
+            echo ""
+            echo "Commands:"
+            echo "\tadd/insert/a/i [-t:table_name] [-n:entry_name] [-l:code_language] [-c:code_block]"
+            echo "\t\t Inserts line.\n\t\t nt add -t:pystuff -n:def -c:def bork(): pass -l:python3"
+            echo ""
+            echo "\tedit/e [key_to_edit] new_input [-n:name_of_entry] [-n:name_of_table]"
+            echo "\t\t Edits a line\n\t\t Renaming: nt edit name test -n:tsst\n\t\t\t Changing the snippet: nt edit code 'def bork(): pass' -n:test"
+            echo ""
+            echo "\tdel/d [-t:table_name] [-n:entry_name]"
+            echo "\t\t Deletes an entry from a table.\n\t\t nt del -t:pystuff -n:def"
+            echo ""
+            echo "\texec/eval [...sql_code]"
+            echo "\t\tRuns raw SQL code on the database."
+            echo ""
+            echo "\tdrop/dr [-t:table]"
+            echo "\t\t Deletes an entire table."
+            echo ""
+            echo "\ttables"
+            echo "\t\tLists tables in the database."
+            echo ""
+            echo "\tlist/ls [-t:table]"
+            echo "\t\tLists entries in a table. If table=* then lists all entries in all tables."
+            echo ""
+            echo "\tget/g [-n:name] OR getc/gc [-c:code_block] OR getl/gl [-l:language] WITH [-t:table_name]"
+            echo "\t\t Finds entries based on search"
+            echo ""
+            echo "\tshow/s [-n:name] [-t:table_name]"
+            echo "\t\tOutput entry with name matching."
+            
+        of "exec", "eval", "sql":
+            ntexec(args.join(" "), false)
+        of "add", "a", "insert":
+            if lang == "":
+                lang = "none"
+            if name == "" or code == "" or table == "":
+                echo "Value(s) missing for `add`: name, code, table"
+            else:
+                ntadd(table, name, lang, code)
+        of "edit", "e", "mod", "modify", "m":
+            ntedit(table, name, args[0], args[1..args.len-1].join(" "))
+        of "del", "d":
+            if table == "" or name == "":
+                echo "Value(s) missing for `del`: table, name"
+            else:
+                ntdel(table, name)
+        of "drop", "dr":
+            ntdrop(table)
+        of "tables":
+            ntlst()
+        of "list", "ls":
+            if table == "*":
+                ntls()
+            else:
+                ntls(table)
+        of "show", "s":
+            if name == "" and args.len == 0:
+                echo "Value missing: name"
+                return
+
+            if name == "":
+                name = args.join(" ")
+                
+            ntshow(table, name)
+        of "get", "g":
+            if name == "" and args.len == 0:
+                echo "Value missing: name"
+                return
+
+            if name == "":
+                name = args.join(" ")
+                
+            ntget(table, name)
+        of "getc", "gc", "getcode":
+            if code == "" and args.len == 0:
+                echo "Value missing: code"
+                return
+
+            if code == "":
+                code = args.join(" ")
+                
+            ntgetc(table, code)
+            
+        of "getl", "gl", "getlang", "getlanguage":
+            if lang == "" and args.len == 0:
+                echo "Value missing: lang"
+                return
+
+            if lang == "":
+                lang = args.join(" ")
+
+            ntgetl(table, lang)
+            
+        of "create", "cr":
+            if table == "" or table == "nt":
+                echo "Can't create empty table."
+            else:
+                ntcreate(table)
+
+            
+db = open(getHomeDir()&"/nt.db", "", "", "")
+spinLine()
+db.close()