Better selection of sourcecode

pixijs · Aug 25, 2016 · 168c09c · 168c09c
1 parent ca2c1f0
commit 168c09c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/scripts/linenumber.js b/src/scripts/linenumber.js
@@ -1,21 +1,21 @@
 (function() {
  var counter = 0;
  var numbered;
- var source = document.getElementsByClassName('prettyprint source');
+ var source = document.getElementById('source-code');
 
- if (source && source[0]) {
+ if (source) {
  var linenums = config.linenums;
 
  if (linenums) {
- source = source[0].getElementsByTagName('ol')[0];
+ source = source.getElementsByTagName('ol')[0];
 
  numbered = Array.prototype.slice.apply(source.children);
  numbered = numbered.map(function(item) {
  counter++;
  item.id = 'line' + counter;
  });
  } else {
- source = source[0].getElementsByTagName('code')[0];
+ source = source.getElementsByTagName('code')[0];
 
  numbered = source.innerHTML.split('\n');
  numbered = numbered.map(function(item) {

diff --git a/tmpl/source.tmpl b/tmpl/source.tmpl
@@ -3,6 +3,6 @@
 ?>
  <section>
  <article>
- <pre class="prettyprint source <?js= env.conf.templates.linenums ? 'linenums' : '' ?>"><code><?js= data.code ?></code></pre>
+ <pre id="source-code" class="prettyprint source <?js= env.conf.templates.linenums ? 'linenums' : '' ?>"><code><?js= data.code ?></code></pre>
  </article>
  </section>