templ.go

/*
 *  This program is free software: you can redistribute it and/or
 *  modify it under the terms of the GNU Affero General Public License as
 *  published by the Free Software Foundation, either version 3 of the
 *  License, or (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  General Public License for more details.
 *
 *  You should have received a copy of the GNU Affero General Public
 *  License along with this program.  If not, see
 *  <http://www.gnu.org/licenses/>.
 *
 *  (c) Vincenzo "KatolaZ" Nicosia 2017 -- <katolaz@freaknet.org>
 *
 *
 *  This file is part of "binnit", a minimal no-fuss pastebin-like
 *  server written in golang
 *
 */

/*
 *
 * minimal Templating support for binnit
 *
 */

package main

import (
	"errors"
	"io/ioutil"
	"os"
	"regexp"
	"strconv"
	"strings"
)

func format_rows(content string) string {

	var ret string

	lines := strings.Split(content, "\n")

	ret += "<table class='content'>"

	for l_num, l := range lines {
		ret += "<tr>\n"
		ret += "<td class='lineno'><pre>" + strconv.Itoa(l_num+1) + "</pre></td>"
		ret += "<td class='line'><pre>" + l + "</pre></td>"
		ret += "</tr>"
	}
	ret += "</table>"
	return ret
}

func prepare_paste_page(title, date, content, templ_dir string) (string, error) {

	s := ""

	// insert header

	head_file := templ_dir + "/header.html"

	f_head, err := os.Open(head_file)
	defer f_head.Close()

	if err == nil {
		cont, err := ioutil.ReadFile(head_file)
		if err == nil {
			s += string(cont)
		}
	}

	// insert content

	// ...Let's read the template
	templ_file := templ_dir + "/templ.html"
	f_templ, err := os.Open(templ_file)
	defer f_templ.Close()

	if cont, err := ioutil.ReadFile(templ_file); err == nil {
		tmpl := string(cont)
		re, _ := regexp.Compile("{{TITLE}}")
		tmpl = string(re.ReplaceAllLiteralString(tmpl, title))

		re, _ = regexp.Compile("{{DATE}}")
		tmpl = string(re.ReplaceAllLiteralString(tmpl, date))

		re, _ = regexp.Compile("{{CONTENT}}")
		tmpl = string(re.ReplaceAllLiteralString(tmpl, format_rows(content)))

		re, _ = regexp.Compile("{{RAW_CONTENT}}")
		tmpl = string(re.ReplaceAllLiteralString(tmpl, content))

		s += tmpl

	} else {
		return "", errors.New("Error opening template file")
	}

	// insert footer
	foot_file := templ_dir + "/footer.html"
	f_foot, err := os.Open(foot_file)
	defer f_foot.Close()

	if err == nil {
		cont, err := ioutil.ReadFile(foot_file)
		if err == nil {
			s += string(cont)
		}
	}
	return s, nil
}