在使用TestNG框架的情况下，报表门户如何截取失败测试用例的截图？

import org.testng.ITestContext;
import org.testng.ITestListener;
import org.testng.ITestResult;
import org.testng.Reporter;

public class TestFailureListener implements ITestListener {

    @Override
    public void onTestFailure(ITestResult result) {
        // 获取当前测试用例的相关信息
        String className = result.getTestClass().getName();
        String methodName = result.getMethod().getMethodName();

        // 获取WebDriver实例，并进行截图
        WebDriver driver = YourWebDriverFactory.getDriver();
        File screenshotFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);

        // 保存截图文件
        String screenshotPath = "path/to/save/screenshot/" + className + "_" + methodName + ".png";
        try {
            FileUtils.copyFile(screenshotFile, new File(screenshotPath));
        } catch (IOException e) {
            e.printStackTrace();
        }

        // 在测试报告中展示截图
        Reporter.log("Failure screenshot: <img src=\"" + screenshotPath + "\" width=\"800\" height=\"600\" />");
    }

    // 其他方法的实现省略...
}

@Listeners(TestFailureListener.class)
public class YourTestNGTestClass {
    // 测试方法的实现...
}