js 文件系统API操作示例

最近有个需求是：自动抓取某网站登录页面的验证码图片并保存，抓取n次。使用chrome插件来实现，其中使用到了js操作文件系统的api，特将代码记录下来，以备查阅。

PS:第一次使用js文件系统的api，百度了很久很多文章，大部分文章都很详细的解释介绍了api具体某个对象、方法的使用，但是很少有全部代码从零到功能实现的全部代码贴出，导致我这个新手一头雾水，甚至在文件保存完后，去哪里看保存的文件这个问题上都折腾了一两个小时。正是这些所谓的基础的东西反而因为没有人介绍导致很多无用功。此处贴出全部的代码，希望少走弯路。

以下是chrome插件的content.js的全部代码：

$(function(){
    if(window.location.href.indexOf('www.***.com/logout.action')!=-1 || window.location.href.indexOf('www.***.com/logon.action')!=-1){//登录页，示例隐藏了真实地址，避免不必要的纠纷
        saveImg();
    }
});
/**
* 保存图片的方法
*/
function saveImg(){
    var count = 0;
    var max = 100;//修改max的值定义需要抓图的张数
    var intev = setInterval(function(){
            window.requestFileSystem = window.requestFileSystem || window.webkitRequestFileSystem;
            window.requestFileSystem(window.TEMPORARY, 5*1024*1024, initFS, errorHandler);//申请文件系统，将initFS方法作为文件处理的方法，方法回调时会见文件系统对象fs作为参数传入
            count ++;
            if(count > max){//抓够张数后停止
                clearInterval(intev);
            }
        },5000);//每5秒刷新一张图
}
/**
* 点击验证码，刷新验证码并返回验证码的值，该网站使用的本地js生成的验证码，验证逻辑也是在本地js中完成，插件为gVerify.js
*/
function refreshImg(){
    var img = $('#verifyCanvas');
    $(img).click();
        //因为chrome插件和原网页的js运行环境不能相互访问（但插件js可以访问员网页的dom），使用向原网页插入script并在script中执行“将原网页js环境的变量绑定到页面dom节点后再访问”的办法获取值。
    var script = document.createElement('script');
        script.type = 'text/javascript';
        script.innerHTML = "document.body.setAttribute('code', verifyCode.options.code);";
        document.head.appendChild(script);
        document.head.removeChild(script);
        var code = document.body.getAttribute('code');
        console.log(code);
        return code;
}
/**
*文件处理方法，注意文件系统不同于操作系统的本地文件系统，只是浏览器运行环境中的一个虚拟的沙箱中的文件系统。在本地文件系统中无法搜索、查看到文件，查看文件的办法：浏览器输入地址查看（中间是网站的域名，每个网站的文件系统相互独立）filesystem:https://www.***.com/temporary/
*/
function initFS(fs){
    createDir(fs.root, '/img/'.split('/'));//创建目录
    var img_code = refreshImg();
        //写文件
    fs.root.getFile('/img/'+img_code + '.png', {create: true, exclusive: true}, function(fileEntry) {
        fileEntry.createWriter(function(fileWriter) {
            fileWriter.onwriteend = function(e) {
              console.log("Write successfully");
            };
            fileWriter.onerror = function(e) {
              console.log("Write error!");
            };
            var img_base64_url = $('#verifyCanvas')[0].toDataURL('image/png');//获取图谱的base64地址
            var content = base64ToBinary(img_base64_url);
            var bob = new Blob([content]);
            fileWriter.write(bob);//查看文件地址：浏览器输入filesystem:https://www.yqt365.com/temporary/
        });
    }, errorHandler);
}
/**
*异常处理
*/
function errorHandler(err){
 var msg = 'An error occured: ';

  switch (err.code) {
    case FileError.NOT_FOUND_ERR:
      msg += 'File or directory not found';
      break;

    case FileError.NOT_READABLE_ERR:
      msg += 'File or directory not readable';
      break;

    case FileError.PATH_EXISTS_ERR:
      msg += 'File or directory already exists';
      break;

    case FileError.TYPE_MISMATCH_ERR:
      msg += 'Invalid filetype';
      break;

    default:
      msg += 'Unknown Error';
      break;
  };

 console.log(msg);
}; 
/**
*创建目录
*/
function createDir(rootDir, folders) {
  rootDir.getDirectory(folders[0], {create: true}, function(dirEntry) {
    if (folders.length) {
      createDir(dirEntry, folders.slice(1));
    }
  }, errorHandler);
};
/**
* 看方法名称都懂啦
*/
function base64ToBinary(imgUrl) {
  var BASE64_MARKER = ';base64,';
  var base64Index = imgUrl.indexOf(BASE64_MARKER) + BASE64_MARKER.length;
  var base64 = imgUrl.substring(base64Index);
  var raw = window.atob(base64);
  var rawLength = raw.length;
  var array = new Uint8Array(new ArrayBuffer(rawLength));

  for (i = 0; i < rawLength; ++i) {
    array[i] = raw.charCodeAt(i);
  }
  return array;
}