Semaphore/api/projects/projects.go

package projects

import (
	"net/http"

	"time"

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

// GetProjects returns all projects in this users context
func GetProjects(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		user := context.Get(r, "user").(*db.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 []db.Project
		if _, err := db.Mysql.Select(&projects, query, args...); err != nil {
			panic(err)
		}

		mulekick.WriteJSON(w, http.StatusOK, projects)

		if (next != nil) {
      next.ServeHTTP(w, r)
    }
	})
}

// AddProject adds a new project to the database
func AddProject(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		var body db.Project
		user := context.Get(r, "user").(*db.User)

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

		err := body.CreateProject()
		if err != nil {
			panic(err)
		}

		if _, err := db.Mysql.Exec("insert into project__user set project_id=?, user_id=?, `admin`=1", body.ID, user.ID); err != nil {
			panic(err)
		}

		desc := "Project Created"
		oType := "Project"
		if err := (db.Event{
			ProjectID:   &body.ID,
			Description: &desc,
			ObjectType:  &oType,
			ObjectID:    &body.ID,
			Created:     db.GetParsedTime(time.Now()),
		}.Insert()); err != nil {
			panic(err)
		}

		mulekick.WriteJSON(w, http.StatusCreated, body)

		if (next != nil) {
      next.ServeHTTP(w, r)
    }
	})
}