Fix #include snippet for c/c++

snippets/c.lua

@@ -1,10 +1,14 @@
-# C LuaSnip Snippets
+-- C LuaSnip Snippets
 
 local luasnip = require('luasnip')
-local snip = luasnip.snippet
+local snippet = luasnip.snippet
 local text = luasnip.text_node
 local insert = luasnip.insert_node
 local func = luasnip.function_node
+local choice = luasnip.choice_node
+local dynamic = luasnip.dynamic_node
+local restore = luasnip.restore_node
+local snip = luasnip.snippet_node
 local key = require("luasnip.nodes.key_indexer").new_key
 
 local function getIncludeGuardName()
@@ -13,38 +17,34 @@ local function getIncludeGuardName()
 end
 
 local snippets = {
-  snip('#include', {
-    text('#include '),
-    insert(1, '<', { key = 'open' }),
-    insert(2, 'header'),
-    func(function(args)
-      if args[1][1] == '<' then
-        return '>'
-      else
-        return '"'
-      end
-    end, key('open')),
-  }),
 
-  snip('once', {
+  snippet('#include', {
+    text('#include '),
+    choice(1, {
+      snip(nil, { text('<'), restore(1, 'header'), text('>') }),
+      snip(nil, { text('"'), restore(1, 'header'), text('"') }),
+    }),
+  }, { stored = { ['header'] = insert(1, 'header') } }),
+
+  snippet('once', {
     text('#ifndef '),
-    func(function(args)
-      return args[1][1]
-    end, key('name')),
+    func(function(opts)
+      return opts[1][1]
+    end, key('guard')),
     text({ '', '#define ' }),
-    insert(1, getIncludeGuardName(), { key = 'name' }),
-    -- FIXME: dynamic(1, function(_, _)
-    --   return snip({}, {
-    --     insert(1, getIncludeGuardName())
-    --   })
-    -- end, { key = "name" }),
+    dynamic(1, function()
+      return snip(nil, {
+        insert(1, getIncludeGuardName(), { key = 'guard' }),
+      })
+    end),
     text({ '', '', '', }),
     insert(0, ''),
     text({ '', '', '#endif  // ' }),
-    func(function(args)
-      return args[1][1]
-    end, key('name')),
-  }),
+    func(function(opts)
+      return opts[1][1]
+    end, key('guard')),
+  }, {}),
+
 }
 
 return snippets