GitHub
9 changed files with 220 additions and 9 deletions
@ -0,0 +1,91 @@
@@ -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 @@
@@ -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