diff options
author | Noah Campbell <noahcampbell@gmail.com> | 2013-11-05 22:28:06 +0000 |
---|---|---|
committer | Noah Campbell <noahcampbell@gmail.com> | 2013-11-05 22:28:06 +0000 |
commit | 86233c00a0a04e8f0130a5970de8d40e6738ef74 (patch) | |
tree | 772e16697927e7bc564e640f6524781699437f02 /transform | |
parent | 1cebce12ad2335e1140646763dd56009c57d6495 (diff) |
Remove the hugo-nav function
Remove the hugo-nav since it relied on a slow library. The current
build reimplements the absurl functionality based on string replace.
Discovered that my prior implementation missed the requirement for
making absolute paths (/path) absolute with the host, whereas a relative
path is left untouched. Updated the test cases to support this if this
is reimplemented.
Diffstat (limited to 'transform')
-rw-r--r-- | transform/absurl.go | 19 | ||||
-rw-r--r-- | transform/chain.go | 27 | ||||
-rw-r--r-- | transform/chain_test.go | 33 | ||||
-rw-r--r-- | transform/nav.go | 12 | ||||
-rw-r--r-- | transform/nav_test.go | 58 | ||||
-rw-r--r-- | transform/posttrans_test.go | 14 |
6 files changed, 49 insertions, 114 deletions
diff --git a/transform/absurl.go b/transform/absurl.go index f66edab9d..5967e597e 100644 --- a/transform/absurl.go +++ b/transform/absurl.go @@ -3,18 +3,29 @@ package transform import ( htmltran "code.google.com/p/go-html-transform/html/transform" "net/url" + "bytes" ) -func AbsURL(absURL string) (trs []*htmltran.Transform, err error) { +func AbsURL(absURL string) (trs []link, err error) { var baseURL *url.URL if baseURL, err = url.Parse(absURL); err != nil { return } - if trs, err = absUrlify(baseURL, elattr{"a", "href"}, elattr{"script", "src"}); err != nil { - return - } + var ( + srcdq = []byte(" src=\""+baseURL.String()+"/") + hrefdq = []byte(" href=\""+baseURL.String()+"/") + srcsq = []byte(" src='"+baseURL.String()+"/") + hrefsq = []byte(" href='"+baseURL.String()+"/") + ) + trs = append(trs, func(content []byte) []byte { + content = bytes.Replace(content, []byte(" src=\"/"), srcdq, -1) + content = bytes.Replace(content, []byte(" src='/"), srcsq, -1) + content = bytes.Replace(content, []byte(" href=\"/"), hrefdq, -1) + content = bytes.Replace(content, []byte(" href='/"), hrefsq, -1) + return content + }) return } diff --git a/transform/chain.go b/transform/chain.go index a4929b70d..fb3c2985c 100644 --- a/transform/chain.go +++ b/transform/chain.go @@ -1,25 +1,30 @@ package transform import ( - htmltran "code.google.com/p/go-html-transform/html/transform" + "bytes" "io" ) -type chain []*htmltran.Transform +type trans func([]byte) []byte -func NewChain(trs ...*htmltran.Transform) chain { +type link trans + +type chain []link + +func NewChain(trs ...link) chain { return trs } func (c *chain) Apply(w io.Writer, r io.Reader) (err error) { - var tr *htmltran.Transformer - - if tr, err = htmltran.NewFromReader(r); err != nil { - return + buffer := new(bytes.Buffer) + buffer.ReadFrom(r) + b := buffer.Bytes() + for _, tr := range *c { + b = tr(b) } - - tr.ApplyAll(*c...) - - return tr.Render(w) + buffer.Reset() + buffer.Write(b) + buffer.WriteTo(w) + return } diff --git a/transform/chain_test.go b/transform/chain_test.go index 594b5a5d4..94135f6d7 100644 --- a/transform/chain_test.go +++ b/transform/chain_test.go @@ -5,41 +5,26 @@ import ( "testing" ) -func TestChainZeroTransformers(t *testing.T) { - tr := NewChain() - in := new(bytes.Buffer) - out := new(bytes.Buffer) - if err := tr.Apply(in, out); err != nil { - t.Errorf("A zero transformer chain returned an error.") - } -} - -func TestChainOneTransformer(t *testing.T) { - absURL, _ := AbsURL("http://base") - tr := NewChain(absURL...) - apply(t.Errorf, tr, abs_url_tests) -} - const H5_JS_CONTENT_ABS_URL_WITH_NAV = "<!DOCTYPE html><html><head><script src=\"/foobar.js\"></script></head><body><nav><ul><li hugo-nav=\"section_0\"></li><li hugo-nav=\"section_1\"></li></ul></nav><article>content <a href=\"/foobar\">foobar</a>. Follow up</article></body></html>" -const CORRECT_OUTPUT_SRC_HREF_WITH_NAV = "<!DOCTYPE html><html><head><script src=\"http://two/foobar.js\"></script></head><body><nav><ul><li hugo-nav=\"section_0\"></li><li hugo-nav=\"section_1\" class=\"active\"></li></ul></nav><article>content <a href=\"http://two/foobar\">foobar</a>. Follow up</article></body></html>" +const CORRECT_OUTPUT_SRC_HREF_WITH_NAV = "<!DOCTYPE html><html><head><script src=\"http://two/foobar.js\"></script></head><body><nav><ul><li hugo-nav=\"section_0\"></li><li hugo-nav=\"section_1\"></li></ul></nav><article>content <a href=\"http://two/foobar\">foobar</a>. Follow up</article></body></html>" var two_chain_tests = []test{ {H5_JS_CONTENT_ABS_URL_WITH_NAV, CORRECT_OUTPUT_SRC_HREF_WITH_NAV}, } -func TestChainTwoTransformer(t *testing.T) { - absURL, _ := AbsURL("http://two") - nav := NavActive("section_1", "hugo-nav") - tr := NewChain(append(absURL, nav...)...) - apply(t.Errorf, tr, two_chain_tests) +func TestChainZeroTransformers(t *testing.T) { + tr := NewChain() + in := new(bytes.Buffer) + out := new(bytes.Buffer) + if err := tr.Apply(in, out); err != nil { + t.Errorf("A zero transformer chain returned an error.") + } } func BenchmarkChain(b *testing.B) { - absURL, _ := AbsURL("http://two") - nav := NavActive("section_1", "hugo-nav") - tr := NewChain(append(absURL, nav...)...) + tr := NewChain(absURL...) b.ResetTimer() for i := 0; i < b.N; i++ { diff --git a/transform/nav.go b/transform/nav.go deleted file mode 100644 index 7783b6175..000000000 --- a/transform/nav.go +++ /dev/null @@ -1,12 +0,0 @@ -package transform - -import ( - htmltran "code.google.com/p/go-html-transform/html/transform" - "fmt" -) - -func NavActive(section, attrName string) (tr []*htmltran.Transform) { - ma := htmltran.MustTrans(htmltran.ModifyAttrib("class", "active"), fmt.Sprintf("li[%s=%s]", attrName, section)) - tr = append(tr, ma) - return -} diff --git a/transform/nav_test.go b/transform/nav_test.go deleted file mode 100644 index 372d3f594..000000000 --- a/transform/nav_test.go +++ /dev/null @@ -1,58 +0,0 @@ -package transform - -import ( - "bytes" - "strings" - "testing" -) - -const HTML_WITH_NAV = `<!DOCTYPE html> -<html> -<head></head> -<body> -<nav> - <ul class="nav navbar-nav"> - <li hugo-nav="section_1"><a href="#">Section 1</a></li> - <li hugo-nav="section_2"><a href="#">Section 2</a></li> - </ul> -</nav> -</body> -</html> -` -const EXPECTED_HTML_WITH_NAV_1 = `<!DOCTYPE html><html><head></head> -<body> -<nav> - <ul class="nav navbar-nav"> - <li hugo-nav="section_1"><a href="#">Section 1</a></li> - <li hugo-nav="section_2" class="active"><a href="#">Section 2</a></li> - </ul> -</nav> - - -</body></html>` - -func TestSetNav(t *testing.T) { - trs := NavActive("section_2", "hugo-nav") - chain := NewChain(trs...) - out := new(bytes.Buffer) - if err := chain.Apply(out, strings.NewReader(HTML_WITH_NAV)); err != nil { - t.Errorf("Unexpected error in Apply() for NavActive: %s", err) - } - - expected := EXPECTED_HTML_WITH_NAV_1 - if out.String() != expected { - t.Errorf("NavActive.Apply output expected and got:\n%q\n%q", expected, out.String()) - } -} - -func BenchmarkTransform(b *testing.B) { - tr := NavActive("section_2", "hugo-nav") - chain := NewChain(tr...) - out := new(bytes.Buffer) - for i := 0; i < b.N; i++ { - if err := chain.Apply(out, strings.NewReader(HTML_WITH_NAV)); err != nil { - b.Errorf("Unexpected error in Apply() for NavActive: %s", err) - } - out.Reset() - } -} diff --git a/transform/posttrans_test.go b/transform/posttrans_test.go index 12f7df1f6..d7d06fee7 100644 --- a/transform/posttrans_test.go +++ b/transform/posttrans_test.go @@ -6,14 +6,18 @@ import ( "testing" ) -const H5_JS_CONTENT_DOUBLE_QUOTE = "<!DOCTYPE html><html><head><script src=\"foobar.js\"></script></head><body><nav><h1>title</h1></nav><article>content <a href='/foobar'>foobar</a>. Follow up</article></body></html>" -const H5_JS_CONTENT_SINGLE_QUOTE = "<!DOCTYPE html><html><head><script src='foobar.js'></script></head><body><nav><h1>title</h1></nav><article>content <a href='/foobar'>foobar</a>. Follow up</article></body></html>" +const H5_JS_CONTENT_DOUBLE_QUOTE = "<!DOCTYPE html><html><head><script src=\"foobar.js\"></script><script src=\"/barfoo.js\"></script></head><body><nav><h1>title</h1></nav><article>content <a href=\"foobar\">foobar</a>. <a href=\"/foobar\">Follow up</a></article></body></html>" + +const H5_JS_CONTENT_SINGLE_QUOTE = "<!DOCTYPE html><html><head><script src='foobar.js'></script><script src='/barfoo.js'></script></head><body><nav><h1>title</h1></nav><article>content <a href='foobar'>foobar</a>. <a href='/foobar'>Follow up</a></article></body></html>" + const H5_JS_CONTENT_ABS_URL = "<!DOCTYPE html><html><head><script src=\"http://user@host:10234/foobar.js\"></script></head><body><nav><h1>title</h1></nav><article>content <a href=\"https://host/foobar\">foobar</a>. Follow up</article></body></html>" // URL doesn't recognize authorities. BUG? //const H5_JS_CONTENT_ABS_URL = "<!DOCTYPE html><html><head><script src=\"//host/foobar.js\"></script></head><body><nav><h1>title</h1></nav><article>content <a href=\"https://host/foobar\">foobar</a>. Follow up</article></body></html>" -const CORRECT_OUTPUT_SRC_HREF = "<!DOCTYPE html><html><head><script src=\"http://base/foobar.js\"></script></head><body><nav><h1>title</h1></nav><article>content <a href=\"http://base/foobar\">foobar</a>. Follow up</article></body></html>" +const CORRECT_OUTPUT_SRC_HREF_DQ = "<!DOCTYPE html><html><head><script src=\"foobar.js\"></script><script src=\"http://base/barfoo.js\"></script></head><body><nav><h1>title</h1></nav><article>content <a href=\"foobar\">foobar</a>. <a href=\"http://base/foobar\">Follow up</a></article></body></html>" + +const CORRECT_OUTPUT_SRC_HREF_SQ = "<!DOCTYPE html><html><head><script src='foobar.js'></script><script src='http://base/barfoo.js'></script></head><body><nav><h1>title</h1></nav><article>content <a href='foobar'>foobar</a>. <a href='http://base/foobar'>Follow up</a></article></body></html>" func TestAbsUrlify(t *testing.T) { tr, _ := AbsURL("http://base") @@ -27,8 +31,8 @@ type test struct { } var abs_url_tests = []test{ - {H5_JS_CONTENT_DOUBLE_QUOTE, CORRECT_OUTPUT_SRC_HREF}, - {H5_JS_CONTENT_SINGLE_QUOTE, CORRECT_OUTPUT_SRC_HREF}, + {H5_JS_CONTENT_DOUBLE_QUOTE, CORRECT_OUTPUT_SRC_HREF_DQ}, + {H5_JS_CONTENT_SINGLE_QUOTE, CORRECT_OUTPUT_SRC_HREF_SQ}, {H5_JS_CONTENT_ABS_URL, H5_JS_CONTENT_ABS_URL}, } |