♻️ specify target types in extension object

Signed-off-by: Nahuel Garbezza <n.garbezza@gmail.com>

Author: ngarbezza

src/extensions/collections.js

@@ -3,6 +3,7 @@
 import { applyExtension } from './extension_applier.js';
 
 const Collection = {
+  targets: [Array, String, Set],
   class: {
   },
   instance: {
@@ -49,6 +50,7 @@ const Collection = {
 };
 
 const SequenceableCollection = {
+  targets: [Array, String],
   class: {
   },
   instance: {
@@ -103,6 +105,7 @@ const SequenceableCollection = {
 };
 
 const HeterogeneousCollection = {
+  targets: [Array, Set],
   class: {
   },
   instance: {
@@ -128,6 +131,7 @@ const HeterogeneousCollection = {
 };
 
 const ArrayExtensions = {
+  targets: [Array],
   class: {
     with(...objects) {
       return objects;
@@ -162,6 +166,7 @@ const ArrayExtensions = {
 };
 
 const StringExtensions = {
+  targets: [String],
   class: {
   },
   instance: {
@@ -189,6 +194,7 @@ const StringExtensions = {
 };
 
 const SetExtensions = {
+  targets: [Set],
   class: {
     with(...objects) {
       return objects.asSet();
@@ -240,10 +246,10 @@ const SetExtensions = {
 };
 
 export const install = () => {
-  applyExtension(Collection, Array, String, Set);
-  applyExtension(SequenceableCollection, Array, String);
-  applyExtension(HeterogeneousCollection, Array, Set);
-  applyExtension(ArrayExtensions, Array);
-  applyExtension(StringExtensions, String);
-  applyExtension(SetExtensions, Set);
+  applyExtension(Collection);
+  applyExtension(SequenceableCollection);
+  applyExtension(HeterogeneousCollection);
+  applyExtension(ArrayExtensions);
+  applyExtension(StringExtensions);
+  applyExtension(SetExtensions);
 };

src/extensions/date.js

@@ -3,9 +3,10 @@
 import { applyExtension } from './extension_applier.js';
 
 const DateExtensions = {
+  targets: [Date],
   class: {
     today() {
-      return new Date();
+      return this.new();
     },
     tomorrow() {
       const result = this.today();
@@ -32,5 +33,5 @@ const DateExtensions = {
 };
 
 export const install = () => {
-  applyExtension(DateExtensions, Date);
+  applyExtension(DateExtensions);
 };

src/extensions/extension_applier.js

@@ -6,11 +6,11 @@ const eachExtensionOf = (extension, block) =>
 const extend = (proto, extension, methodName) =>
   Object.defineProperty(proto, methodName, { value: extension[methodName] });
 
-export const applyExtension = (extension, ...targetTypes) => {
+export const applyExtension = (extension) => {
   eachExtensionOf(extension.instance, methodName =>
-    targetTypes.forEach(type => extend(type.prototype, extension.instance, methodName))
+    extension.targets.forEach(type => extend(type.prototype, extension.instance, methodName))
   );
   eachExtensionOf(extension.class, methodName =>
-    targetTypes.forEach(type => extend(type, extension.class, methodName))
+    extension.targets.forEach(type => extend(type, extension.class, methodName))
   );
 };

src/extensions/function.js

@@ -3,6 +3,7 @@
 import { applyExtension } from './extension_applier.js';
 
 const FunctionExtension = {
+  targets: [Function],
   class: {
   },
   instance: {
@@ -17,5 +18,5 @@ const FunctionExtension = {
 };
 
 export const install = () => {
-  applyExtension(FunctionExtension, Function);
+  applyExtension(FunctionExtension);
 };

src/extensions/number.js

@@ -3,6 +3,7 @@
 import { applyExtension } from './extension_applier.js';
 
 const NumberExtensions = {
+  targets: [Number],
   instance: {
     isDivisibleBy(aNumber) {
       return this % aNumber === 0;
@@ -32,5 +33,5 @@ const NumberExtensions = {
 };
 
 export const install = () => {
-  applyExtension(NumberExtensions, Number);
+  applyExtension(NumberExtensions);
 };

src/extensions/object.js

@@ -3,6 +3,7 @@
 import { applyExtension } from './extension_applier.js';
 
 const ObjectExtension = {
+  targets: [Object],
   class: {
   },
   instance: {
@@ -25,5 +26,5 @@ const ObjectExtension = {
 };
 
 export const install = () => {
-  applyExtension(ObjectExtension, Object);
+  applyExtension(ObjectExtension);
 };