commit 000401

templates/index.html

@@ -87,35 +87,44 @@
           });
         };
 
-      const language_keywords = {
-  python: ['def ', 'print(', 'import ', 'class '],
-  javascript: ['function ', 'console.log(', 'let ', 'const ', 'document.getElementById'],
-  typescript: ['interface ', 'type ', 'let ', 'const ', ': string', ': number'],
+    const language_keywords = {
+  python: ['def ', 'print(', 'import ', 'class ', 'self', 'elif', 'None'],
+  javascript: ['function ', 'console.log(', 'let ', 'const ', 'document.getElementById', '=>'],
+  typescript: ['interface ', 'type ', ': string', ': number', 'let ', 'const '],
   java: ['import java.', 'ArrayList<', 'System.out', 'void main(', 'public class', 'new '],
-  c: ['#include <stdio.h>', 'printf(', 'scanf(', 'int main('],
-  cpp: ['#include', 'std::', 'cout <<', 'cin >>'],
+  c: ['#include <stdio.h>', 'printf(', 'scanf(', 'int main(', 'return 0;'],
+  cpp: ['#include', 'std::', 'cout <<', 'cin >>', 'using namespace std'],
   bash: ['#!/bin/bash', 'echo ', 'cd ', 'ls', 'pwd'],
-  shell: ['#!/bin/sh', 'echo ', 'export ', 'fi'],
+  shell: ['#!/bin/sh', 'echo ', 'export ', 'fi', 'if ['],
   sql: ['SELECT ', 'INSERT ', 'UPDATE ', 'FROM ', 'WHERE ', 'JOIN ', 'DELETE '],
-  html: ['<!DOCTYPE html>', '<html>', '<div>', '<script>'],
-  css: ['color:', 'font-size:', 'margin:', 'padding:'],
-  go: ['package main', 'fmt.Println', 'func main()'],
-  php: ['<?php', 'echo ', '$_', '->'],
-  ruby: ['def ', 'puts ', 'end', 'class '],
-  kotlin: ['fun main(', 'val ', 'var ', 'println('],
-  swift: ['import SwiftUI', 'struct ', 'var body:', 'Text('],
-  rust: ['fn main()', 'println!', 'let mut'],
-  scala: ['object ', 'def ', 'val ', 'println('],
-  dart: ['void main()', 'print(', 'var ', 'class '],
+  html: ['<!DOCTYPE html>', '<html>', '<div>', '<script>', '<head>', '<body>'],
+  css: ['color:', 'font-size:', 'margin:', 'padding:', 'display:', 'position:'],
+  go: ['package main', 'fmt.Println(', 'func main()', 'import "fmt"'],
+  php: ['<?php', 'echo ', '$_', '->', '$_', 'function '],
+  ruby: ['def ', 'puts ', 'end', 'class ', ':symbol', 'require '],
+  kotlin: ['fun main(', 'val ', 'var ', 'println(', 'object '],
+  swift: ['import SwiftUI', 'struct ', 'var body:', 'Text(', 'let '],
+  rust: ['fn main()', 'println!', 'let mut', '::std', 'use '],
+  scala: ['object ', 'def ', 'val ', 'println(', 'case class'],
+  dart: ['void main()', 'print(', 'var ', 'class ', 'import '],
 };
 
 const detectLanguageByKeywords = (code) => {
+  let maxMatches = 0;
+  let bestMatch = 'plaintext';
+
   for (const [lang, keywords] of Object.entries(language_keywords)) {
-    if (keywords.some(kw => code.includes(kw))) {
-      return lang;
+    const matches = keywords.reduce(
+      (count, keyword) => count + (code.includes(keyword) ? 1 : 0),
+      0
+    );
+    if (matches > maxMatches) {
+      maxMatches = matches;
+      bestMatch = lang;
     }
   }
-  return '';
+
+  return bestMatch;
 };
 
 const renderCode = (text) => {