Let's handle file name extensions with Elixir, useful to validate the type of format of files that we would like to allow.
Step 1:
Let's create a new Elixir project by going to the command line and typing the following in your desired directory:
$ mix new file_validator && cd file_validator
Step 2:
Open your test file test/file_validator_test.exs
, remove the autogenerated test and add the following test:
test "file validation should be false" do
file = "somefile.exe"
assert FileValidator.valid?(file) == false
end
test "file validation should be true" do
file = "somefile.png"
assert FileValidator.valid?(file) == true
end
Go to the command line and run the test mix test
Step 3:
Let's make our test pass by opening our main project file lib/file_validator.ex
, remove the autogenerated hello world function and add the following:
@extension_whitelist ~w(.jpg .jpeg .gif .png .pdf)
def valid?(file) do
file_extension = get_ext(file)
@extension_whitelist
|> Enum.member?(file_extension)
end
defp get_ext(file) do
file
|> Path.extname()
|> String.downcase()
end
Get back to the command line and run the test mix test
now our test should pass with no failures.
Code Breakdown
@extension_whitelist ~w(.jpg .jpeg .gif .png .pdf)
The @
symbol means that is a module attribute, that can be used as a constant throughout our module.
The ~w sigil is used to generate lists of words(words are just regular strings). Inside the ~w sigil, words are separated by whitespace.
Path.extname(path)
Returns the extension of the last component of the path of a string.
Enum.member?(list, element)
Checks if an element exists within the list.
Conclusion
That's a pretty simple and straightforward way to validate a file format, by getting the extension from the string with Path.extname(path)
and passing it to Enum.member?(list, element)
along with a whitelisted list of allowed file extensions. Thank you so much for your time, I really appreciate it.
Top comments (0)