mirror of
https://github.com/gchq/CyberChef
synced 2024-11-15 09:07:06 +00:00
move Set Intersection into its own operation class
This commit is contained in:
parent
5f93c667a2
commit
03ecaa81f7
4 changed files with 116 additions and 11 deletions
|
@ -220,7 +220,7 @@
|
||||||
{
|
{
|
||||||
"name": "Sample delimiter",
|
"name": "Sample delimiter",
|
||||||
"type": "binaryString",
|
"type": "binaryString",
|
||||||
"value": "\n\n"
|
"value": "\\n\\n"
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"name": "Item delimiter",
|
"name": "Item delimiter",
|
||||||
|
@ -256,7 +256,7 @@
|
||||||
"flowControl": false,
|
"flowControl": false,
|
||||||
"args": [
|
"args": [
|
||||||
{
|
{
|
||||||
"name": "Samples delimiter",
|
"name": "Sample delimiter",
|
||||||
"type": "binaryString",
|
"type": "binaryString",
|
||||||
"value": "\\n\\n"
|
"value": "\\n\\n"
|
||||||
},
|
},
|
||||||
|
|
|
@ -1,19 +1,63 @@
|
||||||
import SetOp from "./SetOps";
|
import Utils from "../Utils";
|
||||||
|
import Operation from "../Operation";
|
||||||
|
|
||||||
/**
|
/**
|
||||||
*
|
* Set Intersection operation
|
||||||
*/
|
*/
|
||||||
class SetIntersection extends SetOp {
|
class SetIntersection extends Operation {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
*
|
* Set Intersection constructor
|
||||||
*/
|
*/
|
||||||
constructor() {
|
constructor() {
|
||||||
super();
|
super();
|
||||||
this.setOp = this.runIntersection;
|
|
||||||
|
|
||||||
this.name = "Set Intersection";
|
this.name = "Set Intersection";
|
||||||
|
this.module = "Default";
|
||||||
this.description = "Get the intersection of two sets";
|
this.description = "Get the intersection of two sets";
|
||||||
|
this.inputType = "string";
|
||||||
|
this.outputType = "string";
|
||||||
|
this.args = [
|
||||||
|
{
|
||||||
|
name: "Sample delimiter",
|
||||||
|
type: "binaryString",
|
||||||
|
value: Utils.escapeHtml("\\n\\n")
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Item delimiter",
|
||||||
|
type: "binaryString",
|
||||||
|
value: ","
|
||||||
|
},
|
||||||
|
];
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Validate input length
|
||||||
|
* @param {Object[]} sets
|
||||||
|
* @throws {Error} if not two sets
|
||||||
|
*/
|
||||||
|
validateSampleNumbers(sets) {
|
||||||
|
if (!sets || (sets.length !== 2)) {
|
||||||
|
throw "Incorrect number of sets, perhaps you need to modify the sample delimiter or add more samples?";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Run the intersection operation
|
||||||
|
* @param input
|
||||||
|
* @param args
|
||||||
|
*/
|
||||||
|
run(input, args) {
|
||||||
|
[this.sampleDelim, this.itemDelimiter] = args;
|
||||||
|
const sets = input.split(this.sampleDelim);
|
||||||
|
|
||||||
|
try {
|
||||||
|
this.validateSampleNumbers(sets);
|
||||||
|
} catch (e) {
|
||||||
|
return e;
|
||||||
|
}
|
||||||
|
|
||||||
|
return Utils.escapeHtml(this.runIntersect(...sets.map(s => s.split(this.itemDelimiter))));
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -23,11 +67,14 @@ class SetIntersection extends SetOp {
|
||||||
* @param {Object[]} b
|
* @param {Object[]} b
|
||||||
* @returns {Object[]}
|
* @returns {Object[]}
|
||||||
*/
|
*/
|
||||||
runIntersection(a, b) {
|
runIntersect(a, b) {
|
||||||
return a.filter((item) => {
|
return a
|
||||||
return b.indexOf(item) > -1;
|
.filter((item) => {
|
||||||
}).join(this.itemDelimiter);
|
return b.indexOf(item) > -1;
|
||||||
|
})
|
||||||
|
.join(this.itemDelimiter);
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export default SetIntersection;
|
export default SetIntersection;
|
||||||
|
|
|
@ -48,6 +48,8 @@ import "./tests/operations/Base64";
|
||||||
// import "./tests/operations/StrUtils.js";
|
// import "./tests/operations/StrUtils.js";
|
||||||
// import "./tests/operations/SeqUtils.js";
|
// import "./tests/operations/SeqUtils.js";
|
||||||
import "./tests/operations/SetUnion";
|
import "./tests/operations/SetUnion";
|
||||||
|
import "./tests/operations/SetIntersection";
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
let allTestsPassing = true;
|
let allTestsPassing = true;
|
||||||
|
|
56
test/tests/operations/SetIntersection.mjs
Normal file
56
test/tests/operations/SetIntersection.mjs
Normal file
|
@ -0,0 +1,56 @@
|
||||||
|
/**
|
||||||
|
* Set Operations tests.
|
||||||
|
*
|
||||||
|
* @author d98762625
|
||||||
|
*
|
||||||
|
* @copyright Crown Copyright 2017
|
||||||
|
* @license Apache-2.0
|
||||||
|
*/
|
||||||
|
import TestRegister from "../../TestRegister";
|
||||||
|
|
||||||
|
TestRegister.addTests([
|
||||||
|
{
|
||||||
|
name: "Set Intersection",
|
||||||
|
input: "1 2 3 4 5\n\n3 4 5 6 7",
|
||||||
|
expectedOutput: "3 4 5",
|
||||||
|
recipeConfig: [
|
||||||
|
{
|
||||||
|
op: "Set Intersection",
|
||||||
|
args: ["\n\n", " "],
|
||||||
|
},
|
||||||
|
],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Set Intersection: only one set",
|
||||||
|
input: "1 2 3 4 5 6 7 8",
|
||||||
|
expectedOutput: "Incorrect number of sets, perhaps you need to modify the sample delimiter or add more samples?",
|
||||||
|
recipeConfig: [
|
||||||
|
{
|
||||||
|
op: "Set Intersection",
|
||||||
|
args: ["\n\n", " "],
|
||||||
|
},
|
||||||
|
],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Set Intersection: item delimiter",
|
||||||
|
input: "1-2-3-4-5\n\n3-4-5-6-7",
|
||||||
|
expectedOutput: "3-4-5",
|
||||||
|
recipeConfig: [
|
||||||
|
{
|
||||||
|
op: "Set Intersection",
|
||||||
|
args: ["\n\n", "-"],
|
||||||
|
},
|
||||||
|
],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Set Intersection: sample delimiter",
|
||||||
|
input: "1-2-3-4-5z3-4-5-6-7",
|
||||||
|
expectedOutput: "3-4-5",
|
||||||
|
recipeConfig: [
|
||||||
|
{
|
||||||
|
op: "Set Intersection",
|
||||||
|
args: ["z", "-"],
|
||||||
|
},
|
||||||
|
],
|
||||||
|
}
|
||||||
|
]);
|
Loading…
Reference in a new issue