3 # Convert CSV parts list into vendor import format
5 # Copyright © 2015 Keith Packard <keithp@keithp.com>, GPL v3+
11 string[*] not_vendors;
12 string not_vendor_list;
19 void fatal(string format, poly args ...)
21 File::fprintf(stderr, format, args...);
25 string[*] read_line(file f) {
27 string line = fgets(f);
29 string[*] elts = String::parse_csv(line);
30 for (int i = 0; i < dim(elts); i++)
31 if (elts[i][0] == '"')
32 elts[i] = String::dequote(elts[i]);
38 string[*] required_elements = {
49 bool has_header_member(string member) {
50 for (int i = 0; i < dim(header); i++)
51 if (header[i] == member)
56 bool has_vendor(string[*] vendors, string vendor) {
57 for (int i = 0; i < dim(vendors); i++)
58 if (vendors[i] == vendor)
63 void read_header(file f) {
64 header = read_line(f);
66 for (int i = 0; i < dim(required_elements); i++)
67 if (!has_header_member(required_elements[i]))
68 fatal("Missing header element \"%s\"\n", required_elements[i]);
71 string[string] read_entry(file f) {
72 string[*] elements = read_line(f);
73 string[string] entry = {};
75 if (dim(header) != dim(elements))
76 fatal("line %d: has %d instead of %d elements (%V)\n",
77 lineno, dim(elements), dim(header), elements);
79 for (int i = 0; i < dim(header); i++) {
80 if (elements[i] == "")
81 elements[i] = "unknown";
82 entry[header[i]] = elements[i];
87 string part_number(string[string] entry)
90 return entry["mfg_part_number"];
92 return entry["vendor_part_number"];
95 string quoted(string v)
97 if (String::index(v, "\"") >= 0 || String::index(v, ",") >= 0) {
98 if (String::index(v, "\"") >= 0) {
100 for (int i = 0; i < String::length(v); i++) {
103 ret = ret + String::new(v[i]);
108 return "\"" + v + "\"";
115 void process_seeed(string[string] entry)
117 if (entry["loadstatus"] == "noload")
120 static bool start = true;
122 printf("Part/Designator,Manufacturer Part Number/Seeed SKU, Quantity\n");
126 string[*] refdes = String::wordsplit(entry["refdes"], " \t");
129 for (int i = 0; i < dim(refdes); i++) {
130 printf("%s", refdes[i]);
131 if (i < dim(refdes) - 1)
136 printf(",%s,%s\n", quoted(entry["mfg_part_number"]), entry["quantity"]);
139 void process_digikey(string[string] entry)
141 if (entry["loadstatus"] == "noload")
143 printf("%s,%s,%s %s\n",
145 quoted(part_number(entry)),
146 quoted(entry["device"]),
147 quoted(entry["value"]));
150 void process_mouser(string[string] entry)
152 if (entry["loadstatus"] == "noload")
154 /* printf("%s|%s\n", part_number(entry), entry["quantity"]); */
156 static bool start = true;
159 printf("Mouser Part Number,Manufacturer Part Number,Quantity 1\n");
164 quoted(entry["vendor_part_number"]),
165 quoted(entry["mfg_part_number"]),
169 void process_other(string[string] entry) {
170 if (entry["loadstatus"] == "noload")
172 printf("%s,%s,%s,%s %s\n",
175 quoted(part_number(entry)),
176 quoted(entry["device"]),
180 void process_file(file f) {
182 while (!File::end(f)) {
183 string[string] entry = read_entry(f);
184 string vendor = entry["vendor"];
185 if (!is_uninit(&vendors) && has_vendor(vendors, "seeed")) {
186 process_seeed(entry);
187 } else if ((!is_uninit(&vendors) && has_vendor(vendors, vendor)) ||
188 (!is_uninit(¬_vendors) && !has_vendor(not_vendors, vendor))) {
189 switch (entry["vendor"]) {
191 process_digikey(entry);
194 process_mouser(entry);
197 process_other(entry);
204 ParseArgs::argdesc argd = {
206 { .var = { .arg_flag = &mfg_part },
209 .desc = "Display manufacturer part number"},
210 { .var = { .arg_string = &vendor_list },
213 .expr_name = "vendors",
214 .desc = "Vendors to match"},
215 { .var = { .arg_string = ¬_vendor_list },
217 .name = "not-vendor",
218 .expr_name = "not-vendor",
219 .desc = "Vendors to exclude"},
225 ParseArgs::parseargs(&argd, &argv);
226 if (!is_uninit(&vendor_list))
227 vendors = String::parse_csv(vendor_list);
229 if (!is_uninit(¬_vendor_list))
230 not_vendors = String::parse_csv(not_vendor_list);
232 if (!is_uninit(&argi)) {
233 for (int i = argi; i < dim(argv); i++)
234 twixt(file f = File::open(argv[i], "r"); File::close(f))