Semaphore/api/projects/projects.go

package projects

import (
	log "github.com/Sirupsen/logrus"
	"github.com/ansible-semaphore/semaphore/db"
	"github.com/ansible-semaphore/semaphore/models"
	"net/http"

	"github.com/ansible-semaphore/semaphore/util"
	"github.com/gorilla/context"
	"github.com/masterminds/squirrel"
)

// GetProjects returns all projects in this users context
func GetProjects(w http.ResponseWriter, r *http.Request) {
	user := context.Get(r, "user").(*models.User)

	query, args, err := squirrel.Select("p.*").
		From("project as p").
		Join("project__user as pu on pu.project_id=p.id").
		Where("pu.user_id=?", user.ID).
		OrderBy("p.name").
		ToSql()

	util.LogWarning(err)
	var projects []models.Project
	if _, err := context.Get(r, "store").(db.Store).Sql().Select(&projects, query, args...); err != nil {
		panic(err)
	}

	util.WriteJSON(w, http.StatusOK, projects)
}

// AddProject adds a new project to the database
func AddProject(w http.ResponseWriter, r *http.Request) {
	var body models.Project

	user := context.Get(r, "user").(*models.User)

	err := util.Bind(w, r, &body)
	if err != nil {
		return
	}

	body, err = context.Get(r, "store").(db.Store).CreateProject(body)
	if err != nil {
		panic(err)
	}

	_, err = context.Get(r, "store").(db.Store).CreateProjectUser(models.ProjectUser{ProjectID: body.ID, UserID: user.ID, Admin: true})

	if err != nil {
		panic(err)
	}

	desc := "Project Created"
	oType := "Project"
	_, err = context.Get(r, "store").(db.Store).CreateEvent(models.Event{
		ProjectID:   &body.ID,
		Description: &desc,
		ObjectType:  &oType,
		ObjectID:    &body.ID,
	})

	if err != nil {
		log.Error(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}

	util.WriteJSON(w, http.StatusCreated, body)
}