Fix template edit link (#130)

* Fix template edit link

* Fix edit link for generated API pages

Author: alexlafroscia

addon/components/docs-viewer/x-main/component.js

@@ -3,7 +3,7 @@ import Component from '@ember/component';
 import layout from './template';
 import { computed } from '@ember/object';
 import appFiles from 'ember-cli-addon-docs/app-files';
-import { dasherize } from '@ember/string';
+import addonFiles from 'ember-cli-addon-docs/addon-files';
 import config from 'dummy/config/environment';
 import { getOwner } from '@ember/application';
 
@@ -23,20 +23,23 @@ export default Component.extend({
   editCurrentPageUrl: computed('router.currentRouteName', function() {
     let path = this.get('router.currentRouteName');
     if (!path) {
-      // `routing` doesn't exist for old ember versions via ember-try
+      // `router` doesn't exist for old ember versions via ember-try
       return;
     }
+
     path = path.replace(/\./g, '/');
 
-    if (path === 'docs/api/class') {
-      let params = getOwner(this).lookup('route:application').paramsFor('docs.api.class');
-      let klass = dasherize(params.class_id.replace(/-.+$/g, ''));
-      let path = `pods/${path}`;
+    if (path === 'docs/api/item') {
+      let { path } = getOwner(this).lookup('route:application').paramsFor('docs.api.item');
+      let file = addonFiles.find(f => f.match(path));
 
-      return `${packageJson.repository}/edit/master/addon/components/${klass}/component.js`;
+      if (file) {
+        return `${packageJson.repository}/edit/master/addon/${file}`;
+      }
     } else {
-      let templatePath = `pods/${path}`;
-      let file = appFiles.find(file => file.match(`${templatePath}/template.(hbs|md)`));
+      let file = appFiles
+        .filter(file => file.match(/template.(hbs|md)/))
+        .find(file => file.match(path));
 
       return `${packageJson.repository}/edit/master/tests/dummy/app/${file}`;
     }

index.js

@@ -147,8 +147,9 @@ module.exports = {
 
   treeForAddon(tree) {
     let dummyAppFiles = new FindDummyAppFiles([ 'tests/dummy/app' ]);
+    let addonFiles = new FindAddonFiles([ 'addon' ]);
 
-    return this._super(new MergeTrees([ tree, dummyAppFiles ]));
+    return this._super(new MergeTrees([ tree, dummyAppFiles, addonFiles ]));
   },
 
   treeForVendor(vendor) {
@@ -280,6 +281,16 @@ class FindDummyAppFiles extends Plugin {
   }
 }
 
+class FindAddonFiles extends Plugin {
+  build() {
+    let addonPath = this.inputPaths[0];
+    let paths = walkSync(addonPath, { directories: false })
+    let pathsString = JSON.stringify(paths);
+
+    fs.writeFileSync(path.join(this.outputPath, 'addon-files.js'), `export default ${pathsString};`);
+  }
+}
+
 class AutoExportAddonToApp extends Plugin {
   build() {
     let addonPath = this.inputPaths[0];