+import(
+ "fmt"
+ "os"
+
+ "io"
+ "io/ioutil"
+ "bufio"
+
+ "strings"
+ "strconv"
+
+ "net/http"
+ "net/url"
+
+ "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
+
+ red [3]byte
+ bred [3]byte
+
+ green [3]byte
+ bgreen [3]byte
+
+ yellow [3]byte
+ byellow [3]byte
+
+ blue [3]byte
+ bblue [3]byte
+
+ purple [3]byte
+ bpurple [3]byte
+
+ cyan [3]byte
+ bcyan [3]byte
+
+ white [3]byte
+ bwhite [3]byte
+
+ fg [3]byte
+ bg [3]byte
+}
+
+/* the default "control" ANSI color set (boring) */
+var ansiColors = ktPalette {
+
+ black : [3]byte {0,0,0},
+ bblack : [3]byte {128,128,128},
+
+ red : [3]byte {128,0,0},
+ bred : [3]byte {255,0,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},
+
+ blue : [3]byte {0,0,128},
+ bblue : [3]byte {0,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},
+
+ white : [3]byte {128,128,128},
+ bwhite : [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 = "✗"
+
+/* 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
+}
+
+var httpdStatus bool
+
+func ktInit(dirPrepend string, port int, colorfilePath string) error {
+
+ 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)
+
+ 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)
+
+ if pal == ansiColors {}
+
+ 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)
+
+ 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("@g[%s]@{-}\n", checkM)
+
+ return nil
+}
+
+func transposePage(writer http.ResponseWriter, req *http.Request) {
+
+ 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)
+
+ if len(cur) < 7 {
+
+
+ } else if(cur[0:6] == "href=\\") {
+
+ urlStr := cur[7:strings.Index(cur[7:], "\\") + 7]
+
+ 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
+}