Add custom load support

Author: TrySound

index.js

@@ -1,17 +1,17 @@
-var fs = require("fs")
 var path = require("path")
 var assign = require("object-assign")
 var postcss = require("postcss")
 var joinMedia = require("./lib/join-media")
 var resolveId = require("./lib/resolve-id")
+var loadContent = require("./lib/load-content")
 var parseStatements = require("./lib/parse-statements")
 
 function AtImport(options) {
   options = assign({
     root: process.cwd(),
     path: [],
     skipDuplicates: true,
-    encoding: "utf8",
+    load: loadContent,
   }, options)
 
   options.root = path.resolve(options.root)
@@ -281,14 +281,8 @@ function loadImportContent(
     state.importedFiles[filename][media] = true
   }
 
-  return new Promise(function(resolve, reject) {
-    fs.readFile(filename, options.encoding, function(err, data) {
-      if (err) {
-        return reject(err)
-      }
-      resolve(data)
-    })
-  }).then(function(content) {
+  return Promise.resolve(options.load(filename, options))
+  .then(function(content) {
     if (typeof options.transform === "function") {
       content = options.transform(content, filename)
     }
@@ -333,8 +327,10 @@ function loadImportContent(
       state,
       media,
       processor
-    ).then(function(statements) {
-      return processor.process(newStyles).then(function(newResult) {
+    )
+    .then(function(statements) {
+      return processor.process(newStyles)
+      .then(function(newResult) {
         result.messages = result.messages.concat(newResult.messages)
 
         return statements

lib/load-content.js

@@ -0,0 +1,12 @@
+var fs = require("fs")
+
+module.exports = function(filename) {
+  return new Promise(function(resolve, reject) {
+    fs.readFile(filename, "utf-8", function(err, data) {
+      if (err) {
+        return reject(err)
+      }
+      resolve(data)
+    })
+  })
+}

test/custom-load.js

@@ -0,0 +1,18 @@
+import test from "ava"
+import compareFixtures from "./lib/compare-fixtures"
+
+test("should accept content", t => {
+  return compareFixtures(t, "custom-load", {
+    load: () => {
+      return "custom-content {}"
+    },
+  })
+})
+
+test("should accept promised content", t => {
+  return compareFixtures(t, "custom-load", {
+    load: () => {
+      return Promise.resolve("custom-content {}")
+    },
+  })
+})

test/fixtures/custom-load.css

@@ -0,0 +1 @@
+@import "foo"

test/fixtures/custom-load.expected.css

@@ -0,0 +1 @@
+custom-content {}