pixeldrain/gocqlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Adjust permisions of output file and directory (#303)

Browse Source 
- Bind output file and directory permisions to CLI.
- Use `os.WriteFile` instead of deprecated `ioutil.WriteFile`.
This commit is contained in:
Daniel Bershatsky
2025-04-20 17:15:00 +03:00
committed by GitHub
parent 741627dba7
commit fa2799b8f5
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
cmd/schemagen/schemagen.go
View File

@@ -7,7 +7,7 @@ import (
"fmt" "fmt"
"go/format" "go/format"
"html/template" "html/template"
"io/ioutil" "io/fs"
"log" "log"
"os" "os"
"path" "path"
@@ -34,6 +34,8 @@ var (
flagKeyspace = cmd.String("keyspace", "", "keyspace to inspect") flagKeyspace = cmd.String("keyspace", "", "keyspace to inspect")
flagPkgname = cmd.String("pkgname", "models", "the name you wish to assign to your generated package") flagPkgname = cmd.String("pkgname", "models", "the name you wish to assign to your generated package")
flagOutput = cmd.String("output", "models", "the name of the folder to output to") flagOutput = cmd.String("output", "models", "the name of the folder to output to")
flagOutputDirPerm = cmd.Uint64("output-dir-perm", 0o755, "output directory permissions")
flagOutputFilePerm = cmd.Uint64("output-file-perm", 0o644, "output file permissions")
flagUser = cmd.String("user", "", "user for password authentication") flagUser = cmd.String("user", "", "user for password authentication")
flagPassword = cmd.String("password", "", "password for password authentication") flagPassword = cmd.String("password", "", "password for password authentication")
flagIgnoreNames = cmd.String("ignore-names", "", "a comma-separated list of table, view or index names to ignore") flagIgnoreNames = cmd.String("ignore-names", "", "a comma-separated list of table, view or index names to ignore")
@@ -65,7 +67,7 @@ func main() {
} }
func schemagen() error { func schemagen() error {
if err := os.MkdirAll(*flagOutput, os.ModePerm); err != nil { if err := os.MkdirAll(*flagOutput, os.FileMode(*flagOutputDirPerm)); err != nil {
return fmt.Errorf("create output directory: %w", err) return fmt.Errorf("create output directory: %w", err)
} }
@@ -83,7 +85,7 @@ func schemagen() error {
} }
outputPath := path.Join(*flagOutput, *flagPkgname+".go") outputPath := path.Join(*flagOutput, *flagPkgname+".go")
return ioutil.WriteFile(outputPath, b, os.ModePerm) return os.WriteFile(outputPath, b, fs.FileMode(*flagOutputFilePerm))
} }
func renderTemplate(md *gocql.KeyspaceMetadata) ([]byte, error) { func renderTemplate(md *gocql.KeyspaceMetadata) ([]byte, error) {