CloudSlang/cs-actions

/*
 * Copyright 2020-2024 Open Text
 * This program and the accompanying materials
 * are made available under the terms of the Apache License v2.0 which accompany this distribution.
 *
 * The Apache License is available at
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.cloudslang.content.filesystem.services;

import io.cloudslang.content.filesystem.entities.MD5SumInputs;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.HashMap;
import java.util.Map;

import static io.cloudslang.content.constants.OutputNames.RETURN_CODE;
import static io.cloudslang.content.constants.OutputNames.RETURN_RESULT;
import static io.cloudslang.content.filesystem.constants.Constants.*;
import static io.cloudslang.content.filesystem.constants.ExceptionMsgs.WORKS_ONLY_FOR_FILES;
import static io.cloudslang.content.filesystem.constants.ResultsName.CHECKSUM;

public class MD5SumService {

    public static Map<String, String> execute(MD5SumInputs md5SumInputs) throws NoSuchAlgorithmException, IOException {

        Map<String, String> result = new HashMap<>();
        File f = new File(md5SumInputs.getSource());
        if(f.isDirectory())
            throw new IllegalArgumentException(WORKS_ONLY_FOR_FILES);

        MessageDigest md5 = MessageDigest.getInstance(MD5);
        FileInputStream fStream = null;

        try {
            fStream = new FileInputStream(md5SumInputs.getSource());

            byte[] buf = new byte[1024];
            int nRead = fStream.read(buf);
            while (nRead > 0) {
                md5.update(buf, 0, nRead);
                nRead = fStream.read(buf);
            }
        } finally {
            if (fStream != null) {
                fStream.close();
            }
        }
        byte[] digest = md5.digest();
        StringBuffer hash = new StringBuffer();
        for (byte b : digest) {
            hash.append(Integer.toHexString((b & 0xFF) | 0x100).toLowerCase().substring(1, 3));
        }


        result.put(CHECKSUM, hash.toString());
        if (hash.toString().equals(md5SumInputs.getCompareTo())){
            result.put(RETURN_CODE, EQUALS_VALUE);
            result.put(RETURN_RESULT, String.format(EQUALS_VALUE_RETURN_RESULT,md5SumInputs.getCompareTo()));
        }
        else {
            result.put(RETURN_RESULT, String.format(NOT_EQUAL_VALUE_RETURN_RESULT,md5SumInputs.getCompareTo()));
            result.put(RETURN_CODE, LESS);
        }

        return result;

    }
}