GitHub
9 changed files with 220 additions and 9 deletions
@ -0,0 +1,91 @@ |
|||||||
|
package main |
||||||
|
|
||||||
|
import ( |
||||||
|
"reflect" |
||||||
|
|
||||||
|
"github.com/ugorji/go/codec" |
||||||
|
"samhofi.us/x/keybase/types/chat1" |
||||||
|
) |
||||||
|
|
||||||
|
// KvStorePutStruct marshals an interface to JSON and sends to kvstore
|
||||||
|
func (b *bot) KVStorePutStruct(convIDstr chat1.ConvIDStr, v interface{}) error { |
||||||
|
// marshal the struct to JSON
|
||||||
|
kvstoreDataString, err := encodeStructToJSONString(v) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
// put the string in kvstore
|
||||||
|
err = b.KVStorePut(string(convIDstr), getTypeName(v), kvstoreDataString) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
// KVStoreGetStruct gets a string from kvstore and unmarshals the JSON to a struct
|
||||||
|
func (b *bot) KVStoreGetStruct(convIDstr chat1.ConvIDStr, v interface{}) error { |
||||||
|
// get the string from kvstore
|
||||||
|
result, err := b.KVStoreGet(string(convIDstr), getTypeName(v)) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
// if there was no result just return and the struct is unmodified
|
||||||
|
if result == "" { |
||||||
|
return nil |
||||||
|
} |
||||||
|
// unmarshal the string into JSON
|
||||||
|
err = decodeJSONStringToStruct(v, result) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
// KVStorePut puts a string into kvstore given a key and namespace
|
||||||
|
func (b *bot) KVStorePut(namespace string, key string, value string) error { |
||||||
|
_, err := b.k.KVPut(&b.config.KVStoreTeam, namespace, key, value) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
// KVStoreGet gets a string from kvstore given a key and namespace
|
||||||
|
func (b *bot) KVStoreGet(namespace string, key string) (string, error) { |
||||||
|
kvResult, err := b.k.KVGet(&b.config.KVStoreTeam, namespace, key) |
||||||
|
if err != nil { |
||||||
|
return "", err |
||||||
|
} |
||||||
|
return kvResult.EntryValue, nil |
||||||
|
} |
||||||
|
|
||||||
|
// getTypeName returns the name of a type, regardless of if its a pointer or not
|
||||||
|
func getTypeName(v interface{}) string { |
||||||
|
t := reflect.TypeOf(v) |
||||||
|
if t.Kind() == reflect.Ptr { |
||||||
|
return t.Elem().Name() |
||||||
|
} |
||||||
|
return t.Name() |
||||||
|
} |
||||||
|
|
||||||
|
func encodeStructToJSONString(v interface{}) (string, error) { |
||||||
|
jh := codecHandle() |
||||||
|
var bytes []byte |
||||||
|
err := codec.NewEncoderBytes(&bytes, jh).Encode(v) |
||||||
|
if err != nil { |
||||||
|
return "", err |
||||||
|
} |
||||||
|
result := string(bytes) |
||||||
|
return result, nil |
||||||
|
} |
||||||
|
|
||||||
|
func decodeJSONStringToStruct(v interface{}, src string) error { |
||||||
|
bytes := []byte(src) |
||||||
|
jh := codecHandle() |
||||||
|
return codec.NewDecoderBytes(bytes, jh).Decode(v) |
||||||
|
} |
||||||
|
|
||||||
|
func codecHandle() *codec.JsonHandle { |
||||||
|
var jh codec.JsonHandle |
||||||
|
return &jh |
||||||
|
} |
||||||
@ -0,0 +1,8 @@ |
|||||||
|
package main |
||||||
|
|
||||||
|
// ConvOptions stores team specific options like custom servers
|
||||||
|
type ConvOptions struct { |
||||||
|
ConvID string `json:"converation_id,omitempty"` |
||||||
|
NotificationsEnabled bool `json:"notifications_enabled,omitempty"` |
||||||
|
CustomURL string `json:"custom_url,omitempty"` |
||||||
|
} |
||||||
Loading…
Reference in new issue