test: add Layout.rustString()

This commit is contained in:
Michael Vines
2018-10-10 14:16:07 -07:00
parent a2d785398c
commit 715556a611
2 changed files with 38 additions and 39 deletions

View File

@ -16,3 +16,37 @@ export const publicKey = (property: string = 'publicKey'): Object => {
export const uint64 = (property: string = 'uint64'): Object => {
return BufferLayout.blob(8, property);
};
/**
* Layout for a Rust String type
*/
export const rustString = (property: string = 'string') => {
const rsl = BufferLayout.struct(
[
BufferLayout.u32('length'),
BufferLayout.u32('lengthPadding'),
BufferLayout.blob(
BufferLayout.offset(BufferLayout.u32(), -8),
'chars',
),
],
property,
);
const _decode = rsl.decode.bind(rsl);
const _encode = rsl.encode.bind(rsl);
rsl.decode = (buffer, offset) => {
const data = _decode(buffer, offset);
return data.chars.toString('utf8');
};
rsl.encode = (str, buffer, offset) => {
const data = {
chars: Buffer.from(str, 'utf8'),
};
return _encode(data, buffer, offset);
};
return rsl;
};