mem-db.js 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  1. "use strict";
  2. var defer = require('../lib/defer.js');
  3. var codec = require('../lib/object-codec.js');
  4. var sha1 = require('git-sha1');
  5. module.exports = mixin;
  6. var isHash = /^[0-9a-f]{40}$/;
  7. function mixin(repo) {
  8. var objects = {};
  9. var refs = {};
  10. repo.saveAs = saveAs;
  11. repo.loadAs = loadAs;
  12. repo.saveRaw = saveRaw;
  13. repo.loadRaw = loadRaw;
  14. repo.hasHash = hasHash;
  15. repo.readRef = readRef;
  16. repo.updateRef = updateRef;
  17. repo.listRefs = listRefs;
  18. function readRef(ref, callback) {
  19. return makeAsync(function () {
  20. return refs[ref];
  21. }, callback);
  22. }
  23. function listRefs(prefix, callback) {
  24. return makeAsync(function () {
  25. var regex = prefix && new RegExp("^" + prefix + "[/$]");
  26. var out = {};
  27. Object.keys(refs).forEach(function (name) {
  28. if (regex && !regex.test(name)) return;
  29. out[name] = refs[name];
  30. });
  31. return out;
  32. }, callback);
  33. }
  34. function updateRef(ref, hash, callback) {
  35. return makeAsync(function () {
  36. return (refs[ref] = hash);
  37. }, callback);
  38. }
  39. function hasHash(hash, callback) {
  40. return makeAsync(function () {
  41. if (!isHash.test(hash)) hash = refs[hash];
  42. return hash in objects;
  43. }, callback);
  44. }
  45. function saveAs(type, body, callback) {
  46. return makeAsync(function () {
  47. var buffer = codec.frame({type:type,body:body});
  48. var hash = sha1(buffer);
  49. objects[hash] = buffer;
  50. return hash;
  51. }, callback);
  52. }
  53. function saveRaw(hash, buffer, callback) {
  54. return makeAsync(function () {
  55. objects[hash] = buffer;
  56. }, callback);
  57. }
  58. function loadAs(type, hash, callback) {
  59. return makeAsync(function () {
  60. if (!isHash.test(hash)) hash = refs[hash];
  61. var buffer = objects[hash];
  62. if (!buffer) return [];
  63. var obj = codec.deframe(buffer, true);
  64. if (obj.type !== type) throw new TypeError("Type mismatch");
  65. return obj.body;
  66. }, callback);
  67. }
  68. function loadRaw(hash, callback) {
  69. return makeAsync(function () {
  70. return objects[hash];
  71. }, callback);
  72. }
  73. }
  74. function makeAsync(fn, callback) {
  75. if (!callback) return makeAsync.bind(null, fn);
  76. defer(function () {
  77. var out;
  78. try { out = fn(); }
  79. catch (err) { return callback(err); }
  80. callback(null, out);
  81. });
  82. }