48 lines
1.3 KiB
JavaScript
48 lines
1.3 KiB
JavaScript
'use strict';
|
|
|
|
var Message = require('../message');
|
|
var inherits = require('util').inherits;
|
|
var bitcore = require('flocore-lib');
|
|
var $ = bitcore.util.preconditions;
|
|
var _ = bitcore.deps._;
|
|
|
|
/**
|
|
* @param {Transaction=} arg - An instance of Transaction
|
|
* @param {Object} options
|
|
* @extends Message
|
|
* @constructor
|
|
*/
|
|
function TransactionMessage(arg, options) {
|
|
Message.call(this, options);
|
|
this.command = 'tx';
|
|
this.Transaction = options.Transaction;
|
|
$.checkArgument(
|
|
_.isUndefined(arg) || arg instanceof this.Transaction,
|
|
'An instance of Transaction or undefined is expected'
|
|
);
|
|
this.transaction = arg;
|
|
if (!this.transaction) {
|
|
this.transaction = new this.Transaction();
|
|
}
|
|
}
|
|
inherits(TransactionMessage, Message);
|
|
|
|
TransactionMessage.prototype.setPayload = function(payload) {
|
|
if (typeof this.Transaction.fromRaw === 'function') {
|
|
this.transaction = this.Transaction.fromRaw(payload);
|
|
} else if (this.Transaction.prototype.fromBuffer) {
|
|
this.transaction = new this.Transaction().fromBuffer(payload);
|
|
} else {
|
|
this.transaction = this.Transaction.fromBuffer(payload);
|
|
}
|
|
};
|
|
|
|
TransactionMessage.prototype.getPayload = function() {
|
|
if (typeof this.Transaction.fromRaw === 'function') {
|
|
return this.transaction.toRaw();
|
|
}
|
|
return this.transaction.toBuffer();
|
|
};
|
|
|
|
module.exports = TransactionMessage;
|