golang-fave/engine/builder/data_table.go

package builder

import (
	"fmt"
	"html"
	"math"
	"strconv"

	"golang-fave/engine/sqlw"
	"golang-fave/engine/wrapper"
)

type DataTableRow struct {
	DBField     string
	DBExp       string
	NameInTable string
	Classes     string
	CallBack    func(values *[]string) string
}

func DataTable(
	wrap *wrapper.Wrapper,
	table string,
	order_by string,
	order_way string,
	data *[]DataTableRow,
	action func(values *[]string) string,
	pagination_url string,
	custom_sql_count func() (int, error),
	custom_sql_data func(limit_offset int, pear_page int) (*sqlw.Rows, error),
	pagination_enabled bool,
) string {
	var num int
	var err error

	if pagination_enabled {
		if custom_sql_count != nil {
			num, err = custom_sql_count()
		} else {
			err = wrap.DB.QueryRow("SELECT COUNT(*) FROM `" + table + "`;").Scan(&num)
			if err != nil {
				return ""
			}
		}
	} else {
		num = 0
	}

	pear_page := 10
	max_pages := int(math.Ceil(float64(num) / float64(pear_page)))
	curr_page := 1
	p := wrap.R.URL.Query().Get("p")
	if p != "" {
		pi, err := strconv.Atoi(p)
		if err != nil {
			curr_page = 1
		} else {
			if pi < 1 {
				curr_page = 1
			} else if pi > max_pages {
				curr_page = max_pages
			} else {
				curr_page = pi
			}
		}
	}
	limit_offset := curr_page*pear_page - pear_page
	result := `<table id="cp-table-` + table + `" class="table data-table table-striped table-bordered table-hover table_` + table + `">`
	result += `<thead>`
	result += `<tr>`
	qsql := ""
	if custom_sql_data == nil {
		qsql = "SELECT"
	}
	for i, column := range *data {
		if column.NameInTable != "" {
			classes := column.Classes
			if classes != "" {
				classes = " " + classes
			}
			result += `<th scope="col" class="col_` + column.DBField + classes + `">` + html.EscapeString(column.NameInTable) + `</th>`
		}
		if custom_sql_data == nil {
			if column.DBExp == "" {
				qsql += " `" + column.DBField + "`"
			} else {
				qsql += " " + column.DBExp + " as `" + column.DBField + "`"
			}
			if i+1 < len(*data) {
				qsql += ","
			}
		}
	}
	if custom_sql_data == nil {
		qsql += " FROM `" + table + "` ORDER BY `" + order_by + "` " + order_way + " LIMIT ?, ?;"
	}
	if action != nil {
		result += `<th scope="col" class="col_action">&nbsp;</th>`
	}
	result += `</tr>`
	result += `</thead>`
	result += `<tbody>`
	if num > 0 || !pagination_enabled {
		have_records := false
		var rows *sqlw.Rows
		var err error
		if custom_sql_data == nil {
			rows, err = wrap.DB.Query(qsql, limit_offset, pear_page)
		} else {
			rows, err = custom_sql_data(limit_offset, pear_page)
		}
		if err == nil {
			values := make([]string, len(*data))
			scan := make([]interface{}, len(values))
			for i := range values {
				scan[i] = &values[i]
			}
			for rows.Next() {
				err = rows.Scan(scan...)
				if err == nil {
					if !have_records {
						have_records = true
					}
					result += `<tr>`
					for i, val := range values {
						if (*data)[i].NameInTable != "" {
							classes := (*data)[i].Classes
							if classes != "" {
								classes = " " + classes
							}
							if (*data)[i].CallBack == nil {
								result += `<td class="col_` + (*data)[i].DBField + classes + `">` + html.EscapeString(string(val)) + `</td>`
							} else {
								result += `<td class="col_` + (*data)[i].DBField + classes + `">` + (*data)[i].CallBack(&values) + `</td>`
							}
						}
					}
					if action != nil {
						result += `<td class="col_action">` + action(&values) + `</td>`
					}
					result += `</tr>`
				}
			}
			rows.Close()
		}
		if !have_records {
			result += `<tr><td colspan="50">No data</td></tr>`
		}
	} else {
		result += `<tr><td colspan="50">No data</td></tr>`
	}
	result += `</tbody></table>`

	// Show page navigation only if pages more then one
	if pagination_enabled {
		if max_pages > 1 {
			result += `<nav>`
			result += `<ul class="pagination" style="margin-bottom:0px;">`
			class := ""
			if curr_page <= 1 {
				class = " disabled"
			}
			result += `<li class="page-item` + class + `">`
			result += `<a class="page-link" href="` + pagination_url + `?p=` + fmt.Sprintf("%d", curr_page-1) + `" aria-label="Previous">`
			result += `<span aria-hidden="true">&laquo;</span>`
			result += `<span class="sr-only">Previous</span>`
			result += `</a>`
			result += `</li>`
			for i := 1; i <= max_pages; i++ {
				class = ""
				if i == curr_page {
					class = " active"
				}
				result += `<li class="page-item` + class + `">`
				result += `<a class="page-link" href="` + pagination_url + `?p=` + fmt.Sprintf("%d", i) + `">` + fmt.Sprintf("%d", i) + `</a>`
				result += `</li>`
			}
			class = ""
			if curr_page >= max_pages {
				class = " disabled"
			}
			result += `<li class="page-item` + class + `">`
			result += `<a class="page-link" href="` + pagination_url + `?p=` + fmt.Sprintf("%d", curr_page+1) + `" aria-label="Next">`
			result += `<span aria-hidden="true">&raquo;</span>`
			result += `<span class="sr-only">Next</span>`
			result += `</a>`
			result += `</li>`
			result += `</ul>`
			result += `</nav>`
		}
	}

	return result
}