combine function and main snippet into one (c)

lua/snippets/c.lua

@@ -19,28 +19,43 @@ return {
   }),
 
   -- function snippet
-  s("fn", {
+  s({ trig = [[fn\|main]], trigEngine = "vim" }, {
-    c(1, {
+    d(1, function(_, snip)
-      t("void"),
+      if snip.trigger == "main" then
-      t("char"),
+        return sn(nil, {
-      t("int"),
+          t({ "int", "main(int argc, char *argv[])" })
-      t("short"),
+        })
-      t("long"),
+      else
-      t("double"),
+        return sn(nil, {
-      t("float"),
+          c(1, {
-      i(nil, "myType"),
+            t(""),
-    }),
+            t("static "),
-    t({ "", "" }),
+          }),
-    c(2, {
+          c(2, {
-      t(""),
+            t("void"),
-      t("*")
+            t("char"),
-    }),
+            t("int"),
-    i(3, "myFunc");
+            t("short"),
-    t("("),
+            t("long"),
-    i(4),
+            t("double"),
-    t(")"),
+            t("float"),
+            i(nil, "myType"),
+          }),
+          t({ "", "" }),
+          c(3, {
+            t(""),
+            t("*")
+          }),
+          i(4, "myFunc");
+          t("("),
+          i(5),
+          t(")"),
+        })
+      end
+    end, {}),
     t({ "", "{", "\t" }),
     i(0),
-    t({ "", "}" }),
+    t({ "", "}" })
+  }),
   })
 }