add filename handling
@@ -127,8 +127,13 @@ func testCleanup() error {
return nil
}
-func GfmrunActionFunc(_ *cli.Context) error {
- file, err := os.Open("README.md")
+func GfmrunActionFunc(c *cli.Context) error {
+ filename := c.Args().Get(0)
+ if filename == "" {
+ filename = "README.md"
+ }
+
+ file, err := os.Open(filename)
if err != nil {
return err
}
@@ -146,11 +151,16 @@ func GfmrunActionFunc(_ *cli.Context) error {
return err
}
- return runCmd("gfmrun", "-c", fmt.Sprint(counter), "-s", "README.md")
+ return runCmd("gfmrun", "-c", fmt.Sprint(counter), "-s", filename)
+}
+
+func TocActionFunc(c *cli.Context) error {
+ filename := c.Args().Get(0)
+ if filename == "" {
+ filename = "README.md"
}
-func TocActionFunc(_ *cli.Context) error {
- err := runCmd("node_modules/.bin/markdown-toc", "-i", "README.md")
+ err := runCmd("node_modules/.bin/markdown-toc", "-i", filename)
if err != nil {
return err
}
