[Golang] Remove C/C++ Style Comments

package rmcmt

import (
	"io/ioutil"
	"regexp"
)

func RemoveCStyleComments(content []byte) []byte {
	// http://blog.ostermiller.org/find-comment
	ccmt := regexp.MustCompile(`/\*([^*]|[\r\n]|(\*+([^*/]|[\r\n])))*\*+/`)
	return ccmt.ReplaceAll(content, []byte(""))
}

func RemoveCppStyleComments(content []byte) []byte {
	cppcmt := regexp.MustCompile(`//.*`)
	return cppcmt.ReplaceAll(content, []byte(""))
}

func RemoveCAndCppComments(srcpath, dstpath string) {
	b, err := ioutil.ReadFile(srcpath)
	if err != nil {
		panic(err)
	}
	b = RemoveCppStyleComments(RemoveCStyleComments(b))
	err = ioutil.WriteFile(dstpath, b, 0644)
	if err != nil {
		panic(err)
	}
}

package rmcmt

import "testing"

var ccode = []byte(`
/*
 * I am C style comments
 */
int main() {
  return 0;
}
`)

var cppcode = []byte(`
// I am Cpp style comments
int main() {
  return 0;
}
`)

var codeAfter = []byte(`

int main() {
  return 0;
}
`)

func TestRemoveCStyleComments(t *testing.T) {
	if string(RemoveCStyleComments(ccode)) != string(codeAfter) {
		t.Error("Remove C style comments failure!")
	}
}

func TestRemoveCppStyleComments(t *testing.T) {
	if string(RemoveCppStyleComments(cppcode)) != string(codeAfter) {
		t.Error("Remove Cpp style comments failure!")
	}
}