package main
-import(
- "fmt"
- "os"
+import (
+ "fmt"
+ "os"
- "io"
-// "io/ioutil"
- "bufio"
+ "bufio"
+ "io"
+ "io/ioutil"
- "strings"
- "strconv"
- "text/scanner"
+ "strconv"
+ "strings"
- "net/http"
+ "net/http"
+ "net/url"
- "github.com/wsxiaoys/terminal/color"
+ "github.com/wsxiaoys/terminal/color"
)
/* holds our palette, or 16 ANSI colors (8 normal colors + bright complements)
and two foreground/background colors. colors are 3 byte arrays (RGB) */
type ktPalette struct {
+ black [3]byte
+ bblack [3]byte
- black [3]byte
- bblack [3]byte
+ red [3]byte
+ bred [3]byte
- red [3]byte
- bred [3]byte
+ green [3]byte
+ bgreen [3]byte
- green [3]byte
- bgreen [3]byte
+ yellow [3]byte
+ byellow [3]byte
- yellow [3]byte
- byellow [3]byte
+ blue [3]byte
+ bblue [3]byte
- blue [3]byte
- bblue [3]byte
+ purple [3]byte
+ bpurple [3]byte
- purple [3]byte
- bpurple [3]byte
+ cyan [3]byte
+ bcyan [3]byte
- cyan [3]byte
- bcyan [3]byte
+ white [3]byte
+ bwhite [3]byte
- white [3]byte
- bwhite [3]byte
-
- fg [3]byte
- bg [3]byte
+ fg [3]byte
+ bg [3]byte
}
/* the default "control" ANSI color set (boring) */
-var ansiColors = ktPalette {
+var ansiColors = ktPalette{
- black : [3]byte {0,0,0},
- bblack : [3]byte {128,128,128},
+ black: [3]byte{0, 0, 0},
+ bblack: [3]byte{128, 128, 128},
- red : [3]byte {128,0,0},
- bred : [3]byte {255,0,0},
+ red: [3]byte{128, 0, 0},
+ bred: [3]byte{255, 0, 0},
- green : [3]byte {0,128,0},
- bgreen : [3]byte {0,255,0},
+ green: [3]byte{0, 128, 0},
+ bgreen: [3]byte{0, 255, 0},
- yellow : [3]byte {128,128,0},
- byellow : [3]byte {255,255,0},
+ yellow: [3]byte{128, 128, 0},
+ byellow: [3]byte{255, 255, 0},
- blue : [3]byte {0,0,128},
- bblue : [3]byte {0,0,255},
+ blue: [3]byte{0, 0, 128},
+ bblue: [3]byte{0, 0, 255},
- purple : [3]byte {128,0,128},
- bpurple : [3]byte {255,0,255},
+ purple: [3]byte{128, 0, 128},
+ bpurple: [3]byte{255, 0, 255},
- cyan : [3]byte {0,128,128},
- bcyan : [3]byte {0,255,255},
+ cyan: [3]byte{0, 128, 128},
+ bcyan: [3]byte{0, 255, 255},
- white : [3]byte {128,128,128},
- bwhite : [3]byte {255,255,255},
+ white: [3]byte{128, 128, 128},
+ bwhite: [3]byte{255, 255, 255},
- fg : [3]byte {0,0,0},
- bg : [3]byte {255,255,255},
+ fg: [3]byte{0, 0, 0},
+ bg: [3]byte{255, 255, 255},
}
/* lets get our money's worth out of this utf-8 crap */
var checkM = "✓"
-var xM = "✗"
+var xM = "✗"
/* parses a colorfile, returns palette struct. given a nil file pointer,
returns standard ANSI color set (our "control") */
func parseColors(colorfile *os.File) (pal ktPalette, err error) {
- if colorfile == nil { return ansiColors, nil }
-
- ret := ktPalette {}
- var e errConsumer
-
- scanner := bufio.NewScanner(colorfile)
- for scanner.Scan() {
-
- if strings.Contains(scanner.Text(), "color") &&
- !strings.Contains(scanner.Text(), "!") {
-
- /* i am so sorry */
- cur := strings.Replace(scanner.Text(), " ", "", -1)
- split := strings.Split(cur, ":")
- hexColor := strings.Replace(split[1], "#", "", -1)
- coNumStr := strings.Replace(split[0], "URxvt*color", "", -1)
-
- r, err := strconv.ParseUint(hexColor[0:2], 16, 8); e.Consume(err)
- g, err := strconv.ParseUint(hexColor[2:4], 16, 8); e.Consume(err)
- b, err := strconv.ParseUint(hexColor[4:], 16, 8); e.Consume(err)
-
- colorNo, err := strconv.Atoi(coNumStr); e.Consume(err)
-
- if e.err != nil {
- return ktPalette{}, e.err
- }
-
- switch(colorNo) {
- case 0: ret.black[0] = byte(r); ret.black[1] = byte(g); ret.black[2] = byte(b)
- case 1: ret.red[0] = byte(r); ret.red[1] = byte(g); ret.red[2] = byte(b)
- case 2: ret.green[0] = byte(r); ret.green[1] = byte(g); ret.green[2] = byte(b)
- case 3: ret.yellow[0] = byte(r); ret.yellow[1] = byte(g); ret.yellow[2] = byte(b)
- case 4: ret.blue[0] = byte(r); ret.blue[1] = byte(g); ret.blue[2] = byte(b)
- case 5: ret.purple[0] = byte(r); ret.purple[1] = byte(g); ret.purple[2] = byte(b)
- case 6: ret.cyan[0] = byte(r); ret.cyan[1] = byte(g); ret.cyan[2] = byte(b)
- case 7: ret.white[0] = byte(r); ret.white[1] = byte(g); ret.white[2] = byte(b)
- case 8: ret.bblack[0] = byte(r); ret.bblack[1] = byte(g); ret.bblack[2] = byte(b)
- case 9: ret.bred[0] = byte(r); ret.bred[1] = byte(g); ret.bred[2] = byte(b)
- case 10: ret.bgreen[0] = byte(r); ret.bgreen[1] = byte(g); ret.bgreen[2] = byte(b)
- case 11: ret.byellow[0] = byte(r); ret.byellow[1] = byte(g); ret.byellow[2] = byte(b)
- case 12: ret.bblue[0] = byte(r); ret.bblue[1] = byte(g); ret.bblue[2] = byte(b)
- case 13: ret.bpurple[0] = byte(r); ret.bpurple[1] = byte(g); ret.bpurple[2] = byte(b)
- case 14: ret.bcyan[0] = byte(r); ret.bcyan[1] = byte(g); ret.bcyan[2] = byte(b)
- case 15: ret.bwhite[0] = byte(r); ret.bwhite[1] = byte(g); ret.bwhite[2] = byte(b)
- }
-
- } else if strings.Contains(scanner.Text(), "background") {
-
- hex := strings.Split(scanner.Text(), "#")
- het := hex[1]
-
- r, err := strconv.ParseUint(het[0:2], 16, 8); e.Consume(err)
- g, err := strconv.ParseUint(het[2:4], 16, 8); e.Consume(err)
- b, err := strconv.ParseUint(het[4:], 16, 8); e.Consume(err)
-
- if e.err != nil {
- return ktPalette{}, e.err
- }
-
- ret.bg[0] = byte(r)
- ret.bg[1] = byte(g)
- ret.bg[2] = byte(b)
-
- } else if strings.Contains(scanner.Text(), "foreground") {
-
- hex := strings.Split(scanner.Text(), "#")
- het := hex[1]
-
- r, err := strconv.ParseUint(het[0:2], 16, 8); e.Consume(err)
- g, err := strconv.ParseUint(het[2:4], 16, 8); e.Consume(err)
- b, err := strconv.ParseUint(het[4:], 16, 8); e.Consume(err)
-
- if e.err != nil {
- return ktPalette{}, e.err
- }
-
- ret.fg[0] = byte(r)
- ret.fg[1] = byte(g)
- ret.fg[2] = byte(b)
- }
- }
-
- return ret, e.err
+ if colorfile == nil {
+ return ansiColors, nil
+ }
+
+ ret := ktPalette{}
+ var e errConsumer
+
+ scanner := bufio.NewScanner(colorfile)
+ for scanner.Scan() {
+
+ if strings.Contains(scanner.Text(), "color") &&
+ !strings.Contains(scanner.Text(), "!") {
+
+ /* i am so sorry */
+ cur := strings.Replace(scanner.Text(), " ", "", -1)
+ split := strings.Split(cur, ":")
+ hexColor := strings.Replace(split[1], "#", "", -1)
+ coNumStr := strings.Replace(split[0], "URxvt*color", "", -1)
+
+ r, err := strconv.ParseUint(hexColor[0:2], 16, 8)
+ e.Consume(err)
+ g, err := strconv.ParseUint(hexColor[2:4], 16, 8)
+ e.Consume(err)
+ b, err := strconv.ParseUint(hexColor[4:], 16, 8)
+ e.Consume(err)
+
+ colorNo, err := strconv.Atoi(coNumStr)
+ e.Consume(err)
+
+ if e.err != nil {
+ return ktPalette{}, e.err
+ }
+
+ switch colorNo {
+ case 0:
+ ret.black[0] = byte(r)
+ ret.black[1] = byte(g)
+ ret.black[2] = byte(b)
+ case 1:
+ ret.red[0] = byte(r)
+ ret.red[1] = byte(g)
+ ret.red[2] = byte(b)
+ case 2:
+ ret.green[0] = byte(r)
+ ret.green[1] = byte(g)
+ ret.green[2] = byte(b)
+ case 3:
+ ret.yellow[0] = byte(r)
+ ret.yellow[1] = byte(g)
+ ret.yellow[2] = byte(b)
+ case 4:
+ ret.blue[0] = byte(r)
+ ret.blue[1] = byte(g)
+ ret.blue[2] = byte(b)
+ case 5:
+ ret.purple[0] = byte(r)
+ ret.purple[1] = byte(g)
+ ret.purple[2] = byte(b)
+ case 6:
+ ret.cyan[0] = byte(r)
+ ret.cyan[1] = byte(g)
+ ret.cyan[2] = byte(b)
+ case 7:
+ ret.white[0] = byte(r)
+ ret.white[1] = byte(g)
+ ret.white[2] = byte(b)
+ case 8:
+ ret.bblack[0] = byte(r)
+ ret.bblack[1] = byte(g)
+ ret.bblack[2] = byte(b)
+ case 9:
+ ret.bred[0] = byte(r)
+ ret.bred[1] = byte(g)
+ ret.bred[2] = byte(b)
+ case 10:
+ ret.bgreen[0] = byte(r)
+ ret.bgreen[1] = byte(g)
+ ret.bgreen[2] = byte(b)
+ case 11:
+ ret.byellow[0] = byte(r)
+ ret.byellow[1] = byte(g)
+ ret.byellow[2] = byte(b)
+ case 12:
+ ret.bblue[0] = byte(r)
+ ret.bblue[1] = byte(g)
+ ret.bblue[2] = byte(b)
+ case 13:
+ ret.bpurple[0] = byte(r)
+ ret.bpurple[1] = byte(g)
+ ret.bpurple[2] = byte(b)
+ case 14:
+ ret.bcyan[0] = byte(r)
+ ret.bcyan[1] = byte(g)
+ ret.bcyan[2] = byte(b)
+ case 15:
+ ret.bwhite[0] = byte(r)
+ ret.bwhite[1] = byte(g)
+ ret.bwhite[2] = byte(b)
+ }
+
+ } else if strings.Contains(scanner.Text(), "background") {
+
+ hex := strings.Split(scanner.Text(), "#")
+ het := hex[1]
+
+ r, err := strconv.ParseUint(het[0:2], 16, 8)
+ e.Consume(err)
+ g, err := strconv.ParseUint(het[2:4], 16, 8)
+ e.Consume(err)
+ b, err := strconv.ParseUint(het[4:], 16, 8)
+ e.Consume(err)
+
+ if e.err != nil {
+ return ktPalette{}, e.err
+ }
+
+ ret.bg[0] = byte(r)
+ ret.bg[1] = byte(g)
+ ret.bg[2] = byte(b)
+
+ } else if strings.Contains(scanner.Text(), "foreground") {
+
+ hex := strings.Split(scanner.Text(), "#")
+ het := hex[1]
+
+ r, err := strconv.ParseUint(het[0:2], 16, 8)
+ e.Consume(err)
+ g, err := strconv.ParseUint(het[2:4], 16, 8)
+ e.Consume(err)
+ b, err := strconv.ParseUint(het[4:], 16, 8)
+ e.Consume(err)
+
+ if e.err != nil {
+ return ktPalette{}, e.err
+ }
+
+ ret.fg[0] = byte(r)
+ ret.fg[1] = byte(g)
+ ret.fg[2] = byte(b)
+ }
+ }
+
+ return ret, e.err
}
+var httpdStatus bool
+
func ktInit(dirPrepend string, port int, colorfilePath string) error {
- color.Print("@yparsing colorfile :: @{|}")
- file, err := os.Open(colorfilePath)
- if err != nil {
- color.Printf("@r[%s]@{|} - bad colorfile path\n", xM)
- return fmt.Errorf("%s\n", "bad colorfile path")
- }
+ httpdStatus = false
+
+ color.Print("@yparsing colorfile :: @{|}")
+ file, err := os.Open(colorfilePath)
+ if err != nil {
+ color.Printf("@r[%s]@{|} - bad colorfile path\n", xM)
+ return fmt.Errorf("%s\n", "bad colorfile path")
+ }
- pal, err := parseColors(file)
+ pal, err := parseColors(file)
- if err != nil {
- color.Printf("@r[%s]@{|} - malformed colorfile [%s]\n", xM, err)
- return fmt.Errorf("%s\n", "malformed colorfile")
- }
+ if err != nil {
+ color.Printf("@r[%s]@{|} - malformed colorfile [%s]\n", xM, err)
+ return fmt.Errorf("%s\n", "malformed colorfile")
+ }
- color.Printf("@g[%s]@{|}\n", checkM)
+ color.Printf("@g[%s]@{|}\n", checkM)
- if pal == ansiColors {}
+ if pal == ansiColors {
+ }
- color.Printf("@yverifying & preprocessing colorsets@{|} :: @y[SKIP]\n")
- color.Printf("@ygenerating transpositional colorspace@{|} :: @y[SKIP]\n")
+ color.Printf("@yverifying & preprocessing colorsets@{|} :: @y[SKIP]\n")
+ color.Printf("@ygenerating transpositional colorspace@{|} :: @y[SKIP]\n")
- color.Printf("@ystarting httpd on port @b%d@{|} :: ", port)
- http.HandleFunc("/kt/", transposePage)
+ color.Printf("@ystarting httpd on port @b%d@{|} :: ", port)
+ http.HandleFunc("/kt/", transposePage)
- var portString string
- fmt.Sprintf(portString, ":%d", port)
- err = http.ListenAndServe(portString, nil)
- if err != nil {
+ var portString string
+ fmt.Sprintf(portString, ":%d", port)
+ err = http.ListenAndServe(portString, nil)
+ if err != nil {
- color.Printf("@r[%s]@{-} - run me as root!\n", xM)
- return fmt.Errorf("%s\n", "failed to start httpd")
- }
+ color.Printf("@r[%s]@{-} - run me as root!\n", xM)
+ return fmt.Errorf("%s\n", "failed to start httpd")
+ }
- color.Printf("@g[%s]@{-}\n", checkM)
+ color.Printf("@g[%s]@{-}\n", checkM)
- return nil
+ return nil
}
func transposePage(writer http.ResponseWriter, req *http.Request) {
- fqdn := req.URL.Path[4:]
- targetURL := fmt.Sprintf("http://%s", fqdn)
+ if !httpdStatus {
+
+ httpdStatus = true
+ color.Printf("@g[%s]@{|}\n", checkM)
+ }
+
+ if req.URL.Path == "/kt/" {
+ writer.Write([]byte("wtf"))
+ return
+ }
+
+ fqdn := req.URL.Path[4:]
+ targetURL := fmt.Sprintf("http://%s", fqdn)
+
+ resp, err := http.Get(targetURL)
+
+ if err != nil || resp.StatusCode != 200 {
+
+ io.WriteString(writer, "failed to get that page! -kt\n")
+ io.WriteString(writer, targetURL+"\n")
+
+ io.WriteString(writer, resp.Status)
+ return
+ }
+
+ conType := resp.Header.Get("Content-Type")
+
+ switch conType[0:strings.Index(conType, ";")] {
+
+ case "text/html":
+ writer.Write(transposeHTML(bufio.NewScanner(resp.Body), fqdn))
+
+ case "text/css":
+ writer.Write(transposeCSS(bufio.NewScanner(resp.Body), fqdn))
+
+ default:
+ page, _ := ioutil.ReadAll(resp.Body)
+ writer.Write(page)
+ }
+
+ resp.Body.Close()
+}
+
+/* swap href="" & src="" */
+func transposeHTML(scan *bufio.Scanner, fqdn string) []byte {
+
+ var ret []byte
+ var i int
+
+ scan.Split(bufio.ScanWords)
+ for scan.Scan() {
+
+ i++
+ cur := scan.Text()
+
+ //fmt.Printf("%s\n", cur)
- resp, err := http.Get(targetURL)
- if err != nil {
+ if len(cur) < 7 {
- io.WriteString(writer, "failed to get that page! -kt\n")
- io.WriteString(writer, targetURL)
- return
- }
+ } else if cur[0:6] == "href=\\" {
- var s scanner.Scanner
- s.Init(req.Body)
- //tok := s.Scan
+ urlStr := cur[7 : strings.Index(cur[7:], "\\")+7]
- resp.Body.Close()
+ u, err := url.Parse(urlStr)
+ if err != nil {
+ fmt.Printf("malformed URL: %s\n", urlStr)
+ }
+
+ if u.Host == "" {
+
+ u.Host = fmt.Sprintf("localhost/kt/%s", fqdn)
+ // cur = append(cur[0:6],
+ }
+
+ fmt.Printf("[F] URL: %s // PATH: %s\n", u.Host, u.Path)
+ if u == u {
+ }
+
+ } else if cur[0:5] == "href=" {
+
+ urlStr := cur[6 : strings.Index(cur[6:], "\"")+6]
+
+ u, err := url.Parse(urlStr)
+ if err != nil {
+ fmt.Printf("malformed URL: %s\n", urlStr)
+ }
+
+ if u == u {
+ }
+ fmt.Printf("URL: %s // PATH: %s\n", u.Host, u.Path)
+
+ } else if cur[0:5] == "src=\"" {
+
+ //fmt.Printf("%s\n", cur)
+ urlStr := cur[5 : strings.Index(cur[5:], "\"")+5]
+
+ u, err := url.Parse(urlStr)
+ if err != nil {
+ fmt.Printf("malformed URL: %s\n", urlStr)
+ }
+
+ if u.Host == "" {
+
+ u.Host = fmt.Sprintf("localhost/kt/%s", fqdn)
+ // cur = append(cur[0:6],
+
+ fmt.Printf("[S] URL: %s // PATH: %s\n", u.Host, u.Path)
+ if u == u {
+ }
+ }
+
+ }
+
+ ret = append(ret, byte(' '))
+ ret = append(ret, cur...)
+ }
+
+ fmt.Printf("%d\n", i)
+
+ return ret
+}
+
+func transposeCSS(scan *bufio.Scanner, fqdn string) []byte {
+
+ var ret []byte
+
+ return ret
}
func main() {
- err := ktInit("kolors", 999, "/home/kremlin/.Xresources")
+ err := ktInit("kolors", 999, "/home/kremlin/.Xresources")
- if err != nil {}
+ if err != nil {
+ }
}
type errConsumer struct {
- err error
+ err error
}
func (e *errConsumer) Consume(err error) {
- if e.err == nil && err != nil {
- e.err = err
- }
+ if e.err == nil && err != nil {
+ e.err = err
+ }
}
-