framevpm-master/src/main/java/framevpm/learning/model/classmodel/VulNotVul.java
package framevpm.learning.model.classmodel;

import framevpm.organize.model.FileType;

import java.util.ArrayList;
import java.util.List;

public class VulNotVul implements ClassModel {
    @Override
    public List<String> getClassList() {
        List<String> classList = new ArrayList<>();
        classList.add("Vulnerable");
        classList.add("Not Vulnerable");
        return classList;
    }

    @Override
    public String correspondingToTypeFile(FileType fileType) {
        switch (fileType) {
            case Buggy:
            case Clear:
            case BuggyHistory:
            case VulnerableHistory:
                return "Not Vulnerable";
            case Vulnerability:
                return "Vulnerable";
        }
        return null;
    }

    @Override
    public String getName() {
        return NAME;
    }

    public final static String NAME = "Vulnotvul";
}
