Update the simple substitution cipher to split the functions.

dev
KKlochko 2 years ago
parent 178c85b0cb
commit 126a706292

@ -1,6 +1,5 @@
(ns cipher-analytical-machine.ciphers.simple-substitution (ns cipher-analytical-machine.ciphers.simple-substitution
(:require [clojure.string :as cs] (:require [clojure.string :as cs]
[clojure.set :as set]
[cipher-analytical-machine.ciphers.caesar :as caesar]) [cipher-analytical-machine.ciphers.caesar :as caesar])
(:gen-class)) (:gen-class))
@ -20,28 +19,38 @@
(defn find-value-in-table (defn find-value-in-table
"It uses the substitution table to find the value of a char or a number." "It uses the substitution table to find the value of a char or a number."
[char substitution-table symbols] [char substitution-table]
(get substitution-table char)) (get substitution-table char))
(defn encrypt-message (defn encrypt-message
"Encrypt a message using the simple substitution cipher. The function is case-insensitive. If a symbol isn't in the symbols list, then it will be removed." "Encrypt a message using the simple substitution cipher. The function is case-insensitive. If a symbol isn't in the symbols list, then it will be removed."
[message key substitution-table symbols] [message substitution-table]
(let [max-index (count symbols)]
(->> message (->> message
(cs/lower-case) (map (fn [char] (find-value-in-table char substitution-table)))
(remove nil?)
(cs/join \,)))
(defn encrypt-message-with-caesar
"Encrypt a message using the simple substitution cipher and the Caesar cipher. The function is case-insensitive. If a symbol isn't in the symbols list, then it will be removed."
[message key substitution-table symbols]
(-> message
(caesar/encrypt-message key symbols) (caesar/encrypt-message key symbols)
(map (fn [char] (find-value-in-table char substitution-table symbols))) (encrypt-message substitution-table)))
(cs/join \,))))
(defn decrypt-message (defn decrypt-message
"Decrypt a message using the simple substitution cipher. The function is case-insensitive." "Decrypt a message using the simple substitution cipher. The function is case-insensitive. The substitution-table must be "
[message key substitution-table symbols] [message substitution-table]
(let [substitution-table (set/map-invert substitution-table) (let [message (cs/split message #",")]
max-index (count symbols)
message (cs/split message #",")]
(->> message (->> message
(map #(Integer/parseInt %)) (map #(Integer/parseInt %))
(map (fn [char] (find-value-in-table char substitution-table symbols))) (map (fn [char] (find-value-in-table char substitution-table)))
(caesar/encrypt-message key symbols) (remove nil?)
(cs/join)))) (cs/join))))
(defn decrypt-message-with-caesar
"Decrypt a message using the simple substitution cipher and the Caesar cipher. The function is case-insensitive."
[message key substitution-table symbols]
(-> message
(decrypt-message substitution-table)
(caesar/decrypt-message key symbols)))

@ -1,4 +1,4 @@
(ns cipher-analytical-machine.ciphers.simple-substitution (ns cipher-analytical-machine.ciphers.simple-substitution-test
(:require (:require
[clojure.test :refer :all] [clojure.test :refer :all]
[cipher-analytical-machine.ciphers.simple-substitution :refer :all])) [cipher-analytical-machine.ciphers.simple-substitution :refer :all]))
@ -22,22 +22,39 @@
1 \a 1 \a
5 nil)))) 5 nil))))
(deftest encrypt-message-text (deftest encrypt-message-test
(let [symbols "abc"
table {\a 1 \b 2 \c 3}]
(testing "The function must encrypt the message and remove unknown symbols."
(are [message expected]
(= expected (encrypt-message message table))
"abc" "1,2,3"
"aDbdc" "1,2,3"))))
(deftest encrypt-message-with-caesar-test
(let [symbols "abc" (let [symbols "abc"
table {\a 1 \b 2 \c 3}] table {\a 1 \b 2 \c 3}]
(testing "The function must encrypt the message and remove unknown symbols." (testing "The function must encrypt the message and remove unknown symbols."
(are [message key expected] (are [message key expected]
(= expected (encrypt-message message key table symbols)) (= expected (encrypt-message-with-caesar message key table symbols))
"abc" 0 "1,2,3" "abc" 0 "1,2,3"
"abc" 1 "2,3,1" "abc" 1 "2,3,1"
"aDbdc" 0 "1,2,3")))) "aDbdc" 0 "1,2,3"))))
(deftest decrypt-message-text (deftest decrypt-message-test
(let [table {1 \a 2 \b 3 \c}]
(testing "The function must decrypt the message and remove unknown numbers."
(are [message expected]
(= expected (decrypt-message message table))
"1,2,3" "abc"
"1,12,2,3" "abc"))))
(deftest decrypt-message-with-caesar-test
(let [symbols "abc" (let [symbols "abc"
table {\a 1 \b 2 \c 3}] table {1 \a 2 \b 3 \c}]
(testing "The function must decrypt the message and remove unknown numbers." (testing "The function must decrypt the message and remove unknown numbers."
(are [message key expected] (are [message key expected]
(= expected (decrypt-message message key table symbols)) (= expected (decrypt-message-with-caesar message key table symbols))
"1,2,3" 0 "abc" "1,2,3" 0 "abc"
"2,3,1" 1 "abc" "2,3,1" 1 "abc"
"1,12,2,3" 0 "abc")))) "1,12,2,3" 0 "abc"))))

Loading…
Cancel
Save