More robust config

Author: pomber

packages/mdx/src/plugin.ts

@@ -16,7 +16,9 @@ type CodeHikeConfig = {
   lineNumbers?: boolean
 }
 
-export function remarkCodeHike(config: CodeHikeConfig) {
+export function remarkCodeHike(
+  unsafeConfig: CodeHikeConfig
+) {
   return async (tree: Node) => {
     // TODO add opt-in config
     let hasCodeHikeImport = false
@@ -30,6 +32,8 @@ export function remarkCodeHike(config: CodeHikeConfig) {
       }
     })
 
+    const config = addConfigDefaults(unsafeConfig)
+
     addConfig(tree as Parent, config)
 
     if (!hasCodeHikeImport) {
@@ -52,6 +56,12 @@ export function remarkCodeHike(config: CodeHikeConfig) {
   }
 }
 
+function addConfigDefaults(
+  config: Partial<CodeHikeConfig> | undefined
+): CodeHikeConfig {
+  return { ...config, theme: config?.theme || {} }
+}
+
 function addConfig(tree: Parent, config: CodeHikeConfig) {
   tree.children.unshift({
     type: "mdxjsEsm",