Пример #1
0
func CreateSheet(body *music.Sheet, sheet *models.Sheet) {
	sheet.Name = body.Name
	sheet.Privilege = body.Privilege
	sheet.UserID = body.UserID
	sheet.CreatedAt = time.Now().Unix()
	sheet.UpdatedAt = time.Now().Unix()
}
Пример #2
0
func (this *Sheet) Update(body *music.Sheet) datatype.Response {
	sheet := models.Sheet{}
	tx := this.DB.Begin()
	u := models.User{}

	if body.UserID == 0 {
		return badRequest("")
	}

	tx.Where("id = ?", strconv.Itoa(body.UserID)).First(&u)

	if u.ID == 0 {
		return forbidden("no such user")
	}

	tx.Where("user_id = ? AND name = ?", strconv.Itoa(u.ID), body.UpdateName).First(&sheet)
	if sheet.ID != 0 {
		return forbidden("no such sheet")
	}
	//One user cannot create sheet with same name
	tx.Where("user_id = ? AND name = ?", strconv.Itoa(u.ID), body.Name).First(&sheet)

	if sheet.ID == 0 {
		return forbidden("not this user's sheet")
	}
	if body.UpdateName != "" {
		sheet.Name = body.UpdateName
	}
	if body.Privilege != "" {
		sheet.Privilege = body.Privilege
	}
	tx.Save(&sheet)
	tx.Commit()
	return ok("updated", sheet)
}
Пример #3
0
func (this *Audio) Update(body *music.Audio) datatype.Response {
	audio := models.Audio{}
	sheet := models.Sheet{}
	tx := this.DB.Begin()
	u := models.User{}

	if body.UserID == 0 {
		return badRequest("")
	}
	tx.Where("id = ?", strconv.Itoa(body.UserID)).First(&u)

	if u.ID == 0 {
		return forbidden("no such user")
	}

	tx.Where("audio_url = ?", body.AudioUrl).First(&audio)
	if audio.ID == 0 {
		return forbidden("no such audio")
	}

	tx.Where("id = ? AND user_id = ?", strconv.Itoa(audio.SheetID), strconv.Itoa(u.ID)).First(&sheet)
	if sheet.ID == 0 {
		return forbidden("not this user's audio")
	}
	if body.Name != "" {
		audio.Name = body.Name
	}
	if body.ImageUrl != "" {
		audio.ImageUrl = body.ImageUrl
	}

	//if body.Artist != ""{
	//	audio.Artist =  body.Artist
	//}

	if body.SheetID != 0 {
		sheet.ID = 0
		tx.Where("id = ? AND user_id = ?", strconv.Itoa(body.SheetID), strconv.Itoa(u.ID)).First(&sheet)
		if sheet.ID == 0 {
			return forbidden("the terminal sheet doesn't exist or belong to the request user!")

		}
		audio.SheetID = body.SheetID
	}
	tx.Save(&audio)
	tx.Commit()

	return ok("modify success", audio)
}
Пример #4
0
func (this *SheetMigration) Migration(body *music.SheetMigration) datatype.Response {
	sheet := models.Sheet{}
	tx := this.DB.Begin()
	for _, id := range body.IdList {
		audio := models.Audio{}
		tx.Where("id = ?", id).First(&audio)
		tx.Where("id = ?", strconv.Itoa(body.ToSheetId)).First(&sheet)
		if sheet.ID == 0 || sheet.UserID != body.UserID {
			return forbidden("terminal sheet doesn't exist or not belong to user!")

		}
		// if the second select does not find , it will stay in origin, so need to reset sheetid to 1
		sheet.ID = 0
		tx.Where("id = ?", strconv.Itoa(audio.SheetID)).First(&sheet)
		fmt.Println(sheet.ID)
		if sheet.ID == 0 || sheet.UserID != body.UserID {
			return forbidden("the audio not belong to request user!")
		}
		audio.SheetID = body.ToSheetId
		tx.Save(&audio)
	}
	tx.Commit()
	return ok("success", "")
}
Пример #5
0
func (this *Account) Register(body *user.Account) datatype.Response {
	var res datatype.Response
	//check mail
	reg := regexp.MustCompile(`^(\w)+(\.\w+)*@(\w)+((\.\w+)+)$`)
	sel := [...]bool{true, true, true}
	u := models.User{}
	sheet := models.Sheet{}
	salt := models.Salts{}
	flag := 0
	// begin transaction
	tx := this.DB.Begin()
	if body.Password == "" {
		goto BadRequest
	}

	if body.Name == "" || body.Mail == "" || body.Phone == "" {
		goto BadRequest
	}

	if reg.FindAllString(body.Mail, -1) == nil {
		goto BadRequest
	}
	if body.Name != "" {
		tx.Where("name=?", body.Name).First(&u)
		sel[0] = checkUser(u)
	}
	if body.Mail != "" {
		tx.Where("mail=?", body.Mail).First(&u)
		sel[1] = checkUser(u)
	}
	if body.Phone != "" {
		tx.Where("phone=?", body.Phone).First(&u)
		sel[2] = checkUser(u)
	}

	for i, v := range sel {
		if v {
			flag = i
			goto Forbidden
		}
	}

	CreateUser(&u, &salt, body)
	tx.Create(&u)
	//Create default sheet for user
	tx.Where("mail=?", body.Mail).First(&u)
	sheet.UserID = u.ID
	sheet.Name = "default#" + strconv.Itoa(u.ID)
	sheet.Privilege = "privacy"
	sheet.CreatedAt = time.Now().Unix()
	sheet.UpdatedAt = time.Now().Unix()
	tx.Create(&sheet)
	// transaction commit
	salt.UserID = u.ID
	tx.Create(&salt)

	tx.Commit()

	res = datatype.Response{
		Status: http.StatusOK,
		Body:   u,
	}
	return res

BadRequest:
	res = datatype.Response{
		Status: http.StatusBadRequest,
	}
	return res

Forbidden:
	var resText string
	switch flag {
	case 0:
		resText = "Name existed"
	case 1:
		resText = "Mail existed"
	case 2:
		resText = "Phone existed"
	}
	res = datatype.Response{
		Status:       http.StatusForbidden,
		ResponseText: resText,
	}
	return res
}