122 lines
2.6 KiB
Go
122 lines
2.6 KiB
Go
package mysql
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"strings"
|
|
|
|
"github.com/hashicorp/terraform/helper/schema"
|
|
)
|
|
|
|
func resourceGrant() *schema.Resource {
|
|
return &schema.Resource{
|
|
Create: CreateGrant,
|
|
Update: nil,
|
|
Read: ReadGrant,
|
|
Delete: DeleteGrant,
|
|
|
|
Schema: map[string]*schema.Schema{
|
|
"user": &schema.Schema{
|
|
Type: schema.TypeString,
|
|
Required: true,
|
|
ForceNew: true,
|
|
},
|
|
|
|
"host": &schema.Schema{
|
|
Type: schema.TypeString,
|
|
Optional: true,
|
|
ForceNew: true,
|
|
Default: "localhost",
|
|
},
|
|
|
|
"database": &schema.Schema{
|
|
Type: schema.TypeString,
|
|
Required: true,
|
|
ForceNew: true,
|
|
},
|
|
|
|
"privileges": &schema.Schema{
|
|
Type: schema.TypeSet,
|
|
Required: true,
|
|
ForceNew: true,
|
|
Elem: &schema.Schema{Type: schema.TypeString},
|
|
Set: schema.HashString,
|
|
},
|
|
|
|
"grant": &schema.Schema{
|
|
Type: schema.TypeBool,
|
|
Optional: true,
|
|
ForceNew: true,
|
|
Default: false,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func CreateGrant(d *schema.ResourceData, meta interface{}) error {
|
|
conn := meta.(*providerConfiguration).Conn
|
|
|
|
// create a comma-delimited string of privileges
|
|
var privileges string
|
|
var privilegesList []string
|
|
vL := d.Get("privileges").(*schema.Set).List()
|
|
for _, v := range vL {
|
|
privilegesList = append(privilegesList, v.(string))
|
|
}
|
|
privileges = strings.Join(privilegesList, ",")
|
|
|
|
stmtSQL := fmt.Sprintf("GRANT %s on %s.* TO '%s'@'%s'",
|
|
privileges,
|
|
d.Get("database").(string),
|
|
d.Get("user").(string),
|
|
d.Get("host").(string))
|
|
|
|
if d.Get("grant").(bool) {
|
|
stmtSQL = " WITH GRANT OPTION"
|
|
}
|
|
|
|
log.Println("Executing statement:", stmtSQL)
|
|
_, _, err := conn.Query(stmtSQL)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
user := fmt.Sprintf("%s@%s:%s", d.Get("user").(string), d.Get("host").(string), d.Get("database"))
|
|
d.SetId(user)
|
|
|
|
return ReadGrant(d, meta)
|
|
}
|
|
|
|
func ReadGrant(d *schema.ResourceData, meta interface{}) error {
|
|
// At this time, all attributes are supplied by the user
|
|
return nil
|
|
}
|
|
|
|
func DeleteGrant(d *schema.ResourceData, meta interface{}) error {
|
|
conn := meta.(*providerConfiguration).Conn
|
|
|
|
stmtSQL := fmt.Sprintf("REVOKE GRANT OPTION ON %s.* FROM '%s'@'%s'",
|
|
d.Get("database").(string),
|
|
d.Get("user").(string),
|
|
d.Get("host").(string))
|
|
|
|
log.Println("Executing statement:", stmtSQL)
|
|
_, _, err := conn.Query(stmtSQL)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
stmtSQL = fmt.Sprintf("REVOKE ALL ON %s.* FROM '%s'@'%s'",
|
|
d.Get("database").(string),
|
|
d.Get("user").(string),
|
|
d.Get("host").(string))
|
|
|
|
log.Println("Executing statement:", stmtSQL)
|
|
_, _, err = conn.Query(stmtSQL)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|