This repository has been archived on 2022-08-15. You can view files and clone it, but cannot push or open issues or pull requests.
shemu/main.go

113 lines
3.2 KiB
Go

package main
import (
"bufio"
"encoding/csv"
"flag"
"fmt"
"io/ioutil"
"log"
"net/http"
"os"
"path"
"path/filepath"
"time"
)
var students [][]string
var rendusDir string
var title string
func Serve(w http.ResponseWriter, r *http.Request) {
log.Printf("Handling %s request from %s: %s [%s]\n", r.Method, r.RemoteAddr, r.URL.Path, r.UserAgent())
w.Header().Set("Content-Type", "text/html")
w.Write([]byte(`<!DOCTYPE html>
<html>
<head>
<title>` + title + `</title>
<meta http-equiv="Content-Type" content="text/html; charset=UTF-8">
<link rel="stylesheet" href="https://maxcdn.bootstrapcdn.com/bootstrap/3.3.7/css/bootstrap.min.css" integrity="sha384-BVYiiSIFeK1dGmJRAkycuHAHRg32OmUcww7on3RYdg4Va+PmSTsz/K68vbdEjh4u" crossorigin="anonymous">
</head>
<body class="container">
<h1>` + title + `</h1>
<table class="table table-striped table-hover table-condensed">`))
if dirs, err := ioutil.ReadDir(rendusDir); err == nil {
w.Write([]byte(`<thead><tr><td></td>`))
for _, dir := range dirs {
if dir.IsDir() {
w.Write([]byte(`<th>` + dir.Name() + `</th>`))
}
}
w.Write([]byte(`</tr></thead><tbody>`))
for _, student := range students {
login := student[2]
w.Write([]byte(`<tr><th>` + login + `</th>`))
for _, dir := range dirs {
if dir.IsDir() {
if fi, err := os.Stat(path.Join(rendusDir, dir.Name(), login)); err == nil {
w.Write([]byte(`<td class="success">` + fi.ModTime().Format(time.UnixDate) + `</td>`))
} else {
w.Write([]byte(`<td class="danger"><span class="glyphicon glyphicon-remove" aria-hidden="true"></span></td>`))
}
}
}
w.Write([]byte(`</tr>`))
}
}
w.Write([]byte(` </tbody></table>
</body>
</html>
`))
}
func main() {
var studentsFile string
var bind = flag.String("bind", "0.0.0.0:8081", "Bind port/socket")
flag.StringVar(&studentsFile, "students", "./students.csv", "Path to a CSV file containing in the third column the name of the directory to look for")
flag.StringVar(&rendusDir, "path", "./rendu/", "Path to the submissions directory (each subdirectory is a project)")
flag.StringVar(&title, "title", "Rendus VIRLI", "Title of the page")
flag.Parse()
var err error
if rendusDir, err = filepath.Abs(rendusDir); err != nil {
log.Fatal(err)
} else if dirs, err := ioutil.ReadDir(rendusDir); err != nil {
log.Fatal(err)
} else {
var projects []string
nbProj := 0
for _, dir := range dirs {
if dir.IsDir() {
projects = append(projects, dir.Name())
nbProj++
}
}
log.Println(nbProj, "projects found:", projects)
}
// Read and parse students list
log.Println("Reading students files...")
if studentsFile, err = filepath.Abs(studentsFile); err != nil {
log.Fatal(err)
} else if fi, err := os.Open(studentsFile); err != nil {
log.Fatal(err)
} else {
r := csv.NewReader(bufio.NewReader(fi))
if students, err = r.ReadAll(); err != nil {
log.Fatal(err)
}
}
log.Println(len(students), "students loaded.")
log.Println("Registering handlers...")
http.HandleFunc("/", Serve)
log.Println(fmt.Sprintf("Ready, listening on %s", *bind))
if err := http.ListenAndServe(*bind, nil); err != nil {
log.Fatal("Unable to listen and serve: ", err)
}
}