示例#1
0
文件: http.go 项目: exequielbc/clock
func recordHandler(w http.ResponseWriter, r *http.Request) {
	db, err := openDB()
	checkErr(err)
	path := r.URL.Path[len("/entry/"):]
	switch path {
	case "create":
		log.Print("Creating a new entry.")
		s, err := db.Prepare("INSERT INTO records(date,hours,project_id) values(?,?,?)")
		checkErr(err)
		h := r.FormValue("hours")
		m := r.FormValue("minutes")
		pID := r.FormValue("project")
		if h == "" || m == "" || pID == "" {
			errorHTTP(errors.New("A field you entered was blank."), w)
			return
		}
		duration, err := time.ParseDuration(fmt.Sprintf("%sh%sm", h, m))
		errorHTTP(err, w)
		project, err := strconv.ParseInt(pID, 10, 32)
		errorHTTP(err, w)
		_, err = s.Exec(time.Now(), duration.Hours(), project)
		errorHTTP(err, w)
		http.Redirect(w, r, "/entry/", http.StatusFound)
	default:
		projects, err := loadProjects(db)
		errorHTTP(err, w)
		recordsSQL, err := db.Query("SELECT * FROM records")
		errorHTTP(err, w)
		records := []Record{}
		total := make(map[string]float64)
		for recordsSQL.Next() {
			r := Record{}
			r.Project = &Project{}
			recordsSQL.Scan(&r.ID, &r.Date, &r.Hours, &r.Project.ID)
			for _, p := range projects {
				if r.Project.ID == p.ID {
					r.Project.Name = p.Name
				}
			}
			total[r.Project.Name] += r.Hours
			records = append(records, r)
		}
		data := struct {
			Projects []Project
			Records  []Record
			Total    map[string]float64
		}{
			projects,
			records,
			total,
		}
		err = templates.ExecuteTemplate(w, "logging.html", data)
		errorHTTP(err, w)
	}
}
示例#2
0
func getReport(w http.ResponseWriter, r *http.Request) {
	res := &Reports{}

	limit := 20
	if r.FormValue("count") != "" {
		limit, _ = strconv.Atoi(r.FormValue("count"))
	}

	dh := db.GetHandler()
	defer dh.Close()

	order := dh.OrderBy("start", genmai.DESC).Limit(limit + 1)

	var cond [][]interface{}

	re := regexp.MustCompile(`^/api/v1/reports/(\d+)$`)

	reportId := ""
	if m := re.FindStringSubmatch(r.URL.Path); m != nil {
		reportId = m[1]
	}

	if reportId != "" {
		cond = append(cond, []interface{}{"id", "=", reportId})
	}

	if projectId := r.FormValue("projectId"); projectId != "" {
		cond = append(cond, []interface{}{"project_id", "=", projectId})
	}

	if until := r.FormValue("until"); until != "" {
		u, _ := strconv.ParseInt(until, 10, 64)
		cond = append(cond, []interface{}{"start", "<=", time.Unix(u, 0)})
	}

	if since := r.FormValue("since"); since != "" {
		s, _ := strconv.ParseInt(since, 10, 64)
		cond = append(cond, []interface{}{"start", ">=", time.Unix(s, 0)})
	}

	if status := r.FormValue("status"); status != "" {
		cond = append(cond, []interface{}{"status", "=", status})
	}

	where := &genmai.Condition{}
	for i, c := range cond {
		if i < 1 {
			where = dh.Where(c[0], c[1], c[2])
		} else {
			where = where.And(dh.Where(c[0], c[1], c[2]))
		}
	}

	var reports []db.Report

	dh.Select(&reports, where, order)

	if len(reports) > limit {
		res.NextStart = reports[limit].Start.Unix()
		reports = reports[:len(reports)-1]
	}

	for _, report := range reports {
		r := &Report{
			Id:         report.Id,
			Status:     report.Status,
			Branch:     report.Branch,
			CompareUrl: report.CompareUrl,
			Start:      report.Start.Unix(),
			End:        report.End.Unix(),
		}

		var projects []db.Project
		dh.Select(&projects, dh.Where("id", "=", report.ProjectId))
		project := projects[0]

		r.Project = &Project{
			Id:   project.Id,
			Name: project.Name,
			Repo: project.Repo,
		}

		var commits []db.Commit
		dh.Select(&commits, dh.Where("report_id", "=", report.Id))
		for _, commit := range commits {
			r.Commits = append(r.Commits, &Commit{
				Revision: commit.Revision,
				Author:   commit.Author,
				Message:  commit.Message,
				Url:      commit.Url,
			})
		}

		var users []db.User
		dh.Select(&users, dh.Where("id", "=", report.TriggeredBy))
		user := users[0]
		r.TriggeredBy.Name = user.Name
		r.TriggeredBy.Url = user.Url
		r.TriggeredBy.AvatarUrl = user.AvatarUrl

		var stages []db.Stage
		dh.Select(&stages, dh.Where("report_id", "=", report.Id).And(dh.Where("parent_stage_id", "=", 0)))
		for _, stage := range stages {
			s := &Stage{
				Id:     stage.Id,
				Name:   stage.Name,
				Status: stage.Status,
				Log:    stage.Log,
				Start:  stage.Start.Unix(),
				End:    stage.End.Unix(),
			}

			var childStages []db.Stage
			dh.Select(&childStages, dh.Where("report_id", "=", report.Id).And(dh.Where("parent_stage_id", "=", stage.Id)))
			for _, childStage := range childStages {
				s.Stages = append(s.Stages, &Stage{
					Id:     childStage.Id,
					Name:   childStage.Name,
					Status: childStage.Status,
					Log:    childStage.Log,
					Start:  childStage.Start.Unix(),
					End:    childStage.End.Unix(),
				})
			}

			r.Stages = append(r.Stages, s)
		}

		res.Reports = append(res.Reports, r)
	}

	w.Header().Set("Content-Type", "application/json")
	w.WriteHeader(http.StatusOK)

	var b []byte
	if reportId != "" {
		b, _ = json.Marshal(&struct{ Report *Report }{Report: res.Reports[0]})
	} else {
		b, _ = json.Marshal(res)
	}

	fmt.Fprint(w, string(b))
}