Add subPackageName and get rid of hard-coded 0x.js in sourceLink

1 files changed, 11 insertions, 2 deletions

diff --git a/packages/website/ts/pages/documentation/source_link.tsx b/packages/website/ts/pages/documentation/source_link.tsx
index 408dcabc7..74fc6d4d5 100644
--- a/packages/website/ts/pages/documentation/source_link.tsx
+++ b/packages/website/ts/pages/documentation/source_link.tsx
@@ -1,3 +1,4 @@
+import * as _ from 'lodash';
 import {colors} from 'material-ui/styles';
 import * as React from 'react';
 import {Source} from 'ts/types';
@@ -7,14 +8,22 @@ interface SourceLinkProps {
     source: Source;
     baseUrl: string;
     version: string;
+    subPackageName: string;
 }
 
-const SUB_PKG = '0x.js';
+const packagesWithNamespace = [
+    'connect',
+];
 
 export function SourceLink(props: SourceLinkProps) {
     const src = props.source;
     const url = props.baseUrl;
-    const sourceCodeUrl = `${url}/blob/${SUB_PKG}%40${props.version}/packages/${SUB_PKG}/${src.fileName}#L${src.line}`;
+    const pkg = props.subPackageName;
+    let tagPrefix = pkg;
+    if (_.includes(packagesWithNamespace, pkg)) {
+        tagPrefix = `@0xproject/${pkg}`;
+    }
+    const sourceCodeUrl = `${url}/blob/${tagPrefix}%40${props.version}/packages/${pkg}/${src.fileName}#L${src.line}`;
     return (
         <div className="pt2" style={{fontSize: 14}}>
             <a